def test_store_relashionship(self):
with self.app_context:
store = StoreModel('costco')
item = ItemModel('pc', 1500, 1)
store.save_to_db()
item.save_to_db()
self.assertEqual(item.store.name, 'costco')
def test_relationship_items(self):
with self.app_contex():
store = StoreModel('costco')
item = ItemModel('pc', 1500, 1)
store.save_to_db()
item.save_to_db()
self.assertEqual(store.items.count(), 1)
self.assertEqual(store.items.first().name, 'pc')
def test_jason(self):
store = StoreModel('costco')
item = ItemModel('pc', 1500, 1)
store.save_to_db()
item.save_to_db()
expected = {'name': 'costco', 'items': [{'name': 'pc', 'price': 1500}]}
self.assertEqual(store.json(), expected)
def post(self, name):
if StoreModel.find_by_name(name):
return {'message': "A store with name '{}' already exists.".format(name)}, 400
store = StoreModel(name)
try:
store.save_to_db()
except:
return {"message": "An error occurred creating the store."}, 500
return store.json(), 201
def test_crud(self):
with self.app_contex():
store = StoreModel('costco')
self.assertIsNone(StoreModel.find_by_name('costco'))
store.save_to_db()
self.assertIsNotNone(StoreModel.find_by_name('test'))
store.delete_from_db()
self.assertIsNone(StoreModel.find_by_name('costco'))