def test_update(self, client):
""" Test updating an entity using UpdateAPIResource """
# Create a dog object
Dog.create(id=5, name='Johnny', owner='John')
# Update the dog object
rv = client.put('/dogs/5',
data=json.dumps(dict(age=3)),
content_type='application/json')
assert rv.status_code == 200
expected_resp = {
'dog': {
'age': 3,
'id': 5,
'name': 'Johnny',
'owner': 'John'
}
}
assert rv.json == expected_resp
# Test value has been updated in the db
dog = Dog.get(5)
assert dog is not None
assert dog.age == 3
# Delete the dog now
dog = Dog.get(5)
dog.delete()
def test_create(self, client):
""" Test creating an entity using CreateAPIResource """
# Create a dog object
rv = client.post('/dogs',
data=json.dumps(
dict(id=5, name='Johnny', owner='John')),
content_type='application/json')
assert rv.status_code == 201
expected_resp = {
'dog': {
'age': 5,
'id': 5,
'name': 'Johnny',
'owner': 'John'
}
}
assert rv.json == expected_resp
# Test value has been added to db
dog = Dog.get(5)
assert dog is not None
assert dog.id == 5
# Delete the dog now
dog = Dog.get(5)
dog.delete()
def test_show(self, client):
""" Test retrieving an entity using ShowAPIResource"""
# Create a dog object
Dog.create(id=5, name='Johnny', owner='John')
# Fetch this dog by ID
rv = client.get('/dogs/5')
assert rv.status_code == 200
expected_resp = {
'dog': {
'age': 5,
'id': 5,
'name': 'Johnny',
'owner': 'John'
}
}
assert rv.json == expected_resp
# Test search by invalid id
rv = client.get('/dogs/6')
assert rv.status_code == 404
# Delete the dog now
dog = Dog.get(5)
dog.delete()
def test_custom_route(self):
""" Test custom routes using the blueprint resource set """
# Create a human object
Human.create(id=1, name='John')
Dog.create(id=5, name='Johnny', owner='John')
# Get the custom route
rv = self.client.get('/humans/1/my_dogs')
assert rv.status_code == 200
assert rv.json['total'] == 1
assert rv.json['dogs'][0] == {'age': 5, 'id': 5, 'name': 'Johnny', 'owner': 'John'}
def test_delete(self, client):
""" Test deleting an entity using DeleteAPIResource """
# Create a dog object
Dog.create(id=5, name='Johnny', owner='John')
# Delete the dog object
rv = client.delete('/dogs/5')
assert rv.status_code == 204
assert rv.data == b''
# Test value has been updated in the db
with pytest.raises(ObjectNotFoundError):
Dog.get(5)
def test_list(self, client):
""" Test listing an entity using ListAPIResource """
# Create a dog objects
Dog.create(id=1, name='Johnny', owner='John')
Dog.create(id=2, name='Mary', owner='John', age=3)
Dog.create(id=3, name='Grady', owner='Jane', age=8)
Dog.create(id=4, name='Brawny', owner='John', age=2)
# Get the list of dogs
rv = client.get('/dogs?order_by[]=age')
assert rv.status_code == 200
assert rv.json['total'] == 4
assert rv.json['dogs'][0] == {
'age': 2,
'id': 4,
'name': 'Brawny',
'owner': 'John'
}
# Test with filters
rv = client.get('/dogs?owner=Jane')
assert rv.status_code == 200
assert rv.json['total'] == 1
def test_custom_route(self, client):
""" Test custom routes using the resource set """
# Create a human object
Human.create(id=1, name='John')
Dog.create(id=1, name='Johnny', owner='John')
Dog.create(id=2, name='Mary', owner='John', age=3)
Dog.create(id=3, name='Grady', owner='Jane', age=8)
# Get the custom route
rv = client.get('/humans/1/my_dogs')
assert rv.status_code == 200
assert rv.json['total'] == 2
assert rv.json['dogs'][0] == {
'age': 3,
'id': 2,
'name': 'Mary',
'owner': 'John'
}