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

krkn_lib/tests/test_krkn_kubernetes_models.py

@@ -1,10 +1,10 @@
-from krkn_lib.models.krkn import HogConfig, HogType
 from krkn_lib.models.k8s import (
     AffectedNode,
     AffectedNodeStatus,
     AffectedPod,
     PodsStatus,
 )
+from krkn_lib.models.krkn import HogConfig, HogType
 from krkn_lib.tests import BaseTest
 
 

krkn_lib/tests/test_krkn_kubernetes_pods_monitor_models.py

@@ -1,13 +1,12 @@
 import time
 import unittest
 
-
-from krkn_lib.models.k8s import PodsStatus, AffectedPod
+from krkn_lib.models.k8s import AffectedPod, PodsStatus
 from krkn_lib.models.pod_monitor.models import (
-    PodEvent,
-    PodStatus,
     MonitoredPod,
+    PodEvent,
     PodsSnapshot,
+    PodStatus,
 )
 
 

krkn_lib/tests/test_krkn_kubernetes_virt.py

@@ -1,6 +1,39 @@
+#!/usr/bin/env python3
+
+
+"""
+Test suite for KubeVirt/virtualization-related functionality in KrknKubernetes
+
+This test suite covers KubeVirt-specific methods in the KrknKubernetes class,
+including operations on VirtualMachines (VMs), VirtualMachineInstances (VMIs),
+and VirtualMachineSnapshots.
+
+Tested functionality:
+- VM operations: get, list, delete, patch
+- VMI operations: get, list, create, delete, patch
+- Snapshot operations: get, create, delete
+
+Usage:
+    # Run all tests in this file
+    python -m unittest src.krkn_lib.tests.test_krkn_kubernetes_virt -v
+
+    # Run a specific test class
+    python -m unittest src.krkn_lib.tests.test_krkn_kubernetes_virt.TestKrknKubernetesVirt -v
+
+    # Run a specific test method
+    python -m unittest src.krkn_lib.tests.test_krkn_kubernetes_virt.TestKrknKubernetesVirt.test_get_vm_success -v
+
+    # Run with coverage
+    python -m coverage run -a -m unittest src.krkn_lib.tests.test_krkn_kubernetes_virt -v
+
+Assisted By: Claude Code
+"""  # NOQA
+
 import unittest
-from unittest.mock import patch, MagicMock, PropertyMock
+from unittest.mock import MagicMock, PropertyMock, patch
+
 from kubernetes.client.rest import ApiException
+
 from krkn_lib.k8s.krkn_kubernetes import KrknKubernetes
 
 
@@ -171,9 +204,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
             return_value=[namespace],
         ):
             # Configure the mock to return vmis
-            mock_list = (
-                self.mock_custom_client.list_namespaced_custom_object
-            )
+            mock_list = self.mock_custom_client.list_namespaced_custom_object
             mock_list.return_value = vmis_response
 
             # get_vmis returns a list
@@ -198,9 +229,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
             return_value=[namespace],
         ):
             # Configure the mock to raise 404
-            mock_list = (
-                self.mock_custom_client.list_namespaced_custom_object
-            )
+            mock_list = self.mock_custom_client.list_namespaced_custom_object
             mock_list.side_effect = api_exception
 
             result = self.lib_k8s.get_vmis(regex_name, namespace)
@@ -224,9 +253,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
             return_value=namespaces,
         ):
             # Configure mock for different responses
-            mock_list = (
-                self.mock_custom_client.list_namespaced_custom_object
-            )
+            mock_list = self.mock_custom_client.list_namespaced_custom_object
             mock_list.side_effect = [
                 vmis_response_1,
                 vmis_response_2,
@@ -264,9 +291,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
             return_value=[namespace],
         ):
             # Configure the mock to return vms
-            mock_list = (
-                self.mock_custom_client.list_namespaced_custom_object
-            )
+            mock_list = self.mock_custom_client.list_namespaced_custom_object
             mock_list.return_value = vms_response
 
             result = self.lib_k8s.get_vms(regex_name, namespace)
