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')
