Add service command unit tests

Add service command unit tests.

Change-Id: I09e80692f71f109d0808bb718e7c6c642140fbbf
This commit is contained in:
Steve Noyes 2018-06-07 16:06:38 -04:00
parent 703e0d984b
commit 4e26ee0c40
2 changed files with 85 additions and 0 deletions

View File

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

View File

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