@@ -289,9 +314,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
             return_value=[namespace],
         ):
             # Configure the mock to raise 404
-            mock_list = (
-                self.mock_custom_client.list_namespaced_custom_object
-            )
+            mock_list = self.mock_custom_client.list_namespaced_custom_object
             mock_list.side_effect = api_exception
 
             result = self.lib_k8s.get_vms(regex_name, namespace)
@@ -314,9 +337,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
             return_value=namespaces,
         ):
             # Configure mock for different responses
-            mock_list = (
-                self.mock_custom_client.list_namespaced_custom_object
-            )
+            mock_list = self.mock_custom_client.list_namespaced_custom_object
             mock_list.side_effect = [
                 vms_response_1,
                 vms_response_2,
@@ -338,9 +359,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         }
 
         # Configure the mock to return expected response
-        mock_delete = (
-            self.mock_custom_client.delete_namespaced_custom_object
-        )
+        mock_delete = self.mock_custom_client.delete_namespaced_custom_object
         mock_delete.return_value = expected_response
 
         result = self.lib_k8s.delete_vm(vm_name, namespace)
@@ -361,9 +380,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         api_exception = ApiException(status=404)
 
         # Configure the mock to raise 404
-        mock_delete = (
-            self.mock_custom_client.delete_namespaced_custom_object
-        )
+        mock_delete = self.mock_custom_client.delete_namespaced_custom_object
         mock_delete.side_effect = api_exception
 
         result = self.lib_k8s.delete_vm(vm_name, namespace)
@@ -376,9 +393,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         api_exception = ApiException(status=500)
 
         # Configure the mock to raise 500
-        mock_delete = (
-            self.mock_custom_client.delete_namespaced_custom_object
-        )
+        mock_delete = self.mock_custom_client.delete_namespaced_custom_object
         mock_delete.side_effect = api_exception
 
         with self.assertRaises(ApiException):
@@ -389,27 +404,21 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         vmi_name = "test-vmi"
         namespace = "test-ns"
 
-        # Mock logging
-        with patch("krkn_lib.k8s.krkn_kubernetes.logging") as mock_logging:
-            # Configure the mock to return None (success)
-            mock_delete = (
-                self.mock_custom_client.delete_namespaced_custom_object
-            )
-            mock_delete.return_value = None
-
-            result = self.lib_k8s.delete_vmi(vmi_name, namespace)
-
-            mock_delete.assert_called_once_with(
-                group="kubevirt.io",
-                version="v1",
-                namespace=namespace,
-                plural="virtualmachineinstances",
-                name=vmi_name,
-            )
-            # delete_vmi doesn't explicitly return on success (returns None)
-            self.assertIsNone(result)
-            # Verify logging was called
-            mock_logging.info.assert_called_once()
+        # Configure the mock to return None (success)
+        mock_delete = self.mock_custom_client.delete_namespaced_custom_object
+        mock_delete.return_value = None
+
+        result = self.lib_k8s.delete_vmi(vmi_name, namespace)
+
+        mock_delete.assert_called_once_with(
+            group="kubevirt.io",
+            version="v1",
+            namespace=namespace,
+            plural="virtualmachineinstances",
+            name=vmi_name,
+        )
+        # delete_vmi doesn't explicitly return on success (returns None)
+        self.assertIsNone(result)
 
     def test_delete_vmi_not_found(self):
         """Test delete_vmi returns 1 when VMI doesn't exist"""
@@ -417,17 +426,13 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         namespace = "test-ns"
         api_exception = ApiException(status=404)
 
-        # Mock logging
-        with patch("krkn_lib.k8s.krkn_kubernetes.logging"):
-            # Configure the mock to raise 404
-            mock_delete = (
-                self.mock_custom_client.delete_namespaced_custom_object
-            )
-            mock_delete.side_effect = api_exception
+        # Configure the mock to raise 404
+        mock_delete = self.mock_custom_client.delete_namespaced_custom_object
+        mock_delete.side_effect = api_exception
 
