Code Review / yardstick.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | review | tree
raw | inline | side by side
Merge "Rework network_services/vnf_generic/vnf/test_sample_vnf.py:TestVnfSshHelper"
[yardstick.git] / yardstick / tests / unit / common / test_kubernetes_utils.py
diff --git a/yardstick/tests/unit/common/test_kubernetes_utils.py b/yardstick/tests/unit/common/test_kubernetes_utils.py
index bdc2c12..ba6b5f3 100644 (file)
--- a/yardstick/tests/unit/common/test_kubernetes_utils.py
+++ b/yardstick/tests/unit/common/test_kubernetes_utils.py
@@ -121,6 +121,23 @@ class DeleteCustomResourceDefinitionTestCase(base.BaseUnitTestCase):
         mock_delete_crd.delete_custom_resource_definition.\
             assert_called_once_with('name', 'del_obj')
 
+    @mock.patch.object(client, 'V1DeleteOptions', return_value='del_obj')
+    @mock.patch.object(kubernetes_utils, 'get_extensions_v1beta_api')
+    @mock.patch.object(kubernetes_utils, 'LOG')
+    def test_execute_skip_exception(self, mock_log, mock_get_api, mock_delobj):
+        mock_delete_crd = mock.Mock()
+        mock_delete_crd.delete_custom_resource_definition.side_effect = rest.ApiException(
+            status=404)
+
+        mock_get_api.return_value = mock_delete_crd
+        kubernetes_utils.delete_custom_resource_definition('name', skip_codes=[404])
+
+        mock_delobj.assert_called_once()
+        mock_delete_crd.delete_custom_resource_definition.assert_called_once_with(
+            'name', 'del_obj')
+
+        mock_log.info.assert_called_once()
+
 
 class GetCustomResourceDefinitionTestCase(base.BaseUnitTestCase):
 
@@ -159,37 +176,105 @@ class GetCustomResourceDefinitionTestCase(base.BaseUnitTestCase):
             kubernetes_utils.get_custom_resource_definition('kind')
 
 
-class CreateNetworkTestCase(base.BaseUnitTestCase):
+class GetNetworkTestCase(base.BaseUnitTestCase):
     @mock.patch.object(kubernetes_utils, 'get_custom_objects_api')
     def test_execute_correct(self, mock_get_api):
+        mock_api = mock.Mock()
+        mock_get_api.return_value = mock_api
+        group = 'group.com'
+        version = mock.Mock()
+        plural = 'networks'
+        name = 'net_one'
+
+        kubernetes_utils.get_network(
+            constants.SCOPE_CLUSTER, group, version, plural, name)
+        mock_api.get_cluster_custom_object.assert_called_once_with(
+            group, version, plural, name)
+
+        mock_api.reset_mock()
+        kubernetes_utils.get_network(
+            constants.SCOPE_NAMESPACED, group, version, plural, name)
+        mock_api.get_namespaced_custom_object.assert_called_once_with(
+            group, version, 'default', plural, name)
+
+    @mock.patch.object(kubernetes_utils, 'get_custom_objects_api')
+    def test_execute_exception(self, mock_get_api):
+        mock_api = mock.Mock()
+        mock_api.get_cluster_custom_object.side_effect = rest.ApiException(404)
+        mock_api.get_namespaced_custom_object.side_effect = rest.ApiException(404)
+        mock_get_api.return_value = mock_api
+        group = 'group.com'
+        version = mock.Mock()
+        plural = 'networks'
+        name = 'net_one'
+
+        network_obj = kubernetes_utils.get_network(
+            constants.SCOPE_CLUSTER, group, version, plural, name)
+        self.assertIsNone(network_obj)
+
+        mock_api.reset_mock()
+        network_obj = kubernetes_utils.get_network(
+           constants.SCOPE_NAMESPACED, group, version, plural, name)
+        self.assertIsNone(network_obj)
+
+
+class CreateNetworkTestCase(base.BaseUnitTestCase):
+    @mock.patch.object(kubernetes_utils, 'get_custom_objects_api')
+    @mock.patch.object(kubernetes_utils, 'get_network')
+    def test_execute_correct(self, mock_get_net, mock_get_api):
+        mock_get_net.return_value = None
         mock_api = mock.Mock()
         mock_get_api.return_value = mock_api
         group = 'group.com'
         version = mock.Mock()
         plural = 'networks'
         body = mock.Mock()
