Skip to content

Commit e3382cb

Browse files
authored
fix(server): update driver spec test argument (#2022)
* fix(server): update driver spec test argument Signed-off-by: Evan Lezar <elezar@nvidia.com> * test(server): stabilize concurrent lease steal test Signed-off-by: Evan Lezar <elezar@nvidia.com> --------- Signed-off-by: Evan Lezar <elezar@nvidia.com>
1 parent 75a317e commit e3382cb

2 files changed

Lines changed: 8 additions & 4 deletions

File tree

crates/openshell-server/src/compute/lease.rs

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -480,14 +480,18 @@ mod tests {
480480
#[tokio::test]
481481
async fn concurrent_steal_exactly_one_wins() {
482482
let store = test_store().await;
483-
let l_holder = lease(store.clone(), "holder", Duration::ZERO);
483+
let test_ttl = Duration::from_millis(100);
484+
let expiry_slack = Duration::from_millis(25);
485+
486+
let l_holder = lease(store.clone(), "holder", test_ttl);
484487
let _guard = l_holder.try_acquire().await.unwrap();
488+
tokio::time::sleep(test_ttl + expiry_slack).await;
485489

486490
let mut tasks = Vec::new();
487491
for i in 0..5 {
488492
let s = store.clone();
489493
tasks.push(tokio::spawn(async move {
490-
let l = lease(s, &format!("standby-{i}"), Duration::ZERO);
494+
let l = lease(s, &format!("standby-{i}"), test_ttl);
491495
l.try_steal_expired().await
492496
}));
493497
}

crates/openshell-server/src/compute/mod.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2172,8 +2172,8 @@ mod tests {
21722172
..Default::default()
21732173
};
21742174

2175-
let driver =
2176-
driver_sandbox_spec_from_public(&public, None).expect("driver spec should map");
2175+
let driver = driver_sandbox_spec_from_public(&public, "test-driver")
2176+
.expect("driver spec should map");
21772177

21782178
let gpu = driver
21792179
.resource_requirements

0 commit comments

Comments
 (0)