github/krkn
1
0
Fork 0
mirror of https://github.com/krkn-chaos/krkn.git synced 2026-02-14 09:59:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add unit tests for Azure class methods in (#1035)

Browse Source 
Signed-off-by: Sai Sanjay <saisanjay7660@gmail.com>
This commit is contained in:
Sai Sanjay
2026-01-02 13:20:34 +00:00
committed by GitHub
parent d9271a4bcc
commit 3e14fe07b7
1 changed files with 41 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
tests/test_az_node_scenarios.py
View File

@@ -10,9 +10,7 @@ Assisted By: Claude Code
"""
import unittest
from unittest.mock import MagicMock, Mock, patch, PropertyMock, call
import logging
import time
from unittest.mock import Mock, patch
from krkn_lib.k8s import KrknKubernetes
from krkn_lib.models.k8s import AffectedNode, AffectedNodeStatus
@@ -353,6 +351,26 @@ class TestAzure(unittest.TestCase):
self.assertTrue(result)
mock_affected_node.set_affected_node_status.assert_called_once()
@patch('time.sleep')
@patch('logging.info')
@patch('krkn.scenario_plugins.node_actions.az_node_scenarios.NetworkManagementClient')
@patch('krkn.scenario_plugins.node_actions.az_node_scenarios.ComputeManagementClient')
@patch('krkn.scenario_plugins.node_actions.az_node_scenarios.DefaultAzureCredential')
def test_wait_until_stopped_timeout(self, mock_credential, mock_compute, mock_network, mock_logging, mock_sleep):
"""Test wait_until_stopped returns False on timeout"""
azure = Azure()
mock_status = Mock()
mock_status.code = "PowerState/stopping"
mock_instance_view = Mock()
mock_instance_view.statuses = [Mock(), mock_status]
azure.compute_client.virtual_machines.instance_view.return_value = mock_instance_view
result = azure.wait_until_stopped("test-rg", "test-vm", 10, None)
self.assertFalse(result)
@patch('time.sleep')
@patch('logging.info')
@patch('krkn.scenario_plugins.node_actions.az_node_scenarios.NetworkManagementClient')
@@ -436,6 +454,26 @@ class TestAzure(unittest.TestCase):
azure.network_client.network_security_groups.begin_delete.assert_called_once_with("test-rg", "chaos")
@patch('builtins.print')
@patch('krkn.scenario_plugins.node_actions.az_node_scenarios.NetworkManagementClient')
@patch('krkn.scenario_plugins.node_actions.az_node_scenarios.ComputeManagementClient')
@patch('krkn.scenario_plugins.node_actions.az_node_scenarios.DefaultAzureCredential')
def test_delete_security_group_with_result(self, mock_credential, mock_compute, mock_network, mock_print):
"""Test delete_security_group deletes NSG with non-None result"""
azure = Azure()
mock_result = Mock()
mock_result.as_dict.return_value = {"id": "/test-nsg-id", "name": "chaos"}
mock_operation = Mock()
mock_operation.result.return_value = mock_result
azure.network_client.network_security_groups.begin_delete.return_value = mock_operation
azure.delete_security_group("test-rg", "chaos")
azure.network_client.network_security_groups.begin_delete.assert_called_once_with("test-rg", "chaos")
mock_result.as_dict.assert_called_once()
mock_print.assert_called_once_with({"id": "/test-nsg-id", "name": "chaos"})
@patch('krkn.scenario_plugins.node_actions.az_node_scenarios.NetworkManagementClient')
@patch('krkn.scenario_plugins.node_actions.az_node_scenarios.ComputeManagementClient')
@patch('krkn.scenario_plugins.node_actions.az_node_scenarios.DefaultAzureCredential')