def test_crud(self):
with self.app_contex():
store = StoreModel('costco')
self.assertIsNone(StoreModel.find_by_name('costco'))
store.save_to_db()
self.assertIsNotNone(StoreModel.find_by_name('test'))
store.delete_from_db()
self.assertIsNone(StoreModel.find_by_name('costco'))
def test_delete_store(self):
with self.app() as client:
with self.app_context():
StoreModel('costco').save_to_db()
request = client.delete('/store/costco')
self.assertIsNone(StoreModel.find_by_name('costco'))
self.assertEqual(request.status_code, 200)
self.assertDictEqual({'message': 'Store deleted'},
json.loads(request.data))
def test_create_store(self):
with self.app() as client:
with self.app_context():
request = client.post('/store/costco')
self.assertEqual(request.satus_code, 201)
self.assertIsNotNone(StoreModel.find_by_name('costco'))
self.assertDictEqual({
'name': 'test',
'item': []
}, json.loads(request.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 test_create_store(self):
with self.app as client:
# to access the db
with self.app_context():
# client.delete('Superstore')
response = client.post('/store/' + StoreTest.store)
self.assertEqual(response.status_code, 201)
self.assertIsNotNone(StoreModel.find_by_name(StoreTest.store))
self.assertDictEqual({
'items': [],
'name': 'Superstore'
}, json.loads(response.data.decode('utf-8')))
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'}