krkn-lib 5.1.11__py3-none-any.whl → 5.1.13__py3-none-any.whl

krkn_lib/models/telemetry/models.py

@@ -545,6 +545,10 @@ class ChaosRunTelemetry:
     """
     Virt checks of VMIs
     """
+    post_virt_checks: list[VirtCheck] = None
+    """
+    Post Scenario Virt checks of VMIs
+    """
     job_status: bool = True
     """
     Overall job status, will take all scenario's exit status
@@ -597,6 +601,11 @@ class ChaosRunTelemetry:
                 if json_dict.get("virt_checks")
                 else None
             )
+            self.post_virt_checks = (
+                [VirtCheck(k) for k in json_dict.get("post_virt_checks")]
+                if json_dict.get("post_virt_checks")
+                else None
+            )
             self.job_status = json_dict.get("job_status")
             self.build_url = json_dict.get("build_url")
 

krkn_lib/tests/base_test.py

@@ -608,6 +608,19 @@ class BaseTest(unittest.TestCase):
                     "new_ip_address": "0.0.0.3"
                 },
             ],
+            "post_virt_checks": [
+                {
+                    "node_name": "h10-r660",
+                    "namespace": "benchmark-runner",
+                    "ip_address": "0.0.0.0",
+                    "vm_name": "windows-vm-52",
+                    "status": False,
+                    "start_timestamp": "2025-03-12T14:57:55.904352",
+                    "end_timestamp": "2025-03-12T14:57:55.904352",
+                    "duration": 0.00,
+                    "new_ip_address": ""
+                }
+            ],
             "total_node_count": 3,
             "cloud_infrastructure": "AWS",
             "cloud_type": "EC2",

krkn_lib/tests/test_krkn_elastic_models.py

@@ -246,6 +246,27 @@ class TestKrknElasticModels(BaseTest):
         )
         self.assertEqual(elastic_telemetry.virt_checks[0].duration, 20.348474)
 
+        # post_virt_checks
+        self.assertEqual(len(elastic_telemetry.post_virt_checks), 1)
+        self.assertEqual(
+            elastic_telemetry.post_virt_checks[0].vm_name, "windows-vm-52"
+        )
+        self.assertEqual(
+            elastic_telemetry.post_virt_checks[0].ip_address, "0.0.0.0"
+        )
+        self.assertEqual(
+            elastic_telemetry.post_virt_checks[0].new_ip_address, ""
+        )
+
+        self.assertEqual(
+            elastic_telemetry.post_virt_checks[0].namespace, "benchmark-runner"
+        )
+        self.assertEqual(
+            elastic_telemetry.post_virt_checks[0].node_name, "h10-r660"
+        )
+
+        self.assertEqual(elastic_telemetry.post_virt_checks[0].status, False)
+
         self.assertEqual(elastic_telemetry.total_node_count, 3)
         self.assertEqual(elastic_telemetry.cloud_infrastructure, "AWS")
         self.assertEqual(elastic_telemetry.cloud_type, "EC2")