+        name = 'net_one'
 
         kubernetes_utils.create_network(
-            constants.SCOPE_CLUSTER, group, version, plural, body)
+            constants.SCOPE_CLUSTER, group, version, plural, body, name)
         mock_api.create_cluster_custom_object.assert_called_once_with(
             group, version, plural, body)
 
         mock_api.reset_mock()
         kubernetes_utils.create_network(
-            constants.SCOPE_NAMESPACED, group, version, plural, body)
+            constants.SCOPE_NAMESPACED, group, version, plural, body, name)
         mock_api.create_namespaced_custom_object.assert_called_once_with(
             group, version, 'default', plural, body)
 
+    @mock.patch.object(kubernetes_utils, 'get_custom_objects_api')
+    @mock.patch.object(kubernetes_utils, 'get_network')
+    def test_network_already_created(self, mock_get_net, mock_get_api):
+        mock_get_net.return_value = mock.Mock
+        mock_api = mock.Mock()
+        mock_get_api.return_value = mock_api
+        group = 'group.com'
+        version = mock.Mock()
+        plural = 'networks'
+        body = mock.Mock()
+        name = 'net_one'
+
+        mock_api.reset_mock()
+        kubernetes_utils.create_network(
+            constants.SCOPE_CLUSTER, group, version, plural, body, name)
+        mock_api.create_cluster_custom_object.assert_not_called()
+
+        mock_api.reset_mock()
+        kubernetes_utils.create_network(
+            constants.SCOPE_NAMESPACED, group, version, plural, body, name)
+        mock_api.create_namespaced_custom_object.assert_not_called()
 
     @mock.patch.object(kubernetes_utils, 'get_custom_objects_api')
-    def test_execute_exception(self, mock_get_api):
+    @mock.patch.object(kubernetes_utils, 'get_network')
+    def test_execute_exception(self, mock_get_net, mock_get_api):
+        mock_get_net.return_value = None
         mock_api = mock.Mock()
         mock_api.create_cluster_custom_object.side_effect = rest.ApiException
         mock_get_api.return_value = mock_api
         with self.assertRaises(exceptions.KubernetesApiException):
             kubernetes_utils.create_network(
                 constants.SCOPE_CLUSTER, mock.ANY, mock.ANY, mock.ANY,
-                mock.ANY)
+                mock.ANY, mock.ANY)
 
 
 class DeleteNetworkTestCase(base.BaseUnitTestCase):
@@ -223,6 +308,19 @@ class DeleteNetworkTestCase(base.BaseUnitTestCase):
                 constants.SCOPE_CLUSTER, mock.ANY, mock.ANY, mock.ANY,
                 mock.ANY)
 
+    @mock.patch.object(kubernetes_utils, 'get_custom_objects_api')
+    @mock.patch.object(kubernetes_utils, 'LOG')
+    def test_execute_skip_exception(self, mock_log, mock_get_api):
+        mock_api = mock.Mock()
+        mock_api.delete_cluster_custom_object.side_effect = rest.ApiException(status=404)
+
+        mock_get_api.return_value = mock_api
+        kubernetes_utils.delete_network(
+            constants.SCOPE_CLUSTER, mock.ANY, mock.ANY, mock.ANY,
+            mock.ANY, skip_codes=[404])
+
+        mock_log.info.assert_called_once()
+
 
 class DeletePodTestCase(base.BaseUnitTestCase):
     @mock.patch.object(kubernetes_utils, 'get_core_api')
@@ -243,10 +341,107 @@ class DeletePodTestCase(base.BaseUnitTestCase):
         with self.assertRaises(exceptions.KubernetesApiException):
             kubernetes_utils.delete_pod(mock.ANY, skip_codes=[404])
 
+    @mock.patch.object(kubernetes_utils, 'LOG')
     @mock.patch.object(kubernetes_utils, 'get_core_api')
-    def test_execute_skip_exception(self, mock_get_api):
+    def test_execute_skip_exception(self, mock_get_api, *args):
         mock_api = mock.Mock()
         mock_api.delete_namespaced_pod.side_effect = rest.ApiException(status=404)
 
         mock_get_api.return_value = mock_api
         kubernetes_utils.delete_pod(mock.ANY, skip_codes=[404])
