Add service test for kube handler

Partial-Bug: #1437977

Change-Id: I932d446e6f372bf1c9ea6607197b623ccda663f8
This commit is contained in:
Kennan 2015-03-30 21:51:52 +08:00
parent 096d958b45
commit de9369e1e2
1 changed files with 18 additions and 1 deletions

View File

@ -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')