Move unit tests from os_faults/tests/ to os_faults/tests/unit/
Change-Id: I56b25acb032f9a5ecf5d80b0f305b4ca5e9fff1d
This commit is contained in:
Yaroslav Lobankov
361
os_faults/tests/unit/drivers/test_fuel_service.py
Normal file
361
os_faults/tests/unit/drivers/test_fuel_service.py
Normal file
@@ -0,0 +1,361 @@
|
||||
# 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 ddt
|
||||
import mock
|
||||
|
||||
from os_faults.ansible import executor
|
||||
from os_faults.api import error
|
||||
from os_faults.drivers import fuel
|
||||
from os_faults.tests.unit import fakes
|
||||
from os_faults.tests.unit import test
|
||||
|
||||
|
||||
@ddt.ddt
|
||||
class FuelServiceTestCase(test.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
super(FuelServiceTestCase, self).setUp()
|
||||
self.conf = {'address': 'fuel.local', 'username': 'root'}
|
||||
self.fake_ansible_result = fakes.FakeAnsibleResult(
|
||||
payload={'stdout': '[{"ip": "10.0.0.2", "mac": "02", "id": "2"},'
|
||||
' {"ip": "10.0.0.3", "mac": "03", "id": "3"}]'})
|
||||
|
||||
@mock.patch('os_faults.ansible.executor.AnsibleRunner', autospec=True)
|
||||
@ddt.data(('keystone', fuel.KeystoneService),
|
||||
('mysql', fuel.MySQLService),
|
||||
('rabbitmq', fuel.RabbitMQService),
|
||||
('nova-api', fuel.NovaAPIService),
|
||||
('glance-api', fuel.GlanceAPIService),
|
||||
('nova-compute', fuel.NovaComputeService),
|
||||
('nova-scheduler', fuel.NovaSchedulerService),
|
||||
('neutron-openvswitch-agent',
|
||||
fuel.NeutronOpenvswitchAgentService),
|
||||
('neutron-l3-agent', fuel.NeutronL3AgentService),
|
||||
('heat-api', fuel.HeatAPIService),
|
||||
('heat-engine', fuel.HeatEngineService))
|
||||
@ddt.unpack
|
||||
def test_kill(self, service_name, service_cls, mock_ansible_runner):
|
||||
ansible_runner_inst = mock_ansible_runner.return_value
|
||||
ansible_runner_inst.execute.side_effect = [
|
||||
[self.fake_ansible_result],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')]
|
||||
]
|
||||
|
||||
fuel_managment = fuel.FuelManagement(self.conf)
|
||||
|
||||
service = fuel_managment.get_service(service_name)
|
||||
self.assertIsInstance(service, service_cls)
|
||||
|
||||
service.kill()
|
||||
ansible_runner_inst.execute.assert_has_calls([
|
||||
mock.call(['fuel.local'], {'command': 'fuel2 node list -f json'}),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service_cls.GET_NODES_CMD}, []),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service_cls.KILL_CMD}),
|
||||
])
|
||||
|
||||
@mock.patch('os_faults.ansible.executor.AnsibleRunner', autospec=True)
|
||||
@ddt.data(('keystone', fuel.KeystoneService),
|
||||
('mysql', fuel.MySQLService),
|
||||
('rabbitmq', fuel.RabbitMQService),
|
||||
('nova-api', fuel.NovaAPIService),
|
||||
('glance-api', fuel.GlanceAPIService),
|
||||
('nova-compute', fuel.NovaComputeService),
|
||||
('nova-scheduler', fuel.NovaSchedulerService),
|
||||
('neutron-openvswitch-agent',
|
||||
fuel.NeutronOpenvswitchAgentService),
|
||||
('neutron-l3-agent', fuel.NeutronL3AgentService),
|
||||
('heat-api', fuel.HeatAPIService),
|
||||
('heat-engine', fuel.HeatEngineService))
|
||||
@ddt.unpack
|
||||
def test_freeze(self, service_name, service_cls, mock_ansible_runner):
|
||||
ansible_runner_inst = mock_ansible_runner.return_value
|
||||
ansible_runner_inst.execute.side_effect = [
|
||||
[self.fake_ansible_result],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')]
|
||||
]
|
||||
|
||||
fuel_managment = fuel.FuelManagement(self.conf)
|
||||
|
||||
service = fuel_managment.get_service(service_name)
|
||||
self.assertIsInstance(service, service_cls)
|
||||
|
||||
service.freeze()
|
||||
ansible_runner_inst.execute.assert_has_calls([
|
||||
mock.call(['fuel.local'], {'command': 'fuel2 node list -f json'}),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service_cls.GET_NODES_CMD}, []),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service_cls.FREEZE_CMD}),
|
||||
])
|
||||
|
||||
@mock.patch('os_faults.ansible.executor.AnsibleRunner', autospec=True)
|
||||
@ddt.data(('keystone', fuel.KeystoneService),
|
||||
('mysql', fuel.MySQLService),
|
||||
('rabbitmq', fuel.RabbitMQService),
|
||||
('nova-api', fuel.NovaAPIService),
|
||||
('glance-api', fuel.GlanceAPIService),
|
||||
('nova-compute', fuel.NovaComputeService),
|
||||
('nova-scheduler', fuel.NovaSchedulerService),
|
||||
('neutron-openvswitch-agent',
|
||||
fuel.NeutronOpenvswitchAgentService),
|
||||
('neutron-l3-agent', fuel.NeutronL3AgentService),
|
||||
('heat-api', fuel.HeatAPIService),
|
||||
('heat-engine', fuel.HeatEngineService))
|
||||
@ddt.unpack
|
||||
def test_freeze_sec(self, service_name, service_cls, mock_ansible_runner):
|
||||
ansible_runner_inst = mock_ansible_runner.return_value
|
||||
ansible_runner_inst.execute.side_effect = [
|
||||
[self.fake_ansible_result],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')]
|
||||
]
|
||||
|
||||
fuel_managment = fuel.FuelManagement(self.conf)
|
||||
|
||||
service = fuel_managment.get_service(service_name)
|
||||
self.assertIsInstance(service, service_cls)
|
||||
|
||||
delay_sec = 10
|
||||
service.freeze(nodes=None, sec=delay_sec)
|
||||
ansible_runner_inst.execute.assert_has_calls([
|
||||
mock.call(['fuel.local'], {'command': 'fuel2 node list -f json'}),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service_cls.GET_NODES_CMD}, []),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command':
|
||||
service_cls.FREEZE_SEC_CMD.format(delay_sec)}),
|
||||
])
|
||||
|
||||
@mock.patch('os_faults.ansible.executor.AnsibleRunner', autospec=True)
|
||||
@ddt.data(('keystone', fuel.KeystoneService),
|
||||
('mysql', fuel.MySQLService),
|
||||
('rabbitmq', fuel.RabbitMQService),
|
||||
('nova-api', fuel.NovaAPIService),
|
||||
('glance-api', fuel.GlanceAPIService),
|
||||
('nova-compute', fuel.NovaComputeService),
|
||||
('nova-scheduler', fuel.NovaSchedulerService),
|
||||
('neutron-openvswitch-agent',
|
||||
fuel.NeutronOpenvswitchAgentService),
|
||||
('neutron-l3-agent', fuel.NeutronL3AgentService),
|
||||
('heat-api', fuel.HeatAPIService),
|
||||
('heat-engine', fuel.HeatEngineService))
|
||||
@ddt.unpack
|
||||
def test_unfreeze(self, service_name, service_cls, mock_ansible_runner):
|
||||
ansible_runner_inst = mock_ansible_runner.return_value
|
||||
ansible_runner_inst.execute.side_effect = [
|
||||
[self.fake_ansible_result],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')]
|
||||
]
|
||||
|
||||
fuel_managment = fuel.FuelManagement(self.conf)
|
||||
|
||||
service = fuel_managment.get_service(service_name)
|
||||
self.assertIsInstance(service, service_cls)
|
||||
|
||||
service.unfreeze()
|
||||
ansible_runner_inst.execute.assert_has_calls([
|
||||
mock.call(['fuel.local'], {'command': 'fuel2 node list -f json'}),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service_cls.GET_NODES_CMD}, []),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service_cls.UNFREEZE_CMD}),
|
||||
])
|
||||
|
||||
@mock.patch('os_faults.ansible.executor.AnsibleRunner', autospec=True)
|
||||
@ddt.data(('mysql', fuel.MySQLService))
|
||||
@ddt.unpack
|
||||
def test_unplug(self, service_name, service_cls, mock_ansible_runner):
|
||||
ansible_runner_inst = mock_ansible_runner.return_value
|
||||
ansible_runner_inst.execute.side_effect = [
|
||||
[self.fake_ansible_result],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')]
|
||||
]
|
||||
|
||||
fuel_managment = fuel.FuelManagement(self.conf)
|
||||
|
||||
service = fuel_managment.get_service(service_name)
|
||||
self.assertIsInstance(service, service_cls)
|
||||
|
||||
service.unplug()
|
||||
ansible_runner_inst.execute.assert_has_calls([
|
||||
mock.call(['fuel.local'], {'command': 'fuel2 node list -f json'}),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service_cls.GET_NODES_CMD}, []),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command':
|
||||
service_cls.UNPLUG_CMD.format(service_cls.PORT)}),
|
||||
])
|
||||
|
||||
@mock.patch('os_faults.ansible.executor.AnsibleRunner', autospec=True)
|
||||
@ddt.data(('mysql', fuel.MySQLService))
|
||||
@ddt.unpack
|
||||
def test_plug(self, service_name, service_cls, mock_ansible_runner):
|
||||
ansible_runner_inst = mock_ansible_runner.return_value
|
||||
ansible_runner_inst.execute.side_effect = [
|
||||
[self.fake_ansible_result],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')]
|
||||
]
|
||||
|
||||
fuel_managment = fuel.FuelManagement(self.conf)
|
||||
|
||||
service = fuel_managment.get_service(service_name)
|
||||
self.assertIsInstance(service, service_cls)
|
||||
|
||||
service.plug()
|
||||
ansible_runner_inst.execute.assert_has_calls([
|
||||
mock.call(['fuel.local'], {'command': 'fuel2 node list -f json'}),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service_cls.GET_NODES_CMD}, []),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command':
|
||||
service_cls.PLUG_CMD.format(service_cls.PORT)}),
|
||||
])
|
||||
|
||||
@mock.patch('os_faults.ansible.executor.AnsibleRunner', autospec=True)
|
||||
@ddt.data(('keystone', fuel.KeystoneService),
|
||||
('nova-api', fuel.NovaAPIService),
|
||||
('glance-api', fuel.GlanceAPIService),
|
||||
('nova-compute', fuel.NovaComputeService),
|
||||
('nova-scheduler', fuel.NovaSchedulerService),
|
||||
('neutron-openvswitch-agent',
|
||||
fuel.NeutronOpenvswitchAgentService),
|
||||
('neutron-l3-agent', fuel.NeutronL3AgentService),
|
||||
('heat-api', fuel.HeatAPIService),
|
||||
('heat-engine', fuel.HeatEngineService))
|
||||
@ddt.unpack
|
||||
def test_restart(self, service_name, service_cls, mock_ansible_runner):
|
||||
ansible_runner_inst = mock_ansible_runner.return_value
|
||||
ansible_runner_inst.execute.side_effect = [
|
||||
[self.fake_ansible_result],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')]
|
||||
]
|
||||
|
||||
fuel_managment = fuel.FuelManagement(self.conf)
|
||||
|
||||
service = fuel_managment.get_service(service_name)
|
||||
self.assertIsInstance(service, service_cls)
|
||||
|
||||
service.restart()
|
||||
ansible_runner_inst.execute.assert_has_calls([
|
||||
mock.call(['fuel.local'], {'command': 'fuel2 node list -f json'}),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service_cls.GET_NODES_CMD}, []),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service_cls.RESTART_CMD}),
|
||||
])
|
||||
|
||||
@mock.patch('os_faults.ansible.executor.AnsibleRunner', autospec=True)
|
||||
def test_run_task_error(self, mock_ansible_runner):
|
||||
ansible_runner_inst = mock_ansible_runner.return_value
|
||||
ansible_runner_inst.execute.side_effect = [
|
||||
[self.fake_ansible_result],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2'),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2',
|
||||
status=executor.STATUS_FAILED),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3')]
|
||||
]
|
||||
|
||||
fuel_managment = fuel.FuelManagement(self.conf)
|
||||
|
||||
service = fuel_managment.get_service('keystone')
|
||||
exception = self.assertRaises(error.ServiceError, service.restart)
|
||||
self.assertEqual('Task failed on some nodes', str(exception))
|
||||
|
||||
ansible_runner_inst.execute.assert_has_calls([
|
||||
mock.call(['fuel.local'], {'command': 'fuel2 node list -f json'}),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service.GET_NODES_CMD}, []),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service.RESTART_CMD}),
|
||||
])
|
||||
|
||||
@mock.patch('os_faults.ansible.executor.AnsibleRunner', autospec=True)
|
||||
def test_run_node_collection_empty(self, mock_ansible_runner):
|
||||
ansible_runner_inst = mock_ansible_runner.return_value
|
||||
ansible_runner_inst.execute.side_effect = [
|
||||
[self.fake_ansible_result],
|
||||
[fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.2',
|
||||
status=executor.STATUS_FAILED),
|
||||
fakes.FakeAnsibleResult(payload={'stdout': ''},
|
||||
host='10.0.0.3',
|
||||
status=executor.STATUS_FAILED)],
|
||||
]
|
||||
|
||||
fuel_managment = fuel.FuelManagement(self.conf)
|
||||
|
||||
service = fuel_managment.get_service('keystone')
|
||||
exception = self.assertRaises(error.ServiceError, service.restart)
|
||||
self.assertEqual('Node collection is empty', str(exception))
|
||||
|
||||
ansible_runner_inst.execute.assert_has_calls([
|
||||
mock.call(['fuel.local'], {'command': 'fuel2 node list -f json'}),
|
||||
mock.call(['10.0.0.2', '10.0.0.3'],
|
||||
{'command': service.GET_NODES_CMD}, []),
|
||||
])
|
||||
Reference in New Issue
Block a user