From de9369e1e25f62ef8dbee20032e1c79721b7927a Mon Sep 17 00:00:00 2001 From: Kennan Date: Mon, 30 Mar 2015 21:51:52 +0800 Subject: [PATCH] Add service test for kube handler Partial-Bug: #1437977 Change-Id: I932d446e6f372bf1c9ea6607197b623ccda663f8 --- magnum/tests/conductor/handlers/test_kube.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/magnum/tests/conductor/handlers/test_kube.py b/magnum/tests/conductor/handlers/test_kube.py index c1fb67c696..17f789dc54 100644 --- a/magnum/tests/conductor/handlers/test_kube.py +++ b/magnum/tests/conductor/handlers/test_kube.py @@ -230,9 +230,26 @@ class TestKube(base.TestCase): with patch.object(self.kube_handler, 'kube_cli') as mock_kube_cli: mock_kube_cli.service_create.return_value = True - self.kube_handler.service_create({}, expected_service) + self.kube_handler.service_create(self.context, expected_service) mock_kube_cli.service_create.assert_called_once_with( expected_master_url, expected_service) + expected_service.create.assert_called_once_with(self.context) + + @patch('magnum.conductor.handlers.kube._retrieve_k8s_master_url') + def test_service_create_with_failure(self, + mock_retrieve_k8s_master_url): + expected_master_url = 'api_address' + expected_service = self.mock_service() + expected_service.create = mock.MagicMock() + + mock_retrieve_k8s_master_url.return_value = expected_master_url + with patch.object(self.kube_handler, 'kube_cli') as mock_kube_cli: + mock_kube_cli.service_create.return_value = False + + self.kube_handler.service_create(self.context, expected_service) + mock_kube_cli.service_create.assert_called_once_with( + expected_master_url, expected_service) + self.assertFalse(expected_service.create.called) @patch('magnum.conductor.handlers.kube._object_has_stack') @patch('magnum.conductor.handlers.kube._retrieve_k8s_master_url')