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