krkn-lib 5.1.10__tar.gz → 5.1.12__tar.gz

{krkn_lib-5.1.10 → krkn_lib-5.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: krkn-lib
-Version: 5.1.10
+Version: 5.1.12
 Summary: Foundation library for Kraken
 License: Apache-2.0
 License-File: LICENSE
@@ -19,8 +19,8 @@ Requires-Dist: base64io (>=1.0.3,<2.0.0)
 Requires-Dist: coverage (>=7.6.12,<8.0.0)
 Requires-Dist: cython (==3.0)
 Requires-Dist: deprecation (==2.1.0)
-Requires-Dist: elasticsearch (==8.19.1)
-Requires-Dist: elasticsearch-dsl (==8.18.0)
+Requires-Dist: elasticsearch (==7.13.4)
+Requires-Dist: elasticsearch-dsl (==7.4.1)
 Requires-Dist: kubeconfig (>=1.1.1,<2.0.0)
 Requires-Dist: kubernetes (==34.1.0)
 Requires-Dist: numpy (==1.26.4)

{krkn_lib-5.1.10 → krkn_lib-5.1.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "krkn-lib"
-version = "5.1.10"
+version = "5.1.12"
 description = "Foundation library for Kraken"
 authors = ["Red Hat Chaos Team"]
 license = "Apache-2.0"
@@ -20,8 +20,8 @@ tzlocal = "5.1"
 pytz = "^2023.3"
 PyYAML = "6.0.1"
 prometheus-api-client = "^0.5.4"
-elasticsearch = "8.19.1"
-elasticsearch-dsl = "8.18.0"
+elasticsearch = "7.13.4"
+elasticsearch-dsl = "7.4.1"
 wheel = "^0.42.0"
 cython = "3.0"
 numpy= "1.26.4"

{krkn_lib-5.1.10 → krkn_lib-5.1.12}/src/krkn_lib/models/elastic/models.py

@@ -126,7 +126,6 @@ class ElasticVirtChecks(InnerDoc):
     end_timestamp = Date()
     duration = Float()
 
-
 class ElasticChaosRunTelemetry(Document):
     scenarios = Nested(ElasticScenarioTelemetry, multi=True)
     node_summary_infos = Nested(ElasticNodeInfo, multi=True)
@@ -144,6 +143,7 @@ class ElasticChaosRunTelemetry(Document):
     run_uuid = Text(fields={"keyword": Keyword()})
     health_checks = Nested(ElasticHealthChecks, multi=True)
     virt_checks = Nested(ElasticVirtChecks, multi=True)
+    post_virt_checks = Nested(ElasticVirtChecks, multi=True)
 
     class Index:
         name = "chaos_run_telemetry"
@@ -261,6 +261,28 @@ class ElasticChaosRunTelemetry(Document):
         else:
             self.virt_checks = None
 
+        if chaos_run_telemetry.post_virt_checks:
+            self.post_virt_checks = [
+                ElasticVirtChecks(
+                    vm_name=post_info.vm_name,
+                    ip_address=post_info.ip_address,
+                    new_ip_address=post_info.new_ip_address,
+                    namespace=post_info.namespace,
+                    node_name=post_info.node_name,
+                    status=post_info.status,
+                    start_timestamp=datetime.datetime.fromisoformat(
+                        str(post_info.start_timestamp)
+                    ),
+                    end_timestamp=datetime.datetime.fromisoformat(
+                        str(post_info.end_timestamp)
+                    ),
+                    duration=post_info.duration,
+                )
+                for post_info in chaos_run_telemetry.post_virt_checks
+            ]
+        else:
+            self.post_virt_checks = None
+
         self.timestamp = chaos_run_telemetry.timestamp
         self.total_node_count = chaos_run_telemetry.total_node_count
         self.cloud_infrastructure = chaos_run_telemetry.cloud_infrastructure

{krkn_lib-5.1.10 → krkn_lib-5.1.12}/src/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-5.1.10 → krkn_lib-5.1.12}/src/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-5.1.10 → krkn_lib-5.1.12}/src/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")