429 lines
17 KiB
Python
429 lines
17 KiB
Python
# Copyright 2017 The Armada Authors.
|
|
#
|
|
# 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.
|
|
|
|
import mock
|
|
from mock import MagicMock
|
|
|
|
from armada.exceptions import tiller_exceptions as ex
|
|
from armada.handlers import tiller
|
|
from armada.tests.unit import base
|
|
|
|
|
|
class TillerTestCase(base.ArmadaTestCase):
|
|
|
|
@mock.patch.object(tiller.Tiller, '_get_tiller_ip')
|
|
@mock.patch('armada.handlers.tiller.K8s')
|
|
@mock.patch('armada.handlers.tiller.grpc')
|
|
@mock.patch('armada.handlers.tiller.Config')
|
|
@mock.patch('armada.handlers.tiller.InstallReleaseRequest')
|
|
@mock.patch('armada.handlers.tiller.ReleaseServiceStub')
|
|
def test_install_release(self, mock_stub, mock_install_request,
|
|
mock_config, mock_grpc, mock_k8s, mock_ip):
|
|
# instantiate Tiller object
|
|
mock_grpc.insecure_channel.return_value = None
|
|
mock_ip.return_value = '0.0.0.0'
|
|
tiller_obj = tiller.Tiller()
|
|
assert tiller_obj._get_tiller_ip() == '0.0.0.0'
|
|
|
|
# set params
|
|
chart = mock.Mock()
|
|
name = None
|
|
namespace = None
|
|
initial_values = None
|
|
updated_values = mock_config(raw=initial_values)
|
|
wait = False
|
|
timeout = 3600
|
|
|
|
tiller_obj.install_release(
|
|
chart, name, namespace, values=initial_values,
|
|
wait=wait, timeout=timeout)
|
|
|
|
mock_stub.assert_called_with(tiller_obj.channel)
|
|
release_request = mock_install_request(
|
|
chart=chart,
|
|
values=updated_values,
|
|
release=name,
|
|
namespace=namespace,
|
|
wait=wait,
|
|
timeout=timeout
|
|
)
|
|
(mock_stub(tiller_obj.channel).InstallRelease
|
|
.assert_called_with(release_request,
|
|
timeout + 60,
|
|
metadata=tiller_obj.metadata))
|
|
|
|
@mock.patch('armada.handlers.tiller.K8s', autospec=True)
|
|
@mock.patch.object(tiller.Tiller, '_get_tiller_ip', autospec=True)
|
|
@mock.patch.object(tiller.Tiller, '_get_tiller_port', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.grpc', autospec=True)
|
|
def test_get_channel(self, mock_grpc, mock_port, mock_ip, _):
|
|
mock_port.return_value = mock.sentinel.port
|
|
mock_ip.return_value = mock.sentinel.ip
|
|
|
|
# instantiate Tiller object
|
|
mock_grpc.insecure_channel.return_value = 'connected'
|
|
tiller_obj = tiller.Tiller()
|
|
|
|
self.assertIsNotNone(tiller_obj.channel)
|
|
self.assertEqual('connected', tiller_obj.channel)
|
|
|
|
mock_grpc.insecure_channel.assert_called_once_with(
|
|
'%s:%s' % (str(mock.sentinel.ip), str(mock.sentinel.port)),
|
|
options=[('grpc.max_send_message_length',
|
|
tiller.MAX_MESSAGE_LENGTH),
|
|
('grpc.max_receive_message_length',
|
|
tiller.MAX_MESSAGE_LENGTH)]
|
|
)
|
|
|
|
@mock.patch('armada.handlers.tiller.K8s', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.grpc', autospec=True)
|
|
def test_get_tiller_ip_with_host_provided(self, mock_grpc, _):
|
|
tiller_obj = tiller.Tiller('1.1.1.1')
|
|
self.assertIsNotNone(tiller_obj._get_tiller_ip())
|
|
self.assertEqual('1.1.1.1', tiller_obj._get_tiller_ip())
|
|
|
|
@mock.patch.object(tiller.Tiller, '_get_tiller_pod', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.K8s', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.grpc', autospec=True)
|
|
def test_get_tiller_ip_with_mocked_pod(self, mock_grpc, mock_k8s,
|
|
mock_pod):
|
|
status = mock.Mock(pod_ip='1.1.1.1')
|
|
mock_pod.return_value.status = status
|
|
tiller_obj = tiller.Tiller()
|
|
self.assertEqual('1.1.1.1', tiller_obj._get_tiller_ip())
|
|
|
|
@mock.patch.object(tiller.Tiller, '_get_tiller_ip', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.K8s', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.grpc', autospec=True)
|
|
def test_get_tiller_pod_throws_exception(self, mock_grpc, mock_k8s,
|
|
mock_ip):
|
|
|
|
mock_k8s.get_namespace_pod.return_value.items = []
|
|
tiller_obj = tiller.Tiller()
|
|
mock_grpc.insecure_channel.side_effect = ex.ChannelException()
|
|
self.assertRaises(ex.TillerPodNotRunningException,
|
|
tiller_obj._get_tiller_pod)
|
|
|
|
@mock.patch.object(tiller.Tiller, '_get_tiller_ip', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.K8s', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.grpc', autospec=True)
|
|
def test_get_tiller_port(self, mock_grpc, _, mock_ip):
|
|
# instantiate Tiller object
|
|
tiller_obj = tiller.Tiller(None, '8080', None)
|
|
self.assertEqual('8080', tiller_obj._get_tiller_port())
|
|
|
|
@mock.patch.object(tiller.Tiller, '_get_tiller_ip', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.K8s', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.grpc', autospec=True)
|
|
def test_get_tiller_namespace(self, mock_grpc, _, mock_ip):
|
|
# verifies namespace set via instantiation
|
|
tiller_obj = tiller.Tiller(None, None, 'test_namespace2')
|
|
self.assertEqual('test_namespace2',
|
|
tiller_obj._get_tiller_namespace())
|
|
|
|
@mock.patch.object(tiller.Tiller, '_get_tiller_ip', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.K8s', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.grpc', autospec=True)
|
|
def test_get_tiller_status_with_ip_provided(self, mock_grpc, _, mock_ip):
|
|
# instantiate Tiller object
|
|
tiller_obj = tiller.Tiller(None, '8080', None)
|
|
self.assertTrue(tiller_obj.tiller_status())
|
|
|
|
@mock.patch.object(tiller.Tiller, '_get_tiller_ip', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.K8s', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.grpc', autospec=True)
|
|
def test_get_tiller_status_no_ip(self, mock_grpc, _, mock_ip):
|
|
mock_ip.return_value = ''
|
|
# instantiate Tiller object
|
|
tiller_obj = tiller.Tiller()
|
|
self.assertFalse(tiller_obj.tiller_status())
|
|
|
|
@mock.patch.object(tiller.Tiller, '_get_tiller_ip', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.K8s', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.grpc', autospec=True)
|
|
def test_list_releases_empty(self, mock_grpc, _, mock_ip):
|
|
# instantiate Tiller object
|
|
tiller_obj = tiller.Tiller()
|
|
self.assertEqual([], tiller_obj.list_releases())
|
|
|
|
@mock.patch.object(tiller.Tiller, '_get_tiller_ip', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.K8s', autospec=True)
|
|
@mock.patch('armada.handlers.tiller.grpc', autospec=True)
|
|
def test_list_charts_empty(self, mock_grpc, _, mock_ip):
|
|
# instantiate Tiller object
|
|
tiller_obj = tiller.Tiller()
|
|
self.assertEqual([], tiller_obj.list_charts())
|
|
|
|
@mock.patch('armada.handlers.tiller.K8s')
|
|
@mock.patch('armada.handlers.tiller.grpc')
|
|
@mock.patch.object(tiller, 'ListReleasesRequest')
|
|
@mock.patch.object(tiller, 'ReleaseServiceStub')
|
|
def test_list_releases(self, mock_release_service_stub,
|
|
mock_list_releases_request, mock_grpc, _):
|
|
mock_release_service_stub.return_value.ListReleases. \
|
|
return_value = [mock.Mock(releases=['foo', 'bar'])]
|
|
|
|
tiller_obj = tiller.Tiller('host', '8080', None)
|
|
self.assertEqual(['foo', 'bar'], tiller_obj.list_releases())
|
|
|
|
mock_release_service_stub.assert_called_once_with(
|
|
tiller_obj.channel)
|
|
list_releases_stub = mock_release_service_stub.return_value. \
|
|
ListReleases
|
|
list_releases_stub.assert_called_once_with(
|
|
mock_list_releases_request.return_value, tiller_obj.timeout,
|
|
metadata=tiller_obj.metadata)
|
|
|
|
mock_list_releases_request.assert_called_once_with(
|
|
limit=tiller.RELEASE_LIMIT,
|
|
status_codes=[tiller.const.STATUS_DEPLOYED,
|
|
tiller.const.STATUS_FAILED],
|
|
sort_by='LAST_RELEASED',
|
|
sort_order='DESC')
|
|
|
|
@mock.patch('armada.handlers.tiller.K8s')
|
|
@mock.patch('armada.handlers.tiller.grpc')
|
|
@mock.patch.object(tiller, 'GetReleaseContentRequest')
|
|
@mock.patch.object(tiller, 'ReleaseServiceStub')
|
|
def test_get_release_content(self, mock_release_service_stub,
|
|
mock_release_content_request,
|
|
mock_grpc, _):
|
|
mock_release_service_stub.return_value.GetReleaseContent\
|
|
.return_value = {}
|
|
|
|
tiller_obj = tiller.Tiller('host', '8080', None)
|
|
|
|
self.assertEqual({}, tiller_obj.get_release_content('release'))
|
|
get_release_content_stub = mock_release_service_stub. \
|
|
return_value.GetReleaseContent
|
|
get_release_content_stub.assert_called_once_with(
|
|
mock_release_content_request.return_value, tiller_obj.timeout,
|
|
metadata=tiller_obj.metadata)
|
|
|
|
@mock.patch('armada.handlers.tiller.K8s')
|
|
@mock.patch('armada.handlers.tiller.grpc')
|
|
@mock.patch.object(tiller, 'GetVersionRequest')
|
|
@mock.patch.object(tiller, 'ReleaseServiceStub')
|
|
def test_tiller_version(self, mock_release_service_stub,
|
|
mock_version_request,
|
|
mock_grpc, _):
|
|
|
|
mock_version = mock.Mock()
|
|
mock_version.Version.sem_ver = mock.sentinel.sem_ver
|
|
mock_release_service_stub.return_value.GetVersion\
|
|
.return_value = mock_version
|
|
|
|
tiller_obj = tiller.Tiller('host', '8080', None)
|
|
|
|
self.assertEqual(mock.sentinel.sem_ver, tiller_obj.tiller_version())
|
|
|
|
mock_release_service_stub.assert_called_once_with(
|
|
tiller_obj.channel)
|
|
|
|
get_version_stub = mock_release_service_stub.return_value.GetVersion
|
|
get_version_stub.assert_called_once_with(
|
|
mock_version_request.return_value, tiller_obj.timeout,
|
|
metadata=tiller_obj.metadata)
|
|
|
|
@mock.patch('armada.handlers.tiller.K8s')
|
|
@mock.patch('armada.handlers.tiller.grpc')
|
|
@mock.patch.object(tiller, 'GetVersionRequest')
|
|
@mock.patch.object(tiller, 'GetReleaseStatusRequest')
|
|
@mock.patch.object(tiller, 'ReleaseServiceStub')
|
|
def test_get_release_status(self, mock_release_service_stub,
|
|
mock_rel_status_request, mock_version_request,
|
|
mock_grpc, _):
|
|
mock_release_service_stub.return_value.GetReleaseStatus. \
|
|
return_value = {}
|
|
|
|
tiller_obj = tiller.Tiller('host', '8080', None)
|
|
self.assertEqual({}, tiller_obj.get_release_status('release'))
|
|
|
|
mock_release_service_stub.assert_called_once_with(
|
|
tiller_obj.channel)
|
|
get_release_status_stub = mock_release_service_stub.return_value. \
|
|
GetReleaseStatus
|
|
get_release_status_stub.assert_called_once_with(
|
|
mock_rel_status_request.return_value, tiller_obj.timeout,
|
|
metadata=tiller_obj.metadata)
|
|
|
|
@mock.patch('armada.handlers.tiller.K8s')
|
|
@mock.patch('armada.handlers.tiller.grpc')
|
|
@mock.patch.object(tiller, 'UninstallReleaseRequest')
|
|
@mock.patch.object(tiller, 'ReleaseServiceStub')
|
|
def test_uninstall_release(self, mock_release_service_stub,
|
|
mock_uninstall_release_request,
|
|
mock_grpc, _):
|
|
mock_release_service_stub.return_value.UninstallRelease\
|
|
.return_value = {}
|
|
|
|
tiller_obj = tiller.Tiller('host', '8080', None)
|
|
|
|
self.assertEqual({}, tiller_obj.uninstall_release('release'))
|
|
|
|
mock_release_service_stub.assert_called_once_with(
|
|
tiller_obj.channel)
|
|
uninstall_release_stub = mock_release_service_stub.return_value. \
|
|
UninstallRelease
|
|
|
|
uninstall_release_stub.assert_called_once_with(
|
|
mock_uninstall_release_request.return_value, tiller_obj.timeout,
|
|
metadata=tiller_obj.metadata)
|
|
|
|
@mock.patch('armada.handlers.tiller.K8s')
|
|
@mock.patch('armada.handlers.tiller.grpc')
|
|
@mock.patch.object(tiller, 'RollbackReleaseRequest')
|
|
@mock.patch.object(tiller, 'ReleaseServiceStub')
|
|
def test_rollback_release(self, mock_release_service_stub,
|
|
mock_rollback_release_request, _, __):
|
|
mock_release_service_stub.return_value.RollbackRelease\
|
|
.return_value = {}
|
|
|
|
dry_run = True
|
|
|
|
tiller_obj = tiller.Tiller('host', '8080', None, dry_run=dry_run)
|
|
|
|
release = 'release'
|
|
version = 0
|
|
wait = True
|
|
timeout = 123
|
|
recreate_pods = True
|
|
force = True
|
|
|
|
self.assertIsNone(tiller_obj.rollback_release(
|
|
release, version, wait=wait, timeout=timeout, force=force,
|
|
recreate_pods=recreate_pods))
|
|
|
|
mock_rollback_release_request.assert_called_once_with(
|
|
name=release,
|
|
version=version,
|
|
dry_run=dry_run,
|
|
wait=wait,
|
|
timeout=timeout,
|
|
force=force,
|
|
recreate=recreate_pods)
|
|
|
|
mock_release_service_stub.assert_called_once_with(
|
|
tiller_obj.channel)
|
|
rollback_release_stub = mock_release_service_stub.return_value. \
|
|
RollbackRelease
|
|
|
|
rollback_release_stub.assert_called_once_with(
|
|
mock_rollback_release_request.return_value, timeout +
|
|
tiller.GRPC_EPSILON,
|
|
metadata=tiller_obj.metadata)
|
|
|
|
@mock.patch('armada.handlers.tiller.K8s')
|
|
@mock.patch('armada.handlers.tiller.grpc')
|
|
@mock.patch('armada.handlers.tiller.Config')
|
|
@mock.patch.object(tiller, 'UpdateReleaseRequest')
|
|
@mock.patch.object(tiller, 'ReleaseServiceStub')
|
|
def test_update_release(self, mock_release_service_stub,
|
|
mock_update_release_request, mock_config,
|
|
_, __):
|
|
release = 'release'
|
|
chart = {}
|
|
namespace = 'namespace'
|
|
code = 0
|
|
status = 'DEPLOYED'
|
|
description = 'desc'
|
|
version = 2
|
|
values = mock_config(raw=None)
|
|
mock_release_service_stub.return_value.UpdateRelease.return_value =\
|
|
AttrDict(**{
|
|
'release': AttrDict(**{
|
|
'name': release,
|
|
'namespace': namespace,
|
|
'info': AttrDict(**{
|
|
'status': AttrDict(**{
|
|
'Code': AttrDict(**{
|
|
'Name': lambda c:
|
|
status if c == code else None
|
|
}),
|
|
'code': code
|
|
}),
|
|
'Description': description
|
|
}),
|
|
'version': version
|
|
})
|
|
})
|
|
|
|
tiller_obj = tiller.Tiller('host', '8080', None, dry_run=False)
|
|
|
|
# TODO: Test these methods as well, either by unmocking, or adding
|
|
# separate tests for them.
|
|
tiller_obj._pre_update_actions = MagicMock()
|
|
tiller_obj._post_update_actions = MagicMock()
|
|
|
|
pre_actions = {}
|
|
post_actions = {}
|
|
disable_hooks = False
|
|
wait = True
|
|
timeout = 123
|
|
force = True
|
|
recreate_pods = True
|
|
|
|
result = tiller_obj.update_release(
|
|
chart, release, namespace,
|
|
pre_actions=pre_actions,
|
|
post_actions=post_actions,
|
|
disable_hooks=disable_hooks,
|
|
values=values,
|
|
wait=wait,
|
|
timeout=timeout,
|
|
force=force,
|
|
recreate_pods=recreate_pods)
|
|
|
|
tiller_obj._pre_update_actions.assert_called_once_with(
|
|
pre_actions, release, namespace, chart, disable_hooks, values,
|
|
timeout)
|
|
tiller_obj._post_update_actions.assert_called_once_with(
|
|
post_actions, namespace)
|
|
|
|
mock_update_release_request.assert_called_once_with(
|
|
chart=chart,
|
|
name=release,
|
|
dry_run=tiller_obj.dry_run,
|
|
disable_hooks=False,
|
|
values=values,
|
|
wait=wait,
|
|
timeout=timeout,
|
|
force=force,
|
|
recreate=recreate_pods)
|
|
|
|
mock_release_service_stub.assert_called_once_with(
|
|
tiller_obj.channel)
|
|
update_release_stub = mock_release_service_stub.return_value. \
|
|
UpdateRelease
|
|
|
|
update_release_stub.assert_called_once_with(
|
|
mock_update_release_request.return_value, timeout +
|
|
tiller.GRPC_EPSILON,
|
|
metadata=tiller_obj.metadata)
|
|
|
|
expected_result = tiller.TillerResult(
|
|
release,
|
|
namespace,
|
|
status,
|
|
description,
|
|
version)
|
|
|
|
self.assertEqual(expected_result, result)
|
|
|
|
|
|
class AttrDict(dict):
|
|
def __init__(self, *args, **kwargs):
|
|
super(AttrDict, self).__init__(*args, **kwargs)
|
|
self.__dict__ = self
|