import pytest import mock from mock import Mock, MagicMock import os from utils.create_zones import AvailabilityZone return_list = [] def get_agg_mock(host): agg = Mock() agg.name = host agg.id = host return agg class HyperMock(MagicMock): def list(self): mock_hypervisor = [Mock(service={'host': '10.20.0.4'}), Mock(service={'host': '10.20.0.5'})] return mock_hypervisor class AggMock(MagicMock): def get_details(self, agg_id): print "get_details:{0}".format(agg_id) return Mock(hosts=[]) def create(self, host, agg): print "create:{0}:{1}".format(host, agg) return agg def list(self): return return_list def delete(self, agg_id): print "delete:{0}".format(agg_id) pass def add_host(self, aggregate, host): print "add_host:{0}:{1}".format(aggregate, host) pass def remove_host(self, agg_id, host): print "remove_host:{0}:{1}".format(agg_id, host) pass class NovaMock(MagicMock): hypervisors = HyperMock() aggregates = AggMock() class TestClass: @pytest.mark.parametrize("test_input, expected", [ (['compute1', 'compute2'], ['create:compute1:compute1', 'add_host:compute1:10.20.0.4', 'create:compute2:compute2', 'add_host:compute2:10.20.0.5']), (['compute1', 'compute1'], ['create:compute1:compute1', 'add_host:compute1:10.20.0.4']), ]) @mock.patch('utils.create_zones.client', autospec=True) @mock.patch('utils.create_zones.v2', autospec=True) @mock.patch('utils.create_zones.session') def test_create_zones_success(self, mock_keystone_session, mock_keystone_v2, mock_nova_client, test_input, expected, capfd): nova_obj = NovaMock() mock_nova_client.Client.return_value = nova_obj() k = mock.patch.dict(os.environ, {'OS_AUTH_URL': 'http://172.10.0.5:5000', 'OS_USERNAME': 'admin', 'OS_PASSWORD': 'admin', 'OS_TENANT_NAME': 'admin'}) k.start() azone = AvailabilityZone() azone.create_aggs(test_input) k.stop() resout, reserr = capfd.readouterr() for x in expected: assert x in resout @pytest.mark.parametrize("test_input, expected", [ ([get_agg_mock('10.20.0.4'), get_agg_mock('10.20.0.5')], ['get_details:10.20.0.4', 'delete:10.20.0.4', 'get_details:10.20.0.5', 'delete:10.20.0.5']), ([], []), ]) @mock.patch('utils.create_zones.client', autospec=True) @mock.patch('utils.create_zones.v2', autospec=True) @mock.patch('utils.create_zones.session') def test_clean_all_aggregates(self, mock_keystone_session, mock_keystone_v2, mock_nova_client, test_input, expected, capfd): global return_list return_list = test_input nova_obj = NovaMock() mock_nova_client.Client.return_value = nova_obj() k = mock.patch.dict(os.environ, {'OS_AUTH_URL': 'http://172.10.0.5:5000', 'OS_USERNAME': 'admin', 'OS_PASSWORD': 'admin', 'OS_TENANT_NAME': 'admin'}) k.start() azone = AvailabilityZone() azone.clean_all_aggregates() k.stop() resout, reserr = capfd.readouterr() for x in expected: assert x in resout