improve test_healthy_providers_return_healthy_status unit test

Signed-off-by: Vikrant Palle <[email protected]>
wasmCloud · Dec 23, 2024 · 062130e · 062130e
1 parent df0bf72
commit 062130e
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 2 deletions.
diff --git a/crates/wadm/src/scaler/daemonscaler/provider.rs b/crates/wadm/src/scaler/daemonscaler/provider.rs
@@ -635,7 +635,7 @@ mod test {
                     issuer: "issuer".to_string(),
                     reference: provider_ref.to_string(),
                     hosts: HashMap::from([
-                        (host_id_one.to_string(), ProviderStatus::Pending),
+                        (host_id_one.to_string(), ProviderStatus::Failed),
                         (host_id_two.to_string(), ProviderStatus::Running),
                     ]),
                 },
@@ -667,6 +667,34 @@ mod test {
         );
 
         spreadscaler.reconcile().await?;
+        spreadscaler
+            .handle_event(&Event::ProviderHealthCheckFailed(
+                ProviderHealthCheckFailed {
+                    data: ProviderHealthCheckInfo {
+                        provider_id: provider_id.to_string(),
+                        host_id: host_id_one.to_string(),
+                    },
+                },
+            ))
+            .await?;
+
+        store
+            .store(
+                lattice_id,
+                provider_id.to_string(),
+                Provider {
+                    id: provider_id.to_string(),
+                    name: "provider".to_string(),
+                    issuer: "issuer".to_string(),
+                    reference: provider_ref.to_string(),
+                    hosts: HashMap::from([
+                        (host_id_one.to_string(), ProviderStatus::Pending),
+                        (host_id_two.to_string(), ProviderStatus::Running),
+                    ]),
+                },
+            )
+            .await?;
+
         spreadscaler
             .handle_event(&&Event::ProviderHealthCheckPassed(
                 ProviderHealthCheckPassed {

diff --git a/crates/wadm/src/scaler/spreadscaler/provider.rs b/crates/wadm/src/scaler/spreadscaler/provider.rs
@@ -1360,14 +1360,42 @@ mod test {
                     issuer: "issuer".to_string(),
                     reference: provider_ref.to_string(),
                     hosts: HashMap::from([
-                        (host_id_one.to_string(), ProviderStatus::Pending),
+                        (host_id_one.to_string(), ProviderStatus::Failed),
                         (host_id_two.to_string(), ProviderStatus::Running),
                     ]),
                 },
             )
             .await?;
 
         spreadscaler.reconcile().await?;
+        spreadscaler
+            .handle_event(&Event::ProviderHealthCheckFailed(
+                ProviderHealthCheckFailed {
+                    data: ProviderHealthCheckInfo {
+                        provider_id: provider_id.to_string(),
+                        host_id: host_id_one.to_string(),
+                    },
+                },
+            ))
+            .await?;
+
+        store
+            .store(
+                lattice_id,
+                provider_id.to_string(),
+                Provider {
+                    id: provider_id.to_string(),
+                    name: "provider".to_string(),
+                    issuer: "issuer".to_string(),
+                    reference: provider_ref.to_string(),
+                    hosts: HashMap::from([
+                        (host_id_one.to_string(), ProviderStatus::Pending),
+                        (host_id_two.to_string(), ProviderStatus::Running),
+                    ]),
+                },
+            )
+            .await?;
+
         spreadscaler
             .handle_event(&&Event::ProviderHealthCheckPassed(
                 ProviderHealthCheckPassed {