def test_remove_ticket_true(self):
rep = UserRepository()
user_id = rep.create('name', 'password')
rep.assign_ticket(user_id, 'abc')
boolean = rep.remove_ticket(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', 'ticket_id': '894bjhel892'}
if payload["status"] == "buy":
app.logger.info('Покупка билета с идентификатором %s' % payload["ticket_id"])
repo.assign_ticket(user_id, payload["ticket_id"])
else:
app.logger.info('Возврат билета с идентификатором %s' % payload["ticket_id"])
repo.remove_ticket(user_id, payload["ticket_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["ticket_id"], user_id))
else:
app.logger.info('Возврат билета %s для пользователя %s успешно произведен'
% (payload["ticket_id"], user_id))
return response
def test_assign_ticket_false(self):
rep = UserRepository()
boolean = rep.assign_ticket('5bd0a397', 'abc')
self.assertFalse(boolean)