def test_kill_pod_APIcalling(): label_runner = LabelRunner(NodeInventory(None), None, None, RemoteExecutor()) # Patch action of getting nodes to execute kill command on test_node = Node(1) get_node_by_ip_mock = MagicMock(return_value=test_node) label_runner.inventory.get_node_by_ip = get_node_by_ip_mock #Pactch action of switching to APIcalling mode k8s_inventory = MagicMock() k8s_inventory.delete_pods = True label_runner.k8s_inventory = k8s_inventory # Patch action of choosing container k8s_client_mock = MagicMock() label_runner.k8s_inventory.k8s_client = k8s_client_mock delete_pods_mock = MagicMock() label_runner.k8s_inventory.k8s_client.delete_pods = delete_pods_mock metric_collector = MagicMock() label_runner.metric_collector = metric_collector add_pod_killed_metric_mock = MagicMock() label_runner.metric_collector.add_pod_killed_metric = add_pod_killed_metric_mock mock_pod = Pod(name='test', namespace='test') mock_pod.container_ids = ["docker://container1"] label_runner.kill_pod(mock_pod) delete_pods_mock.assert_called_with([mock_pod]) add_pod_killed_metric_mock.assert_called_with(mock_pod)
def test_kill_pod_forced_SSHing(): label_runner = LabelRunner(NodeInventory(None), None, None, SSHExecutor()) # patch metrics collector label_runner.metric_collector = MagicMock() # Patch action of switching to SSHing mode k8s_inventory = MagicMock() label_runner.k8s_inventory = k8s_inventory # Patch action of getting nodes to execute kill command on test_node = Node(1) get_node_by_ip_mock = MagicMock(return_value=test_node) label_runner.inventory.get_node_by_ip = get_node_by_ip_mock # Patch action of choosing container execute_mock = MagicMock() label_runner.executor.execute = execute_mock mock_pod = Pod(name='test', namespace='test') mock_pod.container_ids = ["docker://container1"] mock_pod.labels = {"seal/force-kill": "true"} label_runner.kill_pod(mock_pod) execute_mock.assert_called_once_with( "sudo docker kill -s SIGKILL container1", nodes=[test_node])