fix unit test

aptos-labs · Oct 16, 2024 · cdbebc0 · cdbebc0
1 parent a2d4215
commit cdbebc0
Showing 1 changed file with 20 additions and 8 deletions.
diff --git a/testsuite/forge/src/backend/k8s/swarm.rs b/testsuite/forge/src/backend/k8s/swarm.rs
@@ -872,19 +872,31 @@ mod tests {
     ) -> (Vec<NetworkChaos>, Vec<StressChaos>) {
         let network_chaos = NetworkChaos {
             status: Some(ChaosStatus {
-                conditions: Some(vec![ChaosCondition {
-                    r#type: ChaosConditionType::AllInjected,
-                    status: network_status,
-                }]),
+                conditions: Some(vec![
+                    ChaosCondition {
+                        r#type: ChaosConditionType::AllInjected,
+                        status: network_status.clone(),
+                    },
+                    ChaosCondition {
+                        r#type: ChaosConditionType::Selected,
+                        status: network_status,
+                    },
+                ]),
             }),
             ..NetworkChaos::new("test", Default::default())
         };
         let stress_chaos = StressChaos {
             status: Some(ChaosStatus {
-                conditions: Some(vec![ChaosCondition {
-                    r#type: ChaosConditionType::AllInjected,
-                    status: stress_status,
-                }]),
+                conditions: Some(vec![
+                    ChaosCondition {
+                        r#type: ChaosConditionType::AllInjected,
+                        status: stress_status.clone(),
+                    },
+                    ChaosCondition {
+                        r#type: ChaosConditionType::Selected,
+                        status: stress_status,
+                    },
+                ]),
             }),
             ..StressChaos::new("test", Default::default())
         };