Skip to content

Commit c30e25e

Browse files
committed
Fix test
1 parent 7f546c9 commit c30e25e

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

client/src/raw.rs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1601,6 +1601,7 @@ mod tests {
16011601
operatorservice::v1::DeleteNamespaceRequest, workflowservice::v1::ListNamespacesRequest,
16021602
};
16031603
use tonic::IntoRequest;
1604+
use uuid::Uuid;
16041605

16051606
// Just to help make sure some stuff compiles. Not run.
16061607
#[allow(dead_code)]
@@ -1854,6 +1855,11 @@ mod tests {
18541855
mock_provider
18551856
.expect_deployment_options()
18561857
.return_const(Some(deployment_opts.clone()));
1858+
mock_provider.expect_heartbeat_enabled().return_const(false);
1859+
let uuid = Uuid::new_v4();
1860+
mock_provider
1861+
.expect_worker_instance_key()
1862+
.return_const(uuid);
18571863

18581864
let client_worker_set = Arc::new(ClientWorkerSet::new());
18591865
client_worker_set

client/src/worker_registry/mod.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -461,6 +461,7 @@ mod tests {
461461
mock_provider
462462
.expect_worker_instance_key()
463463
.return_const(worker_instance_key);
464+
mock_provider.expect_deployment_options().return_const(None);
464465

465466
if heartbeat_enabled {
466467
mock_provider

0 commit comments

Comments
 (0)