def test_crud(self):
with self.app_context():
store = StoreModel("Costco")
self.assertIsNotNone(store.name)
self.assertIsNone(StoreModel.find_by_name("Costco"))
store.save_to_db()
self.assertIsNotNone(StoreModel.find_by_name("Costco"))
store.delete_from_db()
self.assertIsNone(StoreModel.find_by_name("Costco"))
def delete(self, name):
store = StoreModel.find_by_name(name)
if (store):
store.delete_from_db()
else:
return {'message': 'Store not found'}
return {'message': 'Store Delted'}
def get(self, name):
store = StoreModel.find_by_name(name)
if store:
return store.json()
return {
'message': '{} does not exist on the database'.format(name)
}, 404
def post(self, name):
if StoreModel.find_by_name(name):
return {'Message': "Store '{}' already present.".format(name)}, 400
store = StoreModel(name)
try:
store.save_to_db()
except:
return {"message": "An error occured while inserting a store"}, 50
return store.json(), 201
def get(self, name):
item = StoreModel.find_by_name(name)
if item:
return {
'items': item.json()
} #item is an object and we have to return a dictionary
return {'message': 'item not found'}, 404
def delete(self,name):
claims = get_jwt_claims()
if not claims['is_admin']:
return {'message':'Admin privileges required for this operation'}
store = StoreModel.find_by_name(name)
if(store):
store.delete_from_db()
else:
return {'message': 'Store not found'}
return {'message': 'Store Delted'}
def test_create_store(self):
with self.app() as client:
with self.app_context():
response = client.post("/store/costco")
self.assertEqual(response.status_code, 201)
self.assertIsNotNone(StoreModel.find_by_name("costco"))
self.assertDictEqual({
"name": "costco",
"item_list": []
}, json.loads(response.data))
def post(self, name):
if (StoreModel.find_by_name(name)):
return {
'message': "store '{}'is already present".format(name)
}, 400
store = StoreModel(name)
try:
store.save_to_db()
except:
return {'message': 'an error occured while insering the item'}, 500
return store.json(), 201
def post(self, name):
if (StoreModel.find_by_name(name)):
return {"message": "Store {} already present".format(name)}
store = StoreModel(name)
try:
store.save_to_db()
except:
return {"message": 'An error occured while inserting data'}, 500
return store.json(), 201
def get(self, name):
"""
Find a store by name
GET /store/<string:name>
"""
exist = StoreModel.find_by_name(name)
if not exist:
return {"message": "Sorry, No Store Found!"}, 404
else:
return exist.json(), 200
def post(self, name):
if StoreModel.find_by_name(name):
return {'message': '{} already exists'.format(name)}, 400
store = StoreModel(name)
try:
store.save_to_db()
except:
return {'message': 'An error occurred'}, 500
return store.json(), 201
def post(self, name):
if StoreModel.find_by_name(name):
return {
'Message': 'Store name {} already exists.'.format(name)
}, 400
store = StoreModel(name)
try:
store.save_db()
except:
return {'Message': 'An Error Occuured While creating store'}
return store.json(), 201
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 delete(self):
"""
Delete a store
DELETE /store?store_name=<string:name>
"""
data = request.args
store_name = data['store_name']
exist = StoreModel.find_by_name(store_name)
if exist:
exist.delete()
return {"message": "Item Removed!"}, 200
return {"message", "Store don't exist!"}, 404
def post(self, name):
store = StoreModel.find_by_name(name)
if store:
return {'message': 'Store already exists.'}, 400 # Bad Request
store = StoreModel(name)
try:
store.save_to_db()
except:
return {
'message':
'An error occurred attempting to save the store to the database'
}, 500 # Internal Server Error
return store.json(), 201 # Created
def post(self):
"""
Create a new Store
POST /store
"""
parser = reqparse.RequestParser()
parser.add_argument('store_name',
type=str,
required=True,
help="Store Name couldn't be blank")
data = parser.parse_args()
name = data['store_name']
exist = StoreModel.find_by_name(name)
if exist:
return {"message": "Store already exist!"}, 409
store = StoreModel(name)
store.save_to_db()
return store.json(), 201
def delete(self, name):
store = StoreModel.find_by_name(name)
if store:
store.delete_from_db()
return {'message': 'item deleted'}, 201
def get(self, name):
store = StoreModel.find_by_name(name)
if (store):
return store.json(), 200
return {"message": "Store not Found"}
def delete(self, name):
store = StoreModel.find_by_name(name)
if store:
store.delete_from_db()
return {"message": "Store has been deleted"}
return {"message": "Store not found"}
def delete(self, name):
store = StoreModel.find_by_name(name)
if store:
store = StoreModel.delete_db()
return {'Message': 'Store deleted'}
def get(self, name):
store = StoreModel.find_by_name(name)
if store:
return store.json()
return {"message": "store not found"}, 404
def get(self, name):
store = StoreModel.find_by_name(name)
if store:
return store.json()
return {'message': 'Store not found.'}, 404 # Not Found
