551 lines
23 KiB
Python
551 lines
23 KiB
Python
# Copyright 2014 NEC Corporation. All rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from magnum.common import exception
|
|
from magnum.conductor.handlers.common import kube_utils
|
|
from magnum.tests import base
|
|
|
|
import mock
|
|
from mock import patch
|
|
|
|
|
|
class TestKubeUtils(base.BaseTestCase):
|
|
def setUp(self):
|
|
super(TestKubeUtils, self).setUp()
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_create_with_data')
|
|
def test_k8s_create_data(self,
|
|
mock_create_with_data):
|
|
expected_data = 'data'
|
|
api_address = 'api_address'
|
|
mock_resource = mock.MagicMock()
|
|
mock_resource.manifest = expected_data
|
|
mock_resource.manifest_url = None
|
|
|
|
kube_utils._k8s_create(api_address, mock_resource)
|
|
mock_create_with_data.assert_called_once_with(api_address,
|
|
expected_data)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_create_with_path')
|
|
def test_k8s_create_url(self,
|
|
mock_create_with_path):
|
|
expected_url = 'url'
|
|
api_address = 'api_address'
|
|
mock_resource = mock.MagicMock()
|
|
mock_resource.manifest = None
|
|
mock_resource.manifest_url = expected_url
|
|
|
|
kube_utils._k8s_create(api_address, mock_resource)
|
|
mock_create_with_path.assert_called_once_with(api_address,
|
|
expected_url)
|
|
|
|
@patch('magnum.openstack.common.utils.trycmd')
|
|
def test_k8s_create_with_path(self, mock_trycmd):
|
|
expected_api_address = 'api_address'
|
|
expected_pod_file = 'pod_file'
|
|
expected_command = [
|
|
'kubectl', 'create',
|
|
'-s', expected_api_address,
|
|
'-f', expected_pod_file
|
|
]
|
|
|
|
kube_utils._k8s_create_with_path(expected_api_address,
|
|
expected_pod_file)
|
|
mock_trycmd.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_create_with_path')
|
|
@patch('tempfile.NamedTemporaryFile')
|
|
def test_k8s_create_with_data(self,
|
|
mock_named_tempfile,
|
|
mock_k8s_create):
|
|
expected_api_address = 'api_address'
|
|
expected_data = 'resource_data'
|
|
expected_filename = 'resource_file'
|
|
|
|
mock_file = mock.MagicMock()
|
|
mock_file.name = expected_filename
|
|
mock_named_tempfile.return_value.__enter__.return_value = mock_file
|
|
|
|
kube_utils._k8s_create_with_data(expected_api_address,
|
|
expected_data)
|
|
|
|
mock_file.write.assert_called_once_with(expected_data)
|
|
mock_k8s_create.assert_called_once_with(expected_api_address,
|
|
expected_filename)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_update_with_data')
|
|
def test_k8s_update_data(self,
|
|
mock_update_with_data):
|
|
expected_data = 'data'
|
|
api_address = 'api_address'
|
|
mock_resource = mock.MagicMock()
|
|
mock_resource.manifest = expected_data
|
|
mock_resource.manifest_url = None
|
|
|
|
kube_utils._k8s_update(api_address, mock_resource)
|
|
mock_update_with_data.assert_called_once_with(api_address,
|
|
expected_data)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_update_with_path')
|
|
def test_k8s_update_url(self,
|
|
mock_update_with_path):
|
|
expected_url = 'url'
|
|
api_address = 'api_address'
|
|
mock_resource = mock.MagicMock()
|
|
mock_resource.manifest = None
|
|
mock_resource.manifest_url = expected_url
|
|
|
|
kube_utils._k8s_update(api_address, mock_resource)
|
|
mock_update_with_path.assert_called_once_with(api_address,
|
|
expected_url)
|
|
|
|
@patch('magnum.openstack.common.utils.trycmd')
|
|
def test_k8s_update_with_path(self, mock_trycmd):
|
|
expected_api_address = 'api_address'
|
|
expected_pod_file = 'pod_file'
|
|
expected_command = [
|
|
'kubectl', 'update',
|
|
'-s', expected_api_address,
|
|
'-f', expected_pod_file
|
|
]
|
|
|
|
kube_utils._k8s_update_with_path(expected_api_address,
|
|
expected_pod_file)
|
|
mock_trycmd.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_update_with_path')
|
|
@patch('tempfile.NamedTemporaryFile')
|
|
def test_k8s_update_with_data(self,
|
|
mock_named_tempfile,
|
|
mock_k8s_update):
|
|
expected_api_address = 'api_address'
|
|
expected_data = 'resource_data'
|
|
expected_filename = 'resource_file'
|
|
|
|
mock_file = mock.MagicMock()
|
|
mock_file.name = expected_filename
|
|
mock_named_tempfile.return_value.__enter__.return_value = mock_file
|
|
|
|
kube_utils._k8s_update_with_data(expected_api_address,
|
|
expected_data)
|
|
|
|
mock_file.write.assert_called_once_with(expected_data)
|
|
mock_k8s_update.assert_called_once_with(expected_api_address,
|
|
expected_filename)
|
|
|
|
|
|
class KubeClientTestCase(base.TestCase):
|
|
def setUp(self):
|
|
super(KubeClientTestCase, self).setUp()
|
|
self.kube_client = kube_utils.KubeClient()
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_create')
|
|
def test_pod_create(self, mock_k8s_create):
|
|
expected_api_address = 'master-address'
|
|
expected_pod_content = mock.MagicMock(manifest='pod_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_pod_content
|
|
]
|
|
mock_k8s_create.return_value = ("", "")
|
|
|
|
result = self.kube_client.pod_create(expected_api_address,
|
|
expected_pod_content)
|
|
self.assertTrue(result)
|
|
mock_k8s_create.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_create')
|
|
def test_pod_create_with_err(self, mock_k8s_create):
|
|
expected_api_address = 'master-address'
|
|
expected_pod_content = mock.MagicMock(manifest='pod_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_pod_content
|
|
]
|
|
mock_k8s_create.return_value = ("", "create failed")
|
|
|
|
result = self.kube_client.pod_create(expected_api_address,
|
|
expected_pod_content)
|
|
self.assertFalse(result)
|
|
mock_k8s_create.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_create')
|
|
def test_pod_create_failure_exception(self, mock_k8s_create):
|
|
expected_api_address = 'master-address'
|
|
expected_pod_content = mock.MagicMock(manifest='pod_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_pod_content
|
|
]
|
|
mock_k8s_create.side_effect = Exception()
|
|
result = self.kube_client.pod_create(expected_api_address,
|
|
expected_pod_content)
|
|
self.assertFalse(result)
|
|
mock_k8s_create.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_update')
|
|
def test_pod_update(self, mock_k8s_update):
|
|
expected_api_address = 'master-address'
|
|
expected_pod_content = mock.MagicMock(manifest='pod_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_pod_content
|
|
]
|
|
mock_k8s_update.return_value = ("", "")
|
|
|
|
result = self.kube_client.pod_update(expected_api_address,
|
|
expected_pod_content)
|
|
self.assertTrue(result)
|
|
mock_k8s_update.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_update')
|
|
def test_pod_update_with_err(self, mock_k8s_update):
|
|
expected_api_address = 'master-address'
|
|
expected_pod_content = mock.MagicMock(manifest='pod_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_pod_content
|
|
]
|
|
mock_k8s_update.return_value = ("", "create failed")
|
|
|
|
result = self.kube_client.pod_update(expected_api_address,
|
|
expected_pod_content)
|
|
self.assertFalse(result)
|
|
mock_k8s_update.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_update')
|
|
def test_pod_update_failure_exception(self, mock_k8s_update):
|
|
expected_api_address = 'master-address'
|
|
expected_pod_content = mock.MagicMock(manifest='pod_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_pod_content
|
|
]
|
|
mock_k8s_update.side_effect = Exception()
|
|
result = self.kube_client.pod_update(expected_api_address,
|
|
expected_pod_content)
|
|
self.assertFalse(result)
|
|
mock_k8s_update.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.openstack.common.utils.trycmd')
|
|
def test_pod_delete(self, mock_trycmd):
|
|
expected_api_address = 'master-address'
|
|
expected_pod_name = 'test-pod'
|
|
expected_command = [
|
|
'kubectl', 'delete', 'pod', expected_pod_name,
|
|
'-s', expected_api_address
|
|
]
|
|
mock_trycmd.return_value = ("", "")
|
|
|
|
result = self.kube_client.pod_delete(expected_api_address,
|
|
expected_pod_name)
|
|
self.assertTrue(result)
|
|
mock_trycmd.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.openstack.common.utils.trycmd')
|
|
def test_pod_delete_failure_err_not_empty(self, mock_trycmd):
|
|
expected_api_address = 'master-address'
|
|
expected_pod_name = 'test-pod'
|
|
expected_command = [
|
|
'kubectl', 'delete', 'pod', expected_pod_name,
|
|
'-s', expected_api_address
|
|
]
|
|
mock_trycmd.return_value = ("", "error")
|
|
|
|
result = self.kube_client.pod_delete(expected_api_address,
|
|
expected_pod_name)
|
|
self.assertFalse(result)
|
|
mock_trycmd.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.openstack.common.utils.trycmd')
|
|
def test_pod_delete_failure_exception(self, mock_trycmd):
|
|
expected_api_address = 'master-address'
|
|
expected_pod_name = 'test-pod'
|
|
expected_command = [
|
|
'kubectl', 'delete', 'pod', expected_pod_name,
|
|
'-s', expected_api_address
|
|
]
|
|
mock_trycmd.side_effect = Exception()
|
|
|
|
result = self.kube_client.pod_delete(expected_api_address,
|
|
expected_pod_name)
|
|
self.assertFalse(result)
|
|
mock_trycmd.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.openstack.common.utils.trycmd')
|
|
def test_pod_delete_not_found_old(self, mock_trycmd):
|
|
expected_api_address = 'master-address'
|
|
expected_pod_name = 'test-pod'
|
|
expected_command = [
|
|
'kubectl', 'delete', 'pod', expected_pod_name,
|
|
'-s', expected_api_address
|
|
]
|
|
mock_trycmd.return_value = ("", 'pod "test-pod" not found')
|
|
|
|
self.assertRaises(exception.PodNotFound, self.kube_client.pod_delete,
|
|
expected_api_address, expected_pod_name)
|
|
|
|
mock_trycmd.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.openstack.common.utils.trycmd')
|
|
def test_pod_delete_not_found_new(self, mock_trycmd):
|
|
expected_api_address = 'master-address'
|
|
expected_pod_name = 'test-pod'
|
|
expected_command = [
|
|
'kubectl', 'delete', 'pod', expected_pod_name,
|
|
'-s', expected_api_address
|
|
]
|
|
mock_trycmd.return_value = ("", 'pods "test-pod" not found')
|
|
|
|
self.assertRaises(exception.PodNotFound, self.kube_client.pod_delete,
|
|
expected_api_address, expected_pod_name)
|
|
|
|
mock_trycmd.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.openstack.common.utils.trycmd')
|
|
def test_service_delete(self, mock_trycmd):
|
|
expected_api_address = 'master-address'
|
|
expected_service_name = 'test-service'
|
|
expected_command = [
|
|
'kubectl', 'delete', 'service', expected_service_name,
|
|
'-s', expected_api_address
|
|
]
|
|
mock_trycmd.return_value = ("", "")
|
|
|
|
result = self.kube_client.service_delete(expected_api_address,
|
|
expected_service_name)
|
|
self.assertTrue(result)
|
|
mock_trycmd.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.openstack.common.utils.trycmd')
|
|
def test_service_delete_failure_exception(self, mock_trycmd):
|
|
expected_api_address = 'master-address'
|
|
expected_service_name = 'test-service'
|
|
expected_command = [
|
|
'kubectl', 'delete', 'service', expected_service_name,
|
|
'-s', expected_api_address
|
|
]
|
|
mock_trycmd.side_effect = Exception()
|
|
|
|
result = self.kube_client.service_delete(expected_api_address,
|
|
expected_service_name)
|
|
self.assertFalse(result)
|
|
mock_trycmd.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_update')
|
|
def test_service_update(self, mock_k8s_update):
|
|
expected_api_address = 'master-address'
|
|
expected_service_content = mock.MagicMock(manifest='service_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_service_content
|
|
]
|
|
mock_k8s_update.return_value = ("", "")
|
|
|
|
result = self.kube_client.service_update(expected_api_address,
|
|
expected_service_content)
|
|
self.assertTrue(result)
|
|
mock_k8s_update.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_update')
|
|
def test_service_update_with_err(self, mock_k8s_update):
|
|
expected_api_address = 'master-address'
|
|
expected_service_content = mock.MagicMock(manifest='service_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_service_content
|
|
]
|
|
mock_k8s_update.return_value = ("", "create failed")
|
|
|
|
result = self.kube_client.service_update(expected_api_address,
|
|
expected_service_content)
|
|
self.assertFalse(result)
|
|
mock_k8s_update.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_update')
|
|
def test_service_update_failure_exception(self, mock_k8s_update):
|
|
expected_api_address = 'master-address'
|
|
expected_service_content = mock.MagicMock(manifest='service_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_service_content
|
|
]
|
|
mock_k8s_update.side_effect = Exception()
|
|
result = self.kube_client.service_update(expected_api_address,
|
|
expected_service_content)
|
|
self.assertFalse(result)
|
|
mock_k8s_update.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_create')
|
|
def test_service_create(self, mock_k8s_create):
|
|
expected_api_address = 'master-address'
|
|
expected_service_content = mock.MagicMock(manifest='service_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_service_content
|
|
]
|
|
mock_k8s_create.return_value = ("", "")
|
|
|
|
result = self.kube_client.service_create(expected_api_address,
|
|
expected_service_content)
|
|
self.assertTrue(result)
|
|
mock_k8s_create.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_create')
|
|
def test_service_create_with_err(self, mock_k8s_create):
|
|
expected_api_address = 'master-address'
|
|
expected_service_content = mock.MagicMock(manifest='service_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_service_content
|
|
]
|
|
mock_k8s_create.return_value = ("", "create failed")
|
|
|
|
result = self.kube_client.service_create(expected_api_address,
|
|
expected_service_content)
|
|
self.assertFalse(result)
|
|
mock_k8s_create.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_create')
|
|
def test_service_create_failure_exception(self, mock_k8s_create):
|
|
expected_api_address = 'master-address'
|
|
expected_service_content = mock.MagicMock(manifest='service_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_service_content
|
|
]
|
|
mock_k8s_create.side_effect = Exception()
|
|
result = self.kube_client.service_create(expected_api_address,
|
|
expected_service_content)
|
|
self.assertFalse(result)
|
|
mock_k8s_create.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_update')
|
|
def test_rc_update(self, mock_k8s_update):
|
|
expected_api_address = 'master-address'
|
|
expected_rc_content = mock.MagicMock(manifest='rc_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_rc_content
|
|
]
|
|
mock_k8s_update.return_value = ("", "")
|
|
|
|
result = self.kube_client.rc_update(expected_api_address,
|
|
expected_rc_content)
|
|
self.assertTrue(result)
|
|
mock_k8s_update.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_update')
|
|
def test_rc_update_with_err(self, mock_k8s_update):
|
|
expected_api_address = 'master-address'
|
|
expected_rc_content = mock.MagicMock(manifest='rc_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_rc_content
|
|
]
|
|
mock_k8s_update.return_value = ("", "update failed")
|
|
|
|
result = self.kube_client.rc_update(expected_api_address,
|
|
expected_rc_content)
|
|
self.assertFalse(result)
|
|
mock_k8s_update.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_update')
|
|
def test_rc_update_failure_exception(self, mock_k8s_update):
|
|
expected_api_address = 'master-address'
|
|
expected_rc_content = mock.MagicMock(manifest='rc_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_rc_content
|
|
]
|
|
mock_k8s_update.side_effect = Exception()
|
|
result = self.kube_client.rc_update(expected_api_address,
|
|
expected_rc_content)
|
|
self.assertFalse(result)
|
|
mock_k8s_update.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_create')
|
|
def test_rc_create(self, mock_k8s_create):
|
|
expected_api_address = 'master-address'
|
|
expected_rc_content = mock.MagicMock(manifest='rc_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_rc_content
|
|
]
|
|
mock_k8s_create.return_value = ("", "")
|
|
|
|
result = self.kube_client.rc_create(expected_api_address,
|
|
expected_rc_content)
|
|
self.assertTrue(result)
|
|
mock_k8s_create.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_create')
|
|
def test_rc_create_with_err(self, mock_k8s_create):
|
|
expected_api_address = 'master-address'
|
|
expected_rc_content = mock.MagicMock(manifest='rc_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_rc_content
|
|
]
|
|
mock_k8s_create.return_value = ("", "create failed")
|
|
|
|
result = self.kube_client.rc_create(expected_api_address,
|
|
expected_rc_content)
|
|
self.assertFalse(result)
|
|
mock_k8s_create.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.conductor.handlers.common.kube_utils._k8s_create')
|
|
def test_rc_create_failure_exception(self, mock_k8s_create):
|
|
expected_api_address = 'master-address'
|
|
expected_rc_content = mock.MagicMock(manifest='rc_content')
|
|
expected_command = [
|
|
expected_api_address,
|
|
expected_rc_content
|
|
]
|
|
mock_k8s_create.side_effect = Exception()
|
|
result = self.kube_client.rc_create(expected_api_address,
|
|
expected_rc_content)
|
|
self.assertFalse(result)
|
|
mock_k8s_create.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.openstack.common.utils.trycmd')
|
|
def test_rc_delete(self, mock_trycmd):
|
|
expected_api_address = 'master-address'
|
|
expected_rc_name = 'test-rc'
|
|
expected_command = [
|
|
'kubectl', 'delete', 'rc', expected_rc_name,
|
|
'-s', expected_api_address
|
|
]
|
|
mock_trycmd.return_value = ("", "")
|
|
|
|
result = self.kube_client.rc_delete(expected_api_address,
|
|
expected_rc_name)
|
|
self.assertTrue(result)
|
|
mock_trycmd.assert_called_once_with(*expected_command)
|
|
|
|
@patch('magnum.openstack.common.utils.trycmd')
|
|
def test_rc_delete_failure_exception(self, mock_trycmd):
|
|
expected_api_address = 'master-address'
|
|
expected_rc_name = 'test-rc'
|
|
expected_command = [
|
|
'kubectl', 'delete', 'rc', expected_rc_name,
|
|
'-s', expected_api_address
|
|
]
|
|
mock_trycmd.side_effect = Exception()
|
|
|
|
result = self.kube_client.rc_delete(expected_api_address,
|
|
expected_rc_name)
|
|
self.assertFalse(result)
|
|
mock_trycmd.assert_called_once_with(*expected_command)
|