def create_pets():
"""
Creates a Pet
This endpoint will create a Pet based the data in the body that is posted
"""
data = {}
# Check for form submission data
if request.headers.get(
'Content-Type') == 'application/x-www-form-urlencoded':
app.logger.info('Getting data from form submit')
data = {
'name': request.form['name'],
'category': request.form['category'],
'available': True
}
else:
app.logger.info('Getting data from API call')
data = request.get_json()
app.logger.info(data)
pet = Pet()
pet.deserialize(data)
pet.save()
message = pet.serialize()
location_url = url_for('get_pets', pet_id=pet.id, _external=True)
return make_response(jsonify(message), status.HTTP_201_CREATED,
{'Location': location_url})
def test_serialize_a_pet(self):
pet = Pet(0, "fido", "dog")
data = pet.serialize()
self.assertNotEqual(data, None)
self.assertIn('id', data)
self.assertEqual(data['id'], 0)
self.assertIn('name', data)
self.assertEqual(data['name'], "fido")
self.assertIn('category', data)
self.assertEqual(data['category'], "dog")
def test_serialize_a_pet(self):
""" Serialize a Pet """
pet = Pet("fido", "dog", False)
data = pet.serialize()
self.assertNotEqual(data, None)
self.assertNotIn('_id', data)
self.assertIn('name', data)
self.assertEqual(data['name'], "fido")
self.assertIn('category', data)
self.assertEqual(data['category'], "dog")
self.assertIn('available', data)
self.assertEqual(data['available'], False)
def test_serialize_a_pet(self):
""" Test serialization of a Pet """
pet = Pet(name="fido", category="dog", available=False)
data = pet.serialize()
self.assertNotEqual(data, None)
self.assertIn('id', data)
self.assertEqual(data['id'], None)
self.assertIn('name', data)
self.assertEqual(data['name'], "fido")
self.assertIn('category', data)
self.assertEqual(data['category'], "dog")
self.assertIn('available', data)
self.assertEqual(data['available'], False)
def post(self):
"""
Creates a Pet
This endpoint will create a Pet based the data in the body that is posted
"""
app.logger.info('Request to Create a Pet')
check_content_type('application/json')
pet = Pet()
app.logger.info('Payload = %s', api.payload)
pet.deserialize(api.payload)
pet.save()
app.logger.info('Pet with new id [%s] saved!', pet.id)
location_url = api.url_for(PetResource, pet_id=pet.id, _external=True)
return pet.serialize(), status.HTTP_201_CREATED, {'Location': location_url}