-            result = self.lib_k8s.delete_vmi(vmi_name, namespace)
-            # Returns 1 on 404
-            self.assertEqual(result, 1)
+        result = self.lib_k8s.delete_vmi(vmi_name, namespace)
+        # Returns 1 on 404
+        self.assertEqual(result, 1)
 
     def test_delete_vmi_api_error(self):
         """Test delete_vmi returns 1 on API error"""
@@ -435,17 +440,13 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         namespace = "test-ns"
         api_exception = ApiException(status=500)
 
-        # Mock logging
-        with patch("krkn_lib.k8s.krkn_kubernetes.logging"):
-            # Configure the mock to raise 500
-            mock_delete = (
-                self.mock_custom_client.delete_namespaced_custom_object
-            )
-            mock_delete.side_effect = api_exception
+        # Configure the mock to raise 500
+        mock_delete = self.mock_custom_client.delete_namespaced_custom_object
+        mock_delete.side_effect = api_exception
 
-            result = self.lib_k8s.delete_vmi(vmi_name, namespace)
-            # Returns 1 on error
-            self.assertEqual(result, 1)
+        result = self.lib_k8s.delete_vmi(vmi_name, namespace)
+        # Returns 1 on error
+        self.assertEqual(result, 1)
 
     def test_get_snapshot_success(self):
         """Test get_snapshot returns snapshot when it exists"""
@@ -467,10 +468,10 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         result = self.lib_k8s.get_snapshot(snapshot_name, namespace)
 
         mock_get.assert_called_once_with(
-            group="kubevirt.io",
-            version="v1",
+            group="snapshot.kubevirt.io",
+            version="v1beta1",
             namespace=namespace,
-            plural="VirtualMachineSnapshot",
+            plural="virtualmachinesnapshots",
             name=snapshot_name,
         )
         self.assertEqual(result, expected_snapshot)
@@ -506,33 +507,20 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         snapshot_name = "test-snapshot"
         namespace = "test-ns"
 
-        # Mock the logger and snapshot_name attributes
-        self.lib_k8s.logger = MagicMock()
-        self.lib_k8s.snapshot_name = snapshot_name
-
         # Configure the mock to return None (success)
-        mock_delete = (
-            self.mock_custom_client.delete_namespaced_custom_object
-        )
+        mock_delete = self.mock_custom_client.delete_namespaced_custom_object
         mock_delete.return_value = None
 
         # Should not raise any exception
         self.lib_k8s.delete_snapshot(snapshot_name, namespace)
 
         mock_delete.assert_called_once_with(
-            group="kubevirt.io",
-            version="v1",
+            group="snapshot.kubevirt.io",
+            version="v1beta1",
             namespace=namespace,
-            plural="VirtualMachineSnapshot",
+            plural="virtualmachinesnapshots",
             name=snapshot_name,
         )
-        # Verify logger was called (uses self.snapshot_name, not parameter)
-        self.lib_k8s.logger.info.assert_any_call(
-            f"Deleting snapshot '{snapshot_name}'..."
-        )
-        self.lib_k8s.logger.info.assert_any_call(
-            f"Snapshot '{snapshot_name}' deleted successfully."
-        )
 
     def test_delete_snapshot_not_found(self):
         """Test delete_snapshot handles deletion gracefully"""
@@ -540,48 +528,26 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         namespace = "test-ns"
         api_exception = ApiException(status=404)
 
-        # Mock the logger and snapshot_name attributes
-        self.lib_k8s.logger = MagicMock()
-        self.lib_k8s.snapshot_name = snapshot_name
-
         # Configure the mock to raise 404
