def _test_flavor_list_extras(self, get_extras=None):
if get_extras:
request = self.mock_rest_request(GET={'get_extras': 'tRuE'})
elif get_extras is None:
request = self.mock_rest_request(GET={})
get_extras = False
else:
request = self.mock_rest_request(GET={'get_extras': 'fAlsE'})
self.mock_flavor_list.return_value = [FakeFlavor("1"), FakeFlavor("2")]
response = nova.Flavors().get(request)
self.assertStatusCode(response, 200)
if get_extras:
self.assertEqual(
{
"items": [{
"extras": {},
"id": "1"
}, {
"extras": {},
"id": "2"
}]
}, response.json)
else:
self.assertEqual({"items": [{
"id": "1"
}, {
"id": "2"
}]}, response.json)
self.mock_flavor_list.assert_called_once_with(request,
is_public=None,
get_extras=get_extras)
def test_flavor_create_with_access_list(self, nc):
flavor_req_data = '{"name": "flavor", ' \
'"ram": 12, ' \
'"vcpus": 1, ' \
'"disk": 2, ' \
'"OS-FLV-EXT-DATA:ephemeral": 3, ' \
'"swap": 4, ' \
'"id": "123", ' \
'"flavor_access": [{"id":"1", "name":"test"}]' \
'}'
nc.flavor_create.return_value = mock.Mock(**{
'id': '1234',
'to_dict.return_value': {'id': '1234', 'name': 'flavor'}
})
flavor_data = {'name': 'flavor',
'memory': 12,
'vcpu': 1,
'disk': 2,
'ephemeral': 3,
'swap': 4,
'flavorid': '123',
'is_public': False}
request = self.mock_rest_request(body=flavor_req_data)
response = nova.Flavors().post(request)
self.assertStatusCode(response, 201)
self.assertEqual('/api/nova/flavors/1234', response['location'])
nc.flavor_create.assert_called_once_with(request, **flavor_data)
nc.add_tenant_to_flavor.assert_called_once_with(request, '1234', '1')
def _test_flavor_list_public(self, nc, is_public=None):
if is_public:
request = self.mock_rest_request(GET={'is_public': 'tRuE'})
elif is_public is None:
request = self.mock_rest_request(GET={})
else:
request = self.mock_rest_request(GET={'is_public': 'fAlsE'})
nc.flavor_list.return_value = [
mock.Mock(**{'to_dict.return_value': {'id': '1'}}),
mock.Mock(**{'to_dict.return_value': {'id': '2'}}),
]
response = nova.Flavors().get(request)
self.assertStatusCode(response, 200)
self.assertEqual(response.content,
'{"items": [{"id": "1"}, {"id": "2"}]}')
nc.flavor_list.assert_called_once_with(request, is_public=is_public,
get_extras=False)
def _test_flavor_list_extras(self, nc, get_extras=None):
if get_extras:
request = self.mock_rest_request(GET={'get_extras': 'tRuE'})
elif get_extras is None:
request = self.mock_rest_request(GET={})
get_extras = False
else:
request = self.mock_rest_request(GET={'get_extras': 'fAlsE'})
nc.flavor_list.return_value = [
mock.Mock(**{
'extras': {},
'to_dict.return_value': {
'id': '1'
}
}),
mock.Mock(**{
'extras': {},
'to_dict.return_value': {
'id': '2'
}
}),
]
response = nova.Flavors().get(request)
self.assertStatusCode(response, 200)
if get_extras:
self.assertEqual(
{
"items": [{
"extras": {},
"id": "1"
}, {
"extras": {},
"id": "2"
}]
}, response.json)
else:
self.assertEqual({"items": [{
"id": "1"
}, {
"id": "2"
}]}, response.json)
nc.flavor_list.assert_called_once_with(request,
is_public=None,
get_extras=get_extras)
