def test_get_flavor_by_ram(self, mock_nova_client, mock_compute): vanilla = fakes.FakeFlavor('1', 'vanilla ice cream', 100) chocolate = fakes.FakeFlavor('1', 'chocolate ice cream', 200) mock_nova_client.flavors.list.return_value = [vanilla, chocolate] mock_response = mock.Mock() mock_response.json.return_value = dict(extra_specs=[]) mock_compute.get.return_value = mock_response flavor = self.cloud.get_flavor_by_ram(ram=150) self.assertEqual(chocolate.id, flavor['id'])
def test_get_flavor_by_ram_and_include( self, mock_nova_client, mock_compute): vanilla = fakes.FakeFlavor('1', 'vanilla ice cream', 100) chocolate = fakes.FakeFlavor('2', 'chocoliate ice cream', 200) strawberry = fakes.FakeFlavor('3', 'strawberry ice cream', 250) mock_response = mock.Mock() mock_response.json.return_value = dict(extra_specs=[]) mock_compute.get.return_value = mock_response mock_nova_client.flavors.list.return_value = [ vanilla, chocolate, strawberry] flavor = self.cloud.get_flavor_by_ram(ram=150, include='strawberry') self.assertEqual(strawberry.id, flavor['id'])
def test_delete_flavor_exception(self, mock_nova): mock_nova.flavors.list.return_value = [ fakes.FakeFlavor('123', 'lemon', 100) ] mock_nova.flavors.delete.side_effect = Exception() self.assertRaises(shade.OpenStackCloudException, self.op_cloud.delete_flavor, '')
def test_delete_flavor(self, mock_nova, mock_compute): mock_response = mock.Mock() mock_response.json.return_value = dict(extra_specs=[]) mock_compute.get.return_value = mock_response mock_nova.flavors.list.return_value = [ fakes.FakeFlavor('123', 'lemon', 100) ] self.assertTrue(self.op_cloud.delete_flavor('lemon')) mock_nova.flavors.delete.assert_called_once_with(flavor='123')
def test_list_flavors(self, nova_mock): nova_mock.flavors.list.return_value = [] self.assertEqual([], self.cloud.list_flavors()) fake_flavor = fakes.FakeFlavor('555', 'vanilla') fake_flavor_dict = meta.obj_to_dict(fake_flavor) nova_mock.flavors.list.return_value = [fake_flavor] self.cloud.list_flavors.invalidate(self.cloud) self.assertEqual([fake_flavor_dict], self.cloud.list_flavors())
def test_get_flavor_string_and_int(self, mock_nova_client, mock_compute): vanilla = fakes.FakeFlavor('1', 'vanilla ice cream', 100) mock_nova_client.flavors.list.return_value = [vanilla] mock_response = mock.Mock() mock_response.json.return_value = dict(extra_specs=[]) mock_compute.get.return_value = mock_response flavor1 = self.cloud.get_flavor('1') self.assertEqual(vanilla.id, flavor1['id']) flavor2 = self.cloud.get_flavor(1) self.assertEqual(vanilla.id, flavor2['id'])
def test_list_flavors(self, nova_mock, mock_compute): nova_mock.flavors.list.return_value = [] nova_mock.flavors.api.client.get.return_value = {} mock_response = mock.Mock() mock_response.json.return_value = dict(extra_specs={}) mock_response.headers.get.return_value = 'request-id' mock_compute.get.return_value = mock_response self.assertEqual([], self.cloud.list_flavors()) fake_flavor = fakes.FakeFlavor('555', 'vanilla', 100) fake_flavor_dict = _utils.normalize_flavors( [meta.obj_to_dict(fake_flavor)])[0] nova_mock.flavors.list.return_value = [fake_flavor] self.cloud.list_flavors.invalidate(self.cloud) self.assertEqual([fake_flavor_dict], self.cloud.list_flavors())
def test_delete_flavor(self, mock_nova): mock_nova.flavors.list.return_value = [ fakes.FakeFlavor('123', 'lemon') ] self.assertTrue(self.op_cloud.delete_flavor('lemon')) mock_nova.flavors.delete.assert_called_once_with(flavor='123')