-        mock_delete = (
-            self.mock_custom_client.delete_namespaced_custom_object
-        )
+        mock_delete = self.mock_custom_client.delete_namespaced_custom_object
         mock_delete.side_effect = api_exception
 
-        # Should not raise exception, but log warning
+        # Should not raise exception
         self.lib_k8s.delete_snapshot(snapshot_name, namespace)
 
-        # Verify warning was logged
-        self.lib_k8s.logger.warning.assert_called_once()
-        warning_call_args = self.lib_k8s.logger.warning.call_args[0][0]
-        self.assertIn("Failed to delete snapshot", warning_call_args)
-
     def test_delete_snapshot_api_error(self):
         """Test delete_snapshot handles API errors gracefully"""
         snapshot_name = "test-snapshot"
         namespace = "test-ns"
         api_exception = ApiException(status=500)
 
-        # Mock the logger and snapshot_name attributes
-        self.lib_k8s.logger = MagicMock()
-        self.lib_k8s.snapshot_name = snapshot_name
-
         # Configure the mock to raise 500
-        mock_delete = (
-            self.mock_custom_client.delete_namespaced_custom_object
-        )
+        mock_delete = self.mock_custom_client.delete_namespaced_custom_object
         mock_delete.side_effect = api_exception
 
-        # Should not raise exception, but log warning
+        # Should not raise exception
         self.lib_k8s.delete_snapshot(snapshot_name, namespace)
 
-        # Verify warning was logged
-        self.lib_k8s.logger.warning.assert_called_once()
-        warning_call_args = self.lib_k8s.logger.warning.call_args[0][0]
-        self.assertIn("Failed to delete snapshot", warning_call_args)
-
     def test_create_vmi_success(self):
         """Test create_vmi successfully creates a VMI"""
         vmi_name = "test-vmi"
@@ -599,9 +565,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         }
 
         # Configure the mock to return expected_vmi
-        mock_create = (
-            self.mock_custom_client.create_namespaced_custom_object
-        )
+        mock_create = self.mock_custom_client.create_namespaced_custom_object
         mock_create.return_value = expected_vmi
 
         result = self.lib_k8s.create_vmi(
@@ -613,7 +577,6 @@ class TestKrknKubernetesVirt(unittest.TestCase):
             version="v1",
             namespace=namespace,
             plural="virtualmachineinstances",
-            name=vmi_name,
             body=vmi_body,
         )
         self.assertEqual(result, expected_vmi)
@@ -627,9 +590,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         api_exception = ApiException(status=404)
 
         # Configure the mock to raise 404
-        mock_create = (
-            self.mock_custom_client.create_namespaced_custom_object
-        )
+        mock_create = self.mock_custom_client.create_namespaced_custom_object
         mock_create.side_effect = api_exception
 
         result = self.lib_k8s.create_vmi(
@@ -646,15 +607,11 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         api_exception = ApiException(status=500)
 
         # Configure the mock to raise 500
-        mock_create = (
-            self.mock_custom_client.create_namespaced_custom_object
-        )
+        mock_create = self.mock_custom_client.create_namespaced_custom_object
         mock_create.side_effect = api_exception
 
         with self.assertRaises(ApiException):
-            self.lib_k8s.create_vmi(
-                vmi_name, namespace, vm_name, vmi_body
-            )
+            self.lib_k8s.create_vmi(vmi_name, namespace, vm_name, vmi_body)
 
     def test_patch_vm_success(self):
         """Test patch_vm successfully patches a VM"""
@@ -672,9 +629,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         }
 
         # Configure the mock to return expected_vm
-        mock_patch = (
-            self.mock_custom_client.patch_namespaced_custom_object
-        )
+        mock_patch = self.mock_custom_client.patch_namespaced_custom_object
         mock_patch.return_value = expected_vm
 
         result = self.lib_k8s.patch_vm(vm_name, namespace, vm_body)
@@ -697,9 +652,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         api_exception = ApiException(status=404)
 
         # Configure the mock to raise 404
