def test_store_list(self):
store1 = StoreModel('Test1')
store2 = StoreModel('Test2')
store1.save_to_db()
store2.save_to_db()
response = self.client.get('/stores')
self.assertEqual(response.status_code, 200)
self.assertDictEqual(json.loads(response.data),
{'stores': [store1.json(), store2.json()]})
def test_store_list_with_items(self):
store = StoreModel('Test')
store.save_to_db()
ItemModel('Test Item', 19.99, 1).save_to_db()
response = self.client.get('/stores')
self.assertEqual(response.status_code, 200)
self.assertDictEqual(json.loads(response.data),
{'stores': [store.json()]})
def test_store_json_with_item(self):
store = StoreModel('Test Store')
item = ItemModel('Test', 19.99, 1)
store.save_to_db()
item.save_to_db()
expected = {
'name': 'Test Store',
'items': [{
'name': 'Test',
'price': 19.99
}]
}
self.assertEqual(store.json(), expected)
def test_find_store(self):
store = StoreModel('Test')
store.save_to_db()
response = self.client.get('/store/Test')
self.assertEqual(response.status_code, 200)
self.assertDictEqual(json.loads(response.data), store.json())