Rework actions tests

Author: Devatoria

tests/test_deployment.py

@@ -0,0 +1,64 @@
+# -*- coding: utf-8 -*-
+from unittest.mock import patch, MagicMock, ANY, call
+
+import pytest
+from chaoslib.exceptions import ActivityFailed
+from kubernetes.client.models import V1DeploymentList, V1Deployment, V1ObjectMeta
+
+from chaosk8s.deployment.actions import create_deployment, delete_deployment, scale_deployment
+
+
+@patch('chaosk8s.has_local_config_file', autospec=True)
+def test_cannot_process_other_than_yaml_and_json(has_conf):
+    has_conf.return_value = False
+    path = "./tests/fixtures/invalid-k8s.txt"
+    with pytest.raises(ActivityFailed) as excinfo:
+        create_deployment(spec_path=path)
+    assert "cannot process {path}".format(path=path) in str(excinfo)
+
+
+@patch('builtins.open', autospec=True)
+@patch('chaosk8s.deployment.actions.json', autospec=True)
+@patch('chaosk8s.deployment.actions.create_k8s_api_client', autospec=True)
+@patch('chaosk8s.deployment.actions.client', autospec=True)
+def test_create_deployment(client, api, json, open):
+    v1 = MagicMock()
+    client.AppsV1beta1Api.return_value = v1
+    json.loads.return_value = {"Kind": "Deployment"}
+
+    create_deployment(spec_path="depl.json")
+
+    v1.create_namespaced_deployment.assert_called_with(ANY, body=json.loads.return_value)
+
+
+@patch('chaosk8s.deployment.actions.create_k8s_api_client', autospec=True)
+@patch('chaosk8s.deployment.actions.client', autospec=True)
+def test_delete_deployment(client, api):
+    depl1 = V1Deployment(metadata=V1ObjectMeta(name="depl1"))
+    depl2 = V1Deployment(metadata=V1ObjectMeta(name="depl2"))
+    v1 = MagicMock()
+    client.AppsV1beta1Api.return_value = v1
+    v1.list_namespaced_deployment.return_value = V1DeploymentList(items=(depl1, depl2))
+
+    delete_deployment("fake_name", "fake_ns")
+
+    v1.list_namespaced_deployment.assert_called_with("fake_ns", label_selector=ANY)
+    v1.delete_namespaced_deployment.assert_has_calls(
+        calls=[
+            call(depl1.metadata.name, "fake_ns", ANY),
+            call(depl2.metadata.name, "fake_ns", ANY)
+        ],
+        any_order=True
+    )
+
+
+@patch('chaosk8s.deployment.actions.create_k8s_api_client', autospec=True)
+@patch('chaosk8s.deployment.actions.client', autospec=True)
+def test_scale_deployment(client, api):
+    v1 = MagicMock()
+    client.ExtensionsV1beta1Api.return_value = v1
+
+    scale_deployment("fake", 3, "fake_ns")
+
+    body = {"spec": {"replicas": 3}}
+    v1.patch_namespaced_deployment_scale.assert_called_with("fake", namespace="fake_ns", body=body)

tests/test_actions.py renamed to tests/test_node.py

@@ -5,20 +5,10 @@
 from chaoslib.exceptions import ActivityFailed
 from kubernetes.client.rest import ApiException
 
-from chaosk8s.actions import start_microservice
 from chaosk8s.node.actions import cordon_node, create_node, delete_nodes, \
     uncordon_node, drain_nodes
 
 
-@patch('chaosk8s.has_local_config_file', autospec=True)
-def test_cannot_process_other_than_yaml_and_json(has_conf):
-    has_conf.return_value = False
-    path = "./tests/fixtures/invalid-k8s.txt"
-    with pytest.raises(ActivityFailed) as excinfo:
-        start_microservice(spec_path=path)
-    assert "cannot process {path}".format(path=path) in str(excinfo)
-
-
 @patch('chaosk8s.has_local_config_file', autospec=True)
 @patch('chaosk8s.node.actions.client', autospec=True)
 @patch('chaosk8s.client')
@@ -92,31 +82,6 @@ def test_delete_nodes(cl, client, has_conf):
     v1.delete_node.assert_called_with("mynode", ANY, grace_period_seconds=None)
 
 
-@patch('chaosk8s.has_local_config_file', autospec=True)
-@patch('chaosk8s.node.actions.client', autospec=True)
-@patch('chaosk8s.client')
-def test_delete_nodes(cl, client, has_conf):
-    has_conf.return_value = False
-
-    v1 = MagicMock()
-    client.CoreV1Api.return_value = v1
-
-    node = MagicMock()
-    node.metadata.name = "mynode"
-
-    result = MagicMock()
-    result.items = [node]
-    v1.list_node.return_value = result
-
-    res = MagicMock()
-    res.status = "Success"
-    v1.delete_node.return_value = res
-
-    delete_nodes(label_selector="k=mynode")
-
-    v1.delete_node.assert_called_with("mynode", ANY, grace_period_seconds=None)
-
-
 @patch('chaosk8s.has_local_config_file', autospec=True)
 @patch('chaosk8s.node.actions.client', autospec=True)
 @patch('chaosk8s.client')

tests/test_replicaset.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+from unittest.mock import patch, MagicMock, ANY, call
+
+from kubernetes.client.models import V1ReplicaSetList, V1ReplicaSet, V1ObjectMeta
+
+from chaosk8s.replicaset.actions import delete_replica_set
+
+
+@patch('chaosk8s.replicaset.actions.create_k8s_api_client', autospec=True)
+@patch('chaosk8s.replicaset.actions.client', autospec=True)
+def test_create_deployment(client, api):
+    v1 = MagicMock()
+    client.ExtensionsV1beta1Api.return_value = v1
+    v1.list_namespaced_replica_set.return_value = V1ReplicaSetList(items=(
+        V1ReplicaSet(metadata=V1ObjectMeta(name="repl1")),
+        V1ReplicaSet(metadata=V1ObjectMeta(name="repl2"))
+    ))
+
+    delete_replica_set("fake", "fake_ns")
+
+    v1.list_namespaced_replica_set.assert_called_with("fake_ns", label_selector="name in (fake)")
+    v1.delete_namespaced_replica_set.assert_has_calls(
+        [
+            call("repl1", "fake_ns", ANY),
+            call("repl2", "fake_ns", ANY)
+        ],
+        any_order=True
+    )

tests/test_service.py

@@ -0,0 +1,15 @@
+# -*- coding: utf-8 -*-
+from unittest.mock import patch, MagicMock
+
+from chaosk8s.service.actions import delete_service
+
+
+@patch('chaosk8s.service.actions.create_k8s_api_client', autospec=True)
+@patch('chaosk8s.service.actions.client', autospec=True)
+def test_delete_service(client, api):
+    v1 = MagicMock()
+    client.CoreV1Api.return_value = v1
+
+    delete_service("fake", "fake_ns")
+
+    v1.delete_namespaced_service.assert_called_with("fake", namespace="fake_ns")