def test_get_right(self):
rep = TicketRepository()
ticket_id = rep.create(seance_id='5bd897f8af13c78fe908cb98', seat_number=1)
ticket1 = rep.get(ticket_id)
ticket2 = Ticket(ticket_id=fields.ObjectId(ticket_id), seance_id='5bd897f8af13c78fe908cb98', seat_number=1)
self.assertEqual(ticket1, ticket2)
rep.delete(ticket_id)
def get(self, ticket_id):
repo = TicketRepository()
app.logger.info(
'Получен запрос на получение информации о билете с идентификатором %s'
% ticket_id)
abort_if_seance_doesnt_exist(ticket_id, repo)
ticket = repo.get(ticket_id)
response = app.make_response("")
response.status_code = 200
response.content_type = "application/json"
response.data = ticket.to_json()
app.logger.info(
'Запрос на получение информации о билете с идентификатором %s успешно обработан'
% ticket_id)
return response
def post(self):
repo = TicketRepository()
app.logger.info('Получен запрос на создание (покупку) билета')
try:
payload = jsonpickle.decode(flask.request.data)
except:
payload = {
"seance_id": "5bd897f8af13c78fe908cb98",
"seat_number": 1
}
ticket_id = repo.create(payload["seance_id"], payload["seat_number"])
ticket = repo.get(ticket_id)
response = app.make_response("")
response.content_type = "application/json"
response.status_code = 201
response.data = ticket.to_json()
app.logger.info('Бмлет с идентификатором %s успешно создан (куплен)' %
ticket_id)
return response
def test_get_error(self):
rep = TicketRepository()
ticket = rep.get('5bd89fd9')
self.assertIsNone(ticket)