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 test_create_item(self): item = ItemModel('test', 19.99, 1) self.assertEqual(item.name, 'test', "The name of the item after creation does not equal the constructor argument.") self.assertEqual(item.price, 19.99, "The price of the item after creation does not equal the constructor argument.") self.assertEqual(item.store_id, 1, "The store_id of the item after creation does not equal the constructor argument.") self.assertIsNone(item.store, "The item's store was not None even though the store was not created.")
def test_item_json(self): item = ItemModel('test', 19.99, 1) expected = { 'id': None, 'name': 'test', 'price': 19.99 } self.assertEqual( item.json(), expected, "The JSON export of the item is incorrect. Received {}, expected {}.".format(item.json(), expected))
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_store_with_items_found(self): with self.app() as c: with self.app_context(): StoreModel('test').save_to_db() ItemModel('test', 2.99, 1).save_to_db() r = c.get('/store/test') self.assertEqual(r.status_code, 200) self.assertDictEqual(d1={ 'id': 1, 'name': 'test', 'items': [{ 'price': 2.99, 'id': 1, 'name': 'test' }] }, d2=json.loads(r.data.decode()))
def test_store_with_items_list(self): with self.app() as c: with self.app_context(): StoreModel('test').save_to_db() ItemModel('test', 17.99, 1).save_to_db() r = c.get('/stores') self.assertDictEqual(d1={ 'stores': [{ 'name': 'test', 'id': 1, 'items': [{ 'id': 1, 'name': 'test', 'price': 17.99 }] }] }, d2=json.loads(r.data.decode()))
def test_crud(self): with self.app_context(): store = StoreModel('test') store.save_to_db() item = ItemModel('test', 19.99, 1) self.assertIsNone( ItemModel.find_by_name('test'), "Found an item with name 'test' before save_to_db") item.save_to_db() self.assertIsNotNone( ItemModel.find_by_name('test'), "Did not find an item with name 'test' after save_to_db") item.delete_from_db() self.assertIsNone( ItemModel.find_by_name('test'), "Found an item with name 'test' after delete_from_db")