def test_store_relationship(self):
with self.app_context():
store = StoreModel('test_store')
item = ItemModel('test', 19.99, 1)
store.save_to_db()
item.save_to_db()
self.assertEqual(item.store.name, 'test_store')
def test_store_relationship(self):
with self.app_context():
store = StoreModel('test')
item = ItemModel('test_item', 19.99, 1)
store.save_to_db()
item.save_to_db()
self.assertEqual(store.items.count(), 1)
self.assertEqual(store.items.first().name, 'test_item')
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_context():
store = StoreModel('test')
self.assertIsNone(
StoreModel.find_by_name('test'),
"Found an store with name 'test' before save_to_db")
store.save_to_db()
self.assertIsNotNone(
StoreModel.find_by_name('test'),
"Did not find an store with name 'test' after save_to_db")
store.delete_from_db()
self.assertIsNone(
StoreModel.find_by_name('test'),
"Found an store with name 'test' after delete_from_db")