+
+
+class DeleteServiceTestCase(base.BaseUnitTestCase):
+    @mock.patch.object(client, "V1DeleteOptions")
+    @mock.patch.object(kubernetes_utils, 'get_core_api')
+    def test_execute_correct(self, mock_get_api, mock_options):
+        mock_api = mock.Mock()
+        mock_get_api.return_value = mock_api
+        mock_options.return_value = None
+        kubernetes_utils.delete_service("name", "default", None)
+        mock_api.delete_namespaced_service.assert_called_once_with(
+            "name", 'default', None)
+
+    @mock.patch.object(kubernetes_utils, 'get_core_api')
+    def test_execute_exception(self, mock_get_api):
+        mock_api = mock.Mock()
+        mock_api.delete_namespaced_service.side_effect = rest.ApiException(status=200)
+
+        mock_get_api.return_value = mock_api
+        with self.assertRaises(exceptions.KubernetesApiException):
+            kubernetes_utils.delete_service(mock.ANY, skip_codes=[404])
+
+    @mock.patch.object(kubernetes_utils, 'LOG')
+    @mock.patch.object(kubernetes_utils, 'get_core_api')
+    def test_execute_skip_exception(self, mock_get_api, *args):
+        mock_api = mock.Mock()
+        mock_api.delete_namespaced_service.side_effect = rest.ApiException(status=404)
+
+        mock_get_api.return_value = mock_api
+        kubernetes_utils.delete_service(mock.ANY, skip_codes=[404])
+
+
+class DeleteReplicationControllerTestCase(base.BaseUnitTestCase):
+    @mock.patch.object(kubernetes_utils, 'get_core_api')
+    def test_execute_correct(self, mock_get_api):
+        mock_api = mock.Mock()
+        mock_get_api.return_value = mock_api
+        kubernetes_utils.delete_replication_controller(
+            "name", "default", body=None)
+
+        mock_api.delete_namespaced_replication_controller.assert_called_once_with(
+            "name", "default", None)
+
+    @mock.patch.object(kubernetes_utils, 'get_core_api')
+    def test_execute_exception(self, mock_get_api):
+        mock_api = mock.Mock()
+        mock_api.delete_namespaced_replication_controller.side_effect = (
+            rest.ApiException(status=200)
+        )
+
+        mock_get_api.return_value = mock_api
+        with self.assertRaises(exceptions.KubernetesApiException):
+            kubernetes_utils.delete_replication_controller(mock.ANY, skip_codes=[404])
+
+    @mock.patch.object(kubernetes_utils, 'get_core_api')
+    @mock.patch.object(kubernetes_utils, 'LOG')
+    def test_execute_skip_exception(self, mock_log, mock_get_api):
+        mock_api = mock.Mock()
+        mock_api.delete_namespaced_replication_controller.side_effect = (
+            rest.ApiException(status=404)
+        )
+
+        mock_get_api.return_value = mock_api
+        kubernetes_utils.delete_replication_controller(mock.ANY, skip_codes=[404])
+
+        mock_log.info.assert_called_once()
+
+
+class DeleteConfigMapTestCase(base.BaseUnitTestCase):
+    @mock.patch.object(kubernetes_utils, 'get_core_api')
+    def test_execute_correct(self, mock_get_api):
+        mock_api = mock.Mock()
+        mock_get_api.return_value = mock_api
+        kubernetes_utils.delete_config_map("name", body=None)
+        mock_api.delete_namespaced_config_map.assert_called_once_with(
+            "name", "default", None
+        )
+
+    @mock.patch.object(kubernetes_utils, 'get_core_api')
+    def test_execute_exception(self, mock_get_api):
+        mock_api = mock.Mock()
+        mock_api.delete_namespaced_config_map.side_effect = rest.ApiException(status=200)
+
+        mock_get_api.return_value = mock_api
+        with self.assertRaises(exceptions.KubernetesApiException):
+            kubernetes_utils.delete_config_map(mock.ANY, skip_codes=[404])
+
+    @mock.patch.object(kubernetes_utils, 'get_core_api')
+    @mock.patch.object(kubernetes_utils, 'LOG')
+    def test_execute_skip_exception(self, mock_log, mock_get_api):
+        mock_api = mock.Mock()
+        mock_api.delete_namespaced_config_map.side_effect = rest.ApiException(status=404)
+
+        mock_get_api.return_value = mock_api
+        kubernetes_utils.delete_config_map(mock.ANY, skip_codes=[404])
+        mock_log.info.assert_called_once()