-        mock_patch = (
-            self.mock_custom_client.patch_namespaced_custom_object
-        )
+        mock_patch = self.mock_custom_client.patch_namespaced_custom_object
         mock_patch.side_effect = api_exception
 
         result = self.lib_k8s.patch_vm(vm_name, namespace, vm_body)
@@ -713,9 +666,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         api_exception = ApiException(status=500)
 
         # Configure the mock to raise 500
-        mock_patch = (
-            self.mock_custom_client.patch_namespaced_custom_object
-        )
+        mock_patch = self.mock_custom_client.patch_namespaced_custom_object
         mock_patch.side_effect = api_exception
 
         with self.assertRaises(ApiException):
@@ -740,9 +691,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         }
 
         # Configure the mock to return expected_vmi
-        mock_patch = (
-            self.mock_custom_client.patch_namespaced_custom_object
-        )
+        mock_patch = self.mock_custom_client.patch_namespaced_custom_object
         mock_patch.return_value = expected_vmi
 
         result = self.lib_k8s.patch_vmi(vmi_name, namespace, vmi_body)
@@ -765,9 +714,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         api_exception = ApiException(status=404)
 
         # Configure the mock to raise 404
-        mock_patch = (
-            self.mock_custom_client.patch_namespaced_custom_object
-        )
+        mock_patch = self.mock_custom_client.patch_namespaced_custom_object
         mock_patch.side_effect = api_exception
 
         result = self.lib_k8s.patch_vmi(vmi_name, namespace, vmi_body)
@@ -781,9 +728,7 @@ class TestKrknKubernetesVirt(unittest.TestCase):
         api_exception = ApiException(status=500)
 
         # Configure the mock to raise 500
-        mock_patch = (
-            self.mock_custom_client.patch_namespaced_custom_object
-        )
+        mock_patch = self.mock_custom_client.patch_namespaced_custom_object
         mock_patch.side_effect = api_exception
 
         with self.assertRaises(ApiException):

krkn_lib/tests/test_krkn_openshift.py

@@ -23,7 +23,6 @@ from krkn_lib.ocp.krkn_openshift import KrknOpenshift
 from krkn_lib.tests import BaseTest
 from krkn_lib.utils import SafeLogger
 
-
 # ==============================================================================
 # UNIT TESTS (Mocked - No external dependencies)
 # ==============================================================================
@@ -256,7 +255,9 @@ class TestGetCloudInfrastructure(unittest.TestCase):
         self.assertEqual(result, "Unknown")
 
     @patch.object(KrknOpenshift, "api_client", new_callable=PropertyMock)
-    def test_get_cloud_infrastructure_no_api_client(self, mock_api_client_prop):
+    def test_get_cloud_infrastructure_no_api_client(
+        self, mock_api_client_prop
+    ):
         """Test when api_client is None."""
         mock_api_client_prop.return_value = None
 
@@ -344,9 +345,7 @@ class TestGetPrometheusApiConnectionData(unittest.TestCase):
                     "items": [
                         {
                             "metadata": {"name": "prometheus-k8s"},
-                            "spec": {
-                                "host": host
-                            },
+                            "spec": {"host": host},
                         }
                     ]
                 }
@@ -503,7 +502,9 @@ class TestFilterMustGatherOcpLogFolder(unittest.TestCase):
                 [r"(\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\.\d+Z).+"],
             )
 
-        self.assertIn("Log destination dir do not exist", str(context.exception))
+        self.assertIn(
+            "Log destination dir do not exist", str(context.exception)
+        )
 
 
 # ==============================================================================

krkn_lib/tests/test_krkn_telemetry_kubernetes.py

@@ -432,9 +432,7 @@ class TestSetParametersBase64(unittest.TestCase):
 
         try:
             scenario_telemetry = ScenarioTelemetry()
