performa/os-faults
Code Issues Proposed changes
os-faults/os_faults/tests/unit/drivers/test_fuel_service.py
Anton Studenov 2ddabc2c18 Move services to common module and reuse them in libvirt driver 
* CloudManagement.execute_on_cloud is abstract now.
* CloudManagement.get_service is not abstract now and
  should be the same for all cloud management drivers.
* Moved basic methods of FuelNodeCollection to NodeCollection.
* Moved FuelService to common/service.py and
  renamed to ServiceAsProcess.
* Cleaned up devstack driver:
  - DevStackNode now uses the same methods as FuelNodeCollection.
  - ServiceAsProcess is used for devstack services.
  - Added execute_on_cloud method to allow using common services.
* Added common Host class.

Change-Id: I50334db300098c5aaf410a2b28cdc3c72ebd065e
2016-10-17 11:59:15 +03:00

349 lines
15 KiB
Python
Raw Blame History

# 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", "fqdn": "node-2"},'
' {"ip": "10.0.0.3", "mac": "03", "fqdn": "node-3"}]'
})
@mock.patch('os_faults.ansible.executor.AnsibleRunner', autospec=True)
@ddt.data(*fuel.FuelManagement.SERVICE_NAME_TO_CLASS.items())
@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()
get_nodes_cmd = 'bash -c "ps ax | grep \'{}\'"'.format(
service_cls.GREP)
ansible_runner_inst.execute.assert_has_calls([
mock.call(['fuel.local'], {'command': 'fuel node --json'}),
mock.call(['10.0.0.2', '10.0.0.3'],
{'command': get_nodes_cmd}, []),
mock.call(['10.0.0.2', '10.0.0.3'],
{'kill': {'grep': service_cls.GREP, 'sig': 9}}),
])
@mock.patch('os_faults.ansible.executor.AnsibleRunner', autospec=True)
@ddt.data(*fuel.FuelManagement.SERVICE_NAME_TO_CLASS.items())
@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()
get_nodes_cmd = 'bash -c "ps ax | grep \'{}\'"'.format(
service_cls.GREP)
ansible_runner_inst.execute.assert_has_calls([
mock.call(['fuel.local'], {'command': 'fuel node --json'}),
mock.call(['10.0.0.2', '10.0.0.3'],
{'command': get_nodes_cmd}, []),
mock.call(['10.0.0.2', '10.0.0.3'],
{'kill': {'grep': service_cls.GREP, 'sig': 19}}),
])
@mock.patch('os_faults.ansible.executor.AnsibleRunner', autospec=True)
@ddt.data(*fuel.FuelManagement.SERVICE_NAME_TO_CLASS.items())
@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)
get_nodes_cmd = 'bash -c "ps ax | grep \'{}\'"'.format(
service_cls.GREP)
ansible_runner_inst.execute.assert_has_calls([
mock.call(['fuel.local'], {'command': 'fuel node --json'}),
mock.call(['10.0.0.2', '10.0.0.3'],
{'command': get_nodes_cmd}, []),
mock.call(['10.0.0.2', '10.0.0.3'],
{'freeze': {'grep': service_cls.GREP,
'sec': delay_sec}}),
])
@mock.patch('os_faults.ansible.executor.AnsibleRunner', autospec=True)
@ddt.data(*fuel.FuelManagement.SERVICE_NAME_TO_CLASS.items())
@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()
get_nodes_cmd = 'bash -c "ps ax | grep \'{}\'"'.format(
service_cls.GREP)
ansible_runner_inst.execute.assert_has_calls([
mock.call(['fuel.local'], {'command': 'fuel node --json'}),
mock.call(['10.0.0.2', '10.0.0.3'],
{'command': get_nodes_cmd}, []),
mock.call(['10.0.0.2', '10.0.0.3'],
{'kill': {'grep': service_cls.GREP, 'sig': 18}}),
])
@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()
get_nodes_cmd = 'bash -c "ps ax | grep \'{}\'"'.format(
service_cls.GREP)
ansible_runner_inst.execute.assert_has_calls([
mock.call(['fuel.local'], {'command': 'fuel node --json'}),
mock.call(['10.0.0.2', '10.0.0.3'],
{'command': get_nodes_cmd}, []),
mock.call(['10.0.0.2', '10.0.0.3'],
{'iptables': {'protocol': service_cls.PORT[0],
'port': service_cls.PORT[1],
'action': 'block',
'service': service_cls.SERVICE_NAME}}),
])
@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()
get_nodes_cmd = 'bash -c "ps ax | grep \'{}\'"'.format(
service_cls.GREP)
ansible_runner_inst.execute.assert_has_calls([
mock.call(['fuel.local'], {'command': 'fuel node --json'}),
mock.call(['10.0.0.2', '10.0.0.3'],
{'command': get_nodes_cmd}, []),
mock.call(['10.0.0.2', '10.0.0.3'],
{'iptables': {'protocol': service_cls.PORT[0],
'port': service_cls.PORT[1],
'action': 'unblock',
'service': service_cls.SERVICE_NAME}}),
])
@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()
get_nodes_cmd = 'bash -c "ps ax | grep \'{}\'"'.format(
service.GREP)
ansible_runner_inst.execute.assert_has_calls([
mock.call(['fuel.local'], {'command': 'fuel node --json'}),
mock.call(['10.0.0.2', '10.0.0.3'],
{'command': 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))
get_nodes_cmd = 'bash -c "ps ax | grep \'{}\'"'.format(
service.GREP)
ansible_runner_inst.execute.assert_has_calls([
mock.call(['fuel.local'], {'command': 'fuel node --json'}),
mock.call(['10.0.0.2', '10.0.0.3'],
{'command': 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))
get_nodes_cmd = 'bash -c "ps ax | grep \'{}\'"'.format(
service.GREP)
ansible_runner_inst.execute.assert_has_calls([
mock.call(['fuel.local'], {'command': 'fuel node --json'}),
mock.call(['10.0.0.2', '10.0.0.3'],
{'command': get_nodes_cmd}, []),
])
View Git Blame Copy Permalink