def test_crud_store(self):
with self.app_context():
store = StoreModel("my store")
self.assertIsNone(store.find_by_name('my store'))
store.save_to_db()
self.assertIsNotNone(store.find_by_name('my store'))
#self.assertEqual(store.json(),{'items':[{'name':'dress','price':19.99}],'name':'my store'})
store.delete_from_db()
self.assertIsNone(store.find_by_name('my store'))
def test_create_store(self):
with self.app as client:
with self.app_context():
resp = client.post('/store/test')
self.assertEqual(resp.status_code, 201)
self.assertIsNotNone(StoreModel.find_by_name('test'))
self.assertDictEqual({
'name': 'test',
'items': []
}, json.loads(resp.data))
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'}