def test_add_flavor_exist(self, value, mock_session, mock_execute): self.fc.get = mock.Mock(return_value=model_flavor.Flavor( flavor_id=value['id'])) # mock the response from cassandra mock_execute.execute.return_value = value new_flavor = flavor.load_from_json(value) self.assertRaises(ValueError, self.fc.add, new_flavor)
def test_add_flavor_exist(self, value, mock_session, mock_execute): self.fc.get = mock.Mock(return_value=model_flavor.Flavor( flavor_id=value['id'] )) # mock the response from cassandra mock_execute.execute.return_value = value new_flavor = flavor.load_from_json(value) self.assertRaises(ValueError, self.fc.add, new_flavor)
def test_get_one(self, value, mock_manager): return_flavor = flavor.load_from_json(value) # mock the storage response mock_response = return_flavor mock_manager.get.return_value = mock_response url = u'/v1.0/flavors/{0}'.format(uri.encode(value['id'])) response = self.app.get(url, headers={'X-Project-ID': self.project_id}) self.assertEqual(200, response.status_code)
def test_get_one(self, value, mock_manager): return_flavor = flavor.load_from_json(value) # mock the storage response mock_response = return_flavor mock_manager.get.return_value = mock_response url = u'/v1.0/flavors/{0}'.format(uri.encode(value['id'])) response = self.app.get(url, headers={'X-Project-ID': self.project_id}) self.assertEqual(200, response.status_code)
def test_add_flavor(self, value, mock_session, mock_execute): self.fc.get = mock.Mock(side_effect=LookupError( "More than one flavor/no record was retrieved.")) # mock the response from cassandra mock_execute.execute.return_value = value new_flavor = flavor.load_from_json(value) actual_response = self.fc.add(new_flavor) self.assertEqual(actual_response, None)
def test_add_flavor(self, value, mock_session, mock_execute): self.fc.get = mock.Mock(side_effect=LookupError( "More than one flavor/no record was retrieved." )) # mock the response from cassandra mock_execute.execute.return_value = value new_flavor = flavor.load_from_json(value) actual_response = self.fc.add(new_flavor) self.assertEqual(actual_response, None)
def post(self): """POST :returns JSON flavor(HTTP 200) or HTTP 400 """ flavors_controller = self.driver.manager.flavors_controller flavor_json = json.loads(pecan.request.body.decode('utf-8')) try: new_flavor = flavor_request.load_from_json(flavor_json) flavors_controller.add(new_flavor) # form the success response flavor_url = str( uri.encode(u'{0}/flavors/{1}'.format( self.base_url, new_flavor.flavor_id))) pecan.response.status = 201 pecan.response.headers["Location"] = flavor_url result = flavors_controller.get(new_flavor.flavor_id) return flavor_response.Model(result, self) except Exception as e: pecan.abort(400, detail=str(e))
def test_add_flavor(self, mock_session, value): new_flavor = flavor.load_from_json(value) actual_response = self.fc.add(new_flavor) self.assertEqual(actual_response, None)
def test_add_flavor(self, mock_session, value): new_flavor = flavor.load_from_json(value) actual_response = self.fc.add(new_flavor) self.assertEqual(actual_response, None)