From 775c3ef2f5d936819af50ea13bea99095575f456 Mon Sep 17 00:00:00 2001 From: Brian Ginsburg Date: Mon, 8 Jan 2024 14:38:20 -0800 Subject: [PATCH] test: Handle get providers event --- homestar-runtime/src/worker.rs | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/homestar-runtime/src/worker.rs b/homestar-runtime/src/worker.rs index 5b2004bb..c98d1254 100644 --- a/homestar-runtime/src/worker.rs +++ b/homestar-runtime/src/worker.rs @@ -564,6 +564,7 @@ mod test { // first time check DHT for workflow info let workflow_info_event = rx.recv_async().await.unwrap(); + let get_workflow_providers_event = rx.recv_async().await.unwrap(); // we should have received 2 receipts let next_run_receipt = rx.recv_async().await.unwrap(); @@ -574,6 +575,11 @@ mod test { _ => panic!("Wrong event type"), }; + match get_workflow_providers_event { + Event::GetProviders(QueryRecord { cid, .. }) => assert_eq!(cid, worker_workflow_cid), + _ => panic!("Wrong event type"), + } + let (next_receipt, _wf_info) = match next_run_receipt { Event::CapturedReceipt(Captured { receipt: next_receipt,