def test_remove_ticket_true(self):
rep = UserRepository()
user_id = rep.create('l_name', 'p_name')
rep.assign_prod(user_id, 'abc')
boolean = rep.remove_prod(user_id, 'abc')
self.assertTrue(boolean)
rep.delete(user_id)
def patch(self, user_id):
repo = UserRepository()
app.logger.info(
'Получен запрос на покупку/возврат товара для пользователя с идентификатором %s'
% user_id)
abort_if_user_doesnt_exist(user_id, repo)
try:
payload = jsonpickle.decode(flask.request.data)
except:
payload = {'status': 'buy', 'prod_id': '894bjhel892'}
if payload["status"] == "buy":
app.logger.info('Покупка товара с идентификатором %s' %
payload["prod_id"])
repo.assign_prod(user_id, payload["prod_id"])
else:
app.logger.info('Возврат товара с идентификатором %s' %
payload["prod_id"])
repo.remove_prod(user_id, payload["prod_id"])
user = repo.get(user_id)
response = app.make_response("")
response.status_code = 201
response.data = user.to_json()
response.content_type = "application/json"
if payload["status"] == "buy":
app.logger.info(
'Покупка товара %s для пользователя %s успешно произведена' %
(payload["prod_id"], user_id))
else:
app.logger.info(
'Возврат товара %s для пользователя %s успешно произведен' %
(payload["prod_id"], user_id))
return response
def test_assign_ticket_false(self):
rep = UserRepository()
boolean = rep.assign_prod('5bd0a397', 'abc')
self.assertFalse(boolean)