def test_store_relationship(self):
with self.app_context():
store = StoreModel("seed store")
item = ItemModel("beans", 2.00, 1)
store.save_to_db()
item.save_to_db()
self.assertEqual(item.store.name, "seed store")
self.assertEqual(store.id, 1)
def test_crud(self):
with self.app_context():
StoreModel("seeds").save_to_db()
item = ItemModel("beans", 2.00, 1)
self.assertIsNone(ItemModel.find_by_name("beans"))
item.save_to_db()
self.assertIsNotNone(ItemModel.find_by_name("beans"))
item.delete_from_db()
self.assertIsNone(ItemModel.find_by_name("seeds"))
def test_store_relationship(self):
with self.app_context():
store = StoreModel('My store')
item = ItemModel('dress', 19.99, 1)
store.save_to_db()
item.save_to_db()
self.assertEqual(store.items.count(), 1)
self.assertEqual(store.items.first().name, 'dress')
self.assertEqual(store.json(), {
'items': [{
'name': 'dress',
'price': 19.99
}],
'name': 'My store'
})
def test_store_json(self):
with self.app_context():
store = StoreModel('My Store')
item = ItemModel('Dress', 19.90, 1)
expected = {'name': 'My Store', 'items': []}
self.assertDictEqual(store.json(), expected)
store.save_to_db()
item.save_to_db()
expected = expected = {
'name': 'My Store',
'items': [{
'name': 'Dress',
'price': 19.90
}]
}
self.assertDictEqual(store.json(), expected)
def test_crud(self):
with self.app_context():
item = ItemModel('test', 19.99)
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_store_found_with_items(self):
with self.app as client:
with self.app_context():
StoreModel('test').save_to_db()
ItemModel('dress', 19.99, 1).save_to_db()
response = client.get('/store/test', data={'name': 'dress'})
self.assertEqual(
{
'items': [{
'name': 'dress',
'price': 19.99
}],
'name': 'test'
}, json.loads(response.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):
item = ItemModel("seeds", 2.00, 1)
self.assertEqual(item.name, "seeds", "the item name is not matching")
self.assertEqual(item.price, 2.00, "the item price is not matching ")
def test_json(self):
item = ItemModel("seeds", 2.00, 1)
self.assertEqual(item.json(), {
"name": "seeds",
"price": 2.00
}), "Json export of the item is incorrect"
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'}, 401
def test_item_model_empty(self):
with self.app_context():
store = StoreModel("my store")
item = ItemModel("dress", 19.99, 1)
self.assertEqual(store.json(), {'name': 'my store', 'items': []})
self.assertListEqual(store.items.all(), [])