def setUp(self):
# create some items
self.person = Person(firstname="Steve", lastname="Loria")
self.person2 = Person(firstname="Monty", lastname="Python")
self.item = Item(name="Foo")
self.item.save()
self.person.items.append(self.item)
self.person.save()
self.item2 = Item(name="Bar")
self.person2.save()
self.item2.save()
def setUp(self):
# create some items
self.person = Person(firstname="Steve", lastname="Loria")
self.person2 = Person(firstname="Monty", lastname="Python")
self.item = Item(name="Foo")
self.item.save()
self.person.items.append(self.item)
self.person.save()
self.item2 = Item(name="Bar")
self.person2.save()
self.item2.save()
class TestMongoengineAPI(TestCase):
TESTING = True
MONGODB_SETTINGS = {
"DB": "_test_inventory"
}
DEBUG = True
def create_app(self):
app.config.from_object(self)
return app
def setUp(self):
# create some items
self.person = Person(firstname="Steve", lastname="Loria")
self.person2 = Person(firstname="Monty", lastname="Python")
self.item = Item(name="Foo")
self.item.save()
self.person.items.append(self.item)
self.person.save()
self.item2 = Item(name="Bar")
self.person2.save()
self.item2.save()
def tearDown(self):
drop_collections()
def test_get_items(self):
url = "/api/v1/items/"
res = self.client.get(url)
data = res.json
assert_equal(res.status_code, 200)
assert_equal(len(data['items']), 2)
assert_equal(data['items'][0]['name'], self.item2.name)
def test_get_item(self):
url = '/api/v1/items/{0}'.format(self.item.id)
res = self.client.get(url)
data = res.json
assert_equal(res.status_code, 200)
assert_equal(data['name'], self.item.name)
assert_equal(data['person']['id'], str(self.person.id))
def test_get_persons(self):
res = self.client.get('/api/v1/people/')
assert_equal(res.status_code, 200)
assert_equal(len(res.json['people']), len(Person.objects))
def test_get_person(self):
res = self.client.get('/api/v1/people/{0}'.format(self.person.id))
assert_equal(res.status_code, 200)
assert_equal(res.json['name'], "{0}, {1}".format(self.person.lastname,
self.person.firstname))
assert_equal(res.json['n_items'], 1)
def test_get_nonexistent_person(self):
res = self.client.get("/api/v1/people/10")
assert_equal(res.status_code, 404)
def _post_json(self, url, data):
return self.client.post(url,
data=json.dumps(data),
content_type='application/json')
def _put_json(self, url, data):
return self.client.put(url,
data=json.dumps(data),
content_type='application/json')
def test_post_item(self):
res = self._post_json("/api/v1/items/", {"name": "Ipad"})
assert_equal(res.status_code, 201)
item = Item.objects.order_by("-updated").first()
assert_true(item is not None)
assert_equal(item.name, "Ipad")
def test_post_item_with_person_id(self):
res = self._post_json('/api/v1/items/',
{"name": "Ipod", "person_id": str(self.person.id)})
assert_equal(res.status_code, 201)
item = Item.objects.first()
person = get_item_person(item)
assert_equal(person, self.person)
def test_post_person(self):
res = self._post_json('/api/v1/people/',
{'firstname': 'Foo', 'lastname': 'Bar'})
assert_equal(res.status_code, 201)
person = Person.objects.order_by("-created").first()
assert_equal(person.firstname, "Foo")
assert_equal(person.lastname, "Bar")
def test_delete_item(self):
all_items = Item.objects
assert_in(self.item, all_items)
res = self.client.delete("/api/v1/items/{0}".format(self.item.id))
all_items = Item.objects
assert_not_in(self.item, all_items)
def test_put_item(self):
res = self._put_json("/api/v1/items/{0}".format(self.item.id),
{"checked_out": True,
"person_id": str(self.person2.id)})
item = Item.objects(id=self.item.id).first()
assert_true(item.checked_out)
item_person = get_item_person(item)
assert_equal(item_person, self.person2)
def test_delete_person(self):
all_persons = Person.objects
assert_in(self.person, all_persons)
self.client.delete('/api/v1/people/{0}'.format(self.person.id))
all_persons = Person.objects
assert_not_in(self.person, all_persons)
def test_recent(self):
self.item.checked_out = True
self.item2.checked_out = False
self.item.save()
self.item2.save()
res = self.client.get("/api/v1/recentcheckouts/")
assert_in(ItemDocSerializer(self.item._data).data, res.json['items'])
assert_not_in(ItemDocSerializer(self.item2._data).data, res.json['items'])
class TestMongoengineAPI(TestCase):
TESTING = True
MONGODB_SETTINGS = {"DB": "_test_inventory"}
DEBUG = True
def create_app(self):
app.config.from_object(self)
return app
def setUp(self):
# create some items
self.person = Person(firstname="Steve", lastname="Loria")
self.person2 = Person(firstname="Monty", lastname="Python")
self.item = Item(name="Foo")
self.item.save()
self.person.items.append(self.item)
self.person.save()
self.item2 = Item(name="Bar")
self.person2.save()
self.item2.save()
def tearDown(self):
drop_collections()
def test_get_items(self):
url = "/api/v1/items/"
res = self.client.get(url)
data = res.json
assert_equal(res.status_code, 200)
assert_equal(len(data['items']), 2)
assert_equal(data['items'][0]['name'], self.item2.name)
def test_get_item(self):
url = '/api/v1/items/{0}'.format(self.item.id)
res = self.client.get(url)
data = res.json
assert_equal(res.status_code, 200)
assert_equal(data['name'], self.item.name)
assert_equal(data['person']['id'], str(self.person.id))
def test_get_persons(self):
res = self.client.get('/api/v1/people/')
assert_equal(res.status_code, 200)
assert_equal(len(res.json['people']), len(Person.objects))
def test_get_person(self):
res = self.client.get('/api/v1/people/{0}'.format(self.person.id))
assert_equal(res.status_code, 200)
assert_equal(
res.json['name'], "{0}, {1}".format(self.person.lastname,
self.person.firstname))
assert_equal(res.json['n_items'], 1)
def test_get_nonexistent_person(self):
res = self.client.get("/api/v1/people/10")
assert_equal(res.status_code, 404)
def _post_json(self, url, data):
return self.client.post(url,
data=json.dumps(data),
content_type='application/json')
def _put_json(self, url, data):
return self.client.put(url,
data=json.dumps(data),
content_type='application/json')
def test_post_item(self):
res = self._post_json("/api/v1/items/", {
"name": "Ipad",
'checked_out': True
})
assert_equal(res.status_code, 201)
item = Item.objects.order_by("-updated").first()
assert_true(item is not None)
assert_equal(item.name, "Ipad")
assert_true(item.checked_out)
def test_post_item_with_person_id(self):
res = self._post_json('/api/v1/items/', {
"name": "Ipod",
"person_id": str(self.person.id)
})
assert_equal(res.status_code, 201)
item = Item.objects.first()
person = get_item_person(item)
assert_equal(person, self.person)
def test_post_person(self):
res = self._post_json('/api/v1/people/', {
'firstname': 'Foo',
'lastname': 'Bar'
})
assert_equal(res.status_code, 201)
person = Person.objects.order_by("-created").first()
assert_equal(person.firstname, "Foo")
assert_equal(person.lastname, "Bar")
def test_delete_item(self):
all_items = Item.objects
assert_in(self.item, all_items)
res = self.client.delete("/api/v1/items/{0}".format(self.item.id))
all_items = Item.objects
assert_not_in(self.item, all_items)
def test_put_item(self):
res = self._put_json("/api/v1/items/{0}".format(self.item.id), {
"checked_out": True,
"person_id": str(self.person2.id)
})
item = Item.objects(id=self.item.id).first()
assert_true(item.checked_out)
item_person = get_item_person(item)
assert_equal(item_person, self.person2)
def test_delete_person(self):
all_persons = Person.objects
assert_in(self.person, all_persons)
self.client.delete('/api/v1/people/{0}'.format(self.person.id))
all_persons = Person.objects
assert_not_in(self.person, all_persons)
def test_recent(self):
self.item.checked_out = True
self.item2.checked_out = False
self.item.save()
self.item2.save()
res = self.client.get("/api/v1/recentcheckouts/")
assert_in(ItemDocSerializer(self.item._data).data, res.json['items'])
assert_not_in(
ItemDocSerializer(self.item2._data).data, res.json['items'])