test: Handle get providers event

ipvm-wg · Jan 8, 2024 · 775c3ef · 775c3ef
1 parent 42319c0
commit 775c3ef
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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,