def test_delete_business(self):
# Tests whether an owner of a business has deleted a business
self.setUpBusinessInstance()
self.setUpBusinessInstance2()
business_status = Business.delete_business(
self.setUpBusinessInstance2().user_id,
self.setUpBusinessInstance2().id)
self.assertTrue(business_status)
def one_business(business_id):
current_user = get_jwt_identity() #Current_user is username
targetbusiness = Business.get_business(business_id)
if request.method == 'GET':
return make_response(jsonify(targetbusiness)), 200
elif request.method == 'DELETE':
if current_user == targetbusiness['username']:
deletebusiness = Business.delete_business(business_id)
return make_response(jsonify(deletebusiness)), 200
else:
return jsonify(
{'message': 'You cannot delete a business that is not yours'})