def test_crud(self):
with self.app_context():
StoreModel('costco').save_to_db()
item = ItemModel('costco', 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 test_put_update_item(self):
with self.app() as client:
with self.app_context():
StoreModel('test').save_to_db()
ItemModel('test', 5.99, 1).save_to_db()
self.assertEqual(ItemModel.find_by_name('test').price, 5.99)
resp = client.put('/item/test',
data={
'price': 17.99,
'store_id': 1
})
self.assertEqual(resp.status_code, 200)
self.assertEqual(ItemModel.find_by_name('test').price, 17.99)
self.assertDictEqual({
'name': 'test',
'price': 17.99
}, json.loads(resp.data))
def put(self, name):
data = Item.parser.parse_args()
item = ItemModel.find_by_name(name)
if item is None:
item = ItemModel(name, **data)
else:
item.price = data['price']
item.save_to_db()
return item.json()
def post(self, name):
if ItemModel.find_by_name(name):
return {'message': "An item with name '{}' already exists.".format(name)}, 400
data = Item.parser.parse_args()
item = ItemModel(name, **data)
try:
item.save_to_db()
except:
return {"message": "An error occurred inserting the item."}, 500
return item.json(), 201
def test_create_item(self):
with self.app_context():
item_test = ItemModel('salt', 3.50)
item_test.save_to_db()
self.assertIsNotNone(ItemModel.find_by_name('salt'))
def delete(self, name):
item = ItemModel.find_by_name(name)
if item:
item.delete_from_db()
return {'message': 'Item deleted'}
def get(self, name):
item = ItemModel.find_by_name(name)
if item:
return item.json()
return {'message': 'Item not found'}, 404
