def test_save_item_into_store(self):
with self.app_context():
store = StoreModel("test")
store.save_to_db()
item = ItemModel("salt", 3.20, 1)
item.save_to_db()
self.assertEqual(store.items.count(), 1)
def test_stores_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_json_with_item(self):
with self.app_context():
store = StoreModel("test")
store.save_to_db()
item = ItemModel("sugar", 4.50, 1)
item.save_to_db()
expected_json = {
"name": "test",
"items": [{"name": "sugar", "price": 4.50}]
}
self.assertDictEqual(expected_json, store.json())
def test_crud(self):
with self.app_context():
StoreModel('test').save_to_db()
item = ItemModel('test', 19.99, 1)
self.assertIsNone(
ItemModel.find_by_name('test'),
"Found an item with name {}, but expected not to.".format(
item.name))
item.save_to_db()
self.assertIsNotNone(ItemModel.find_by_name('test'))
item.delete_from_db()
self.assertIsNone(ItemModel.find_by_name('test'))
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 get(self, name):
store = StoreModel.find_by_name(name)
if store:
return store.json()
return {'message': 'Store not found'}, 404
def delete(self, name):
store = StoreModel.find_by_name(name)
if store:
store.delete_from_db()
return {'message': 'Store deleted'}
def test_create_store(self):
store = StoreModel("test")
self.assertEqual(store.name, "test")
def test_create_store_with_no_items(self):
store = StoreModel("test")
self.assertListEqual(store.items.all(), [])
def test_save_store_into_db(self):
with self.app_context():
store = StoreModel("test")
store.save_to_db()
self.assertIsNotNone(StoreModel.find_by_name("test"))