Add service command unit tests
Add service command unit tests. Change-Id: I09e80692f71f109d0808bb718e7c6c642140fbbf
This commit is contained in:
parent
703e0d984b
commit
4e26ee0c40
|
@ -15,6 +15,7 @@ import testtools
|
|||
|
||||
from kolla_cli.api.group import Group
|
||||
from kolla_cli.api.host import Host
|
||||
from kolla_cli.api.service import Service
|
||||
from kolla_cli.common.ansible.job import AnsibleJob
|
||||
from kolla_cli import shell
|
||||
|
||||
|
@ -35,3 +36,7 @@ class KollaCliUnitTest(testtools.TestCase):
|
|||
def get_fake_group(self, groupname='group1', servicenames=[],
|
||||
hostnames=[]):
|
||||
return Group(groupname, servicenames, hostnames)
|
||||
|
||||
def get_fake_service(self, servicename='service1', parentnames=[],
|
||||
childnames=[], groupnames=[]):
|
||||
return Service(servicename, parentnames, childnames, groupnames)
|
||||
|
|
|
@ -0,0 +1,80 @@
|
|||
# Copyright (c) 2018 OpenStack Foundation
|
||||
#
|
||||
# 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 kolla_cli.tests.unit.common import KollaCliUnitTest
|
||||
|
||||
|
||||
class TestUnit(KollaCliUnitTest):
|
||||
|
||||
@mock.patch('kolla_cli.api.client.ClientApi.service_get_all')
|
||||
@mock.patch('kolla_cli.shell.KollaCli._is_inventory_present',
|
||||
return_value=True)
|
||||
def test_service_list(self, _, mock_service_get_all):
|
||||
# list all services
|
||||
servicename = 'service1'
|
||||
childname = 'child1'
|
||||
fake_service = self.get_fake_service(servicename,
|
||||
childnames=[childname])
|
||||
mock_service_get_all.return_value = [fake_service]
|
||||
ret = self.run_cli_command('service list')
|
||||
self.assertEqual(ret, 0)
|
||||
mock_service_get_all.assert_called_once_with()
|
||||
|
||||
@mock.patch('kolla_cli.api.client.ClientApi.service_get_all')
|
||||
@mock.patch('kolla_cli.shell.KollaCli._is_inventory_present',
|
||||
return_value=True)
|
||||
def test_service_grouplist(self, _, mock_service_get_all):
|
||||
# list all services with their groups
|
||||
servicename = 'service1'
|
||||
groupname = 'group1'
|
||||
fake_service = self.get_fake_service(servicename,
|
||||
groupnames=[groupname])
|
||||
mock_service_get_all.return_value = [fake_service]
|
||||
ret = self.run_cli_command('service listgroups')
|
||||
self.assertEqual(ret, 0)
|
||||
mock_service_get_all.assert_called_once_with()
|
||||
|
||||
@mock.patch('kolla_cli.api.group.Group.add_service')
|
||||
@mock.patch('kolla_cli.api.client.ClientApi.group_get')
|
||||
@mock.patch('kolla_cli.shell.KollaCli._is_inventory_present',
|
||||
return_value=True)
|
||||
def test_service_addgroup(self, _, mock_group_get,
|
||||
mock_group_add_service):
|
||||
servicename = 'service1'
|
||||
groupname = 'group1'
|
||||
fake_group = self.get_fake_group(groupname)
|
||||
mock_group_get.return_value = [fake_group]
|
||||
ret = self.run_cli_command('service addgroup %s %s'
|
||||
% (servicename, groupname))
|
||||
self.assertEqual(ret, 0)
|
||||
mock_group_get.assert_called_once_with([groupname])
|
||||
mock_group_add_service.assert_called_once_with(servicename)
|
||||
|
||||
@mock.patch('kolla_cli.api.group.Group.remove_service')
|
||||
@mock.patch('kolla_cli.api.client.ClientApi.group_get')
|
||||
@mock.patch('kolla_cli.shell.KollaCli._is_inventory_present',
|
||||
return_value=True)
|
||||
def test_service_removegroup(self, _, mock_group_get,
|
||||
mock_group_remove_service):
|
||||
servicename = 'service1'
|
||||
groupname = 'group1'
|
||||
fake_group = self.get_fake_group(groupname,
|
||||
servicenames=[servicename])
|
||||
mock_group_get.return_value = [fake_group]
|
||||
ret = self.run_cli_command('service removegroup %s %s'
|
||||
% (servicename, groupname))
|
||||
self.assertEqual(ret, 0)
|
||||
mock_group_get.assert_called_once_with([groupname])
|
||||
mock_group_remove_service.assert_called_once_with(servicename)
|
Loading…
Reference in New Issue