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)