def test_instance_create_cancellation_called(self): test_name = 'test' CloudshellDriverHelper.get_uuid = Mock(return_value='1234') test_uniq_name = 'test-1234' mock_client2 = Mock() test_nova_instance_service.novaclient.Client = Mock(return_value=mock_client2) # mock_client.Client = Mock(return_vaule=mock_client2) mock_image = Mock() mock_flavor = Mock() mock_client2.images.find = Mock(return_value=mock_image) mock_client2.flavors.find = Mock(return_value=mock_flavor) mock_cp_resource_model = Mock() mock_cp_resource_model.qs_mgmt_os_net_uuid = '1234' mock_cancellation_context = Mock() mock_client2.servers = Mock() mocked_inst = Mock() mock_client2.servers.create = Mock(return_value=mocked_inst) mock_qnet_dict = {'net-id': mock_cp_resource_model.qs_mgmt_os_net_uuid} self.instance_service.instance_waiter = Mock() self.instance_service.instance_waiter.wait = Mock(side_effect=CommandCancellationException) with self.assertRaises(CommandCancellationException): result = self.instance_service.create_instance(openstack_session=self.openstack_session, name=test_name, reservation=Mock(), cp_resource_model=mock_cp_resource_model, deploy_req_model=Mock(), cancellation_context=mock_cancellation_context, logger=self.mock_logger) mock_client2.servers.delete.assert_called_once_with(mocked_inst)
def test_validate_network_attributes(self): mock_cp_resource_model = Mock() mock_cp_resource_model.qs_mgmt_os_net_uuid = 'test-mgmt-uuid' mock_cp_resource_model.floating_ip_subnet_uuid = 'test-subnet-uuid' mock_cp_resource_model.vlan_type = 'vlan' mock_cp_resource_model.reserved_networks = 'reserved-networks' mock_cp_resource_model.provider_network_interface = 'test-if' mock_os_session = Mock() mock_net_client = Mock() test_cp_validator.neutron_client = Mock() test_cp_validator.neutron_client.Client = Mock(return_value=mock_net_client) self.cp_validator.validate_mgmt_network = Mock(return_value=True) self.cp_validator.validate_floating_ip_subnet = Mock(return_value=True) self.cp_validator.validate_vlan_type = Mock(return_value=True) self.cp_validator.validate_reserved_networks = Mock(return_value=True) self.cp_validator.validate_network_attributes(openstack_session=mock_os_session, cp_resource_model=mock_cp_resource_model, logger=self.mock_logger) self.cp_validator.validate_floating_ip_subnet.assert_called_with(mock_net_client, 'test-subnet-uuid', self.mock_logger) self.cp_validator.validate_mgmt_network.assert_called_with(mock_net_client, 'test-mgmt-uuid', self.mock_logger) self.cp_validator.validate_vlan_type.assert_called_with(net_client=mock_net_client, vlan_type='vlan', provider_net_interface='test-if', logger=self.mock_logger) self.cp_validator.validate_reserved_networks.assert_called_with('reserved-networks', self.mock_logger)
def test_instance_create_success(self): test_name = 'test' CloudshellDriverHelper.get_uuid = Mock(return_value='1234') test_uniq_name = 'test-1234' mock_client2 = Mock() test_nova_instance_service.novaclient.Client = Mock(return_value=mock_client2) # mock_client.Client = Mock(return_vaule=mock_client2) mock_image = Mock() mock_flavor = Mock() mock_client2.images.find = Mock(return_value=mock_image) mock_client2.flavors.find = Mock(return_value=mock_flavor) mock_deploy_req_model = Mock() mock_deploy_req_model.affinity_group_id = '' mock_deploy_req_model.availability_zone = 'test-avail-zone' test_nova_instance_service.udev_rules_sh_str = 'test_userdata' mock_cp_resource_model = Mock() mock_cp_resource_model.qs_mgmt_os_net_uuid = '1234' mock_cancellation_context = Mock() mock_client2.servers = Mock() mocked_inst = Mock() mock_client2.servers.create = Mock(return_value=mocked_inst) mock_qnet_dict = {'net-id': mock_cp_resource_model.qs_mgmt_os_net_uuid} result = self.instance_service.create_instance(openstack_session=self.openstack_session, name=test_name, reservation=Mock(), cp_resource_model=mock_cp_resource_model, deploy_req_model=mock_deploy_req_model, cancellation_context=mock_cancellation_context, logger=self.mock_logger) mock_client2.servers.create.assert_called_with(name=test_uniq_name, image=mock_image, flavor=mock_flavor, availability_zone='test-avail-zone', userdata='test_userdata', nics=[mock_qnet_dict]) self.assertEquals(result, mocked_inst) self.instance_service.instance_waiter.wait.assert_called_with(mocked_inst, state=self.instance_service.instance_waiter.ACTIVE, cancellation_context=mock_cancellation_context, logger=self.mock_logger)
def test_get_instance_mgmt_net_name_success(self): mock_client = Mock() test_nova_instance_service.novaclient.Client = Mock(return_value=mock_client) test_net_id = 'test_net_id' test_cp_resource_model = Mock() test_cp_resource_model.qs_mgmt_os_net_uuid = test_net_id mock_net_obj = Mock() mock_net_obj.to_dict = Mock(return_value={'id': test_net_id, 'label': 'test_returned_net'}) mock_client.networks = Mock() mock_client.networks.list = Mock(return_value=[mock_net_obj]) result = self.instance_service.get_instance_mgmt_network_name(instance=Mock(), openstack_session=self.openstack_session, cp_resource_model=test_cp_resource_model) self.assertEqual(result, 'test_returned_net')