-            self.telemetry.set_parameters_base64(
-                scenario_telemetry, temp_path
-            )
+            self.telemetry.set_parameters_base64(scenario_telemetry, temp_path)
 
             decoded = base64.b64decode(
                 scenario_telemetry.parameters_base64.encode()

krkn_lib/tests/test_safe_logger.py

@@ -347,7 +347,9 @@ class TestSafeLoggerWorkerThread(unittest.TestCase):
         # Worker thread should be running
         import threading
 
-        threads = [t for t in threading.enumerate() if t.name == "SafeLogWriter"]
+        threads = [
+            t for t in threading.enumerate() if t.name == "SafeLogWriter"
+        ]
         # Should have at least one SafeLogWriter thread
         self.assertGreater(len(threads), 0)
         # All SafeLogWriter threads should be daemon threads

krkn_lib/tests/test_utils.py

@@ -29,11 +29,10 @@ class UtilFunctionTests(BaseTest):
         test_workdir = os.path.join(workdir_basepath, workdir)
         os.mkdir(test_workdir)
         test_string = "Tester McTesty!"
-        with tempfile.NamedTemporaryFile(
-            dir=test_workdir
-        ) as src, tempfile.NamedTemporaryFile(
-            dir=test_workdir
-        ) as dst:  # NOQA
+        with (
+            tempfile.NamedTemporaryFile(dir=test_workdir) as src,
+            tempfile.NamedTemporaryFile(dir=test_workdir) as dst,
+        ):  # NOQA
             with open(src.name, "w+") as source, open(dst.name, "w+") as dest:
                 encoded_test_byte = base64.b64encode(
                     test_string.encode("utf-8")

krkn_lib/utils/functions.py

@@ -25,9 +25,10 @@ def decode_base64_file(source_filename: str, destination_filename: str):
     :param source_filename: source base64 encoded file
     :param destination_filename: destination decoded file
     """
-    with open(source_filename, "rb") as encoded_source, open(
-        destination_filename, "wb"
-    ) as target:
+    with (
+        open(source_filename, "rb") as encoded_source,
+        open(destination_filename, "wb") as target,
+    ):
         with Base64IO(encoded_source) as source:
             for line in source:
                 target.write(line)

krkn_lib/version/version.py

@@ -1,3 +1,6 @@
-import importlib.metadata
+try:
+    import importlib_metadata
+except ImportError:
+    import importlib.metadata as importlib_metadata
 
-__version__ = importlib.metadata.version("krkn-lib")
+__version__ = importlib_metadata.version("krkn-lib")

{krkn_lib-5.1.13.dist-info → krkn_lib-6.0.1.dist-info}/METADATA

@@ -1,19 +1,16 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: krkn-lib
-Version: 5.1.13
+Version: 6.0.1
 Summary: Foundation library for Kraken
+Home-page: https://github.com/redhat-chaos/krkn
 License: Apache-2.0
-License-File: LICENSE
 Author: Red Hat Chaos Team
-Requires-Python: >=3.9,<4.0
+Requires-Python: >=3.11,<4.0
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Classifier: Programming Language :: Python :: 3.14
 Requires-Dist: PyYAML (==6.0.1)
 Requires-Dist: base64io (>=1.0.3,<2.0.0)
 Requires-Dist: coverage (>=7.6.12,<8.0.0)
@@ -31,8 +28,7 @@ Requires-Dist: requests (>=2.29.0,<3.0.0)
 Requires-Dist: sphinx-rtd-theme (>=1.2.2,<2.0.0)
 Requires-Dist: sphinxnotes-markdown-builder (>=0.5.6,<0.6.0)
 Requires-Dist: tzlocal (==5.1)
-Requires-Dist: wheel (>=0.42.0,<0.43.0)
-Project-URL: Homepage, https://github.com/redhat-chaos/krkn
+Requires-Dist: wheel (==0.46.2)
 Description-Content-Type: text/markdown
 
 ![action](https://github.com/krkn-chaos/krkn-lib/actions/workflows/build.yaml/badge.svg)