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):
repo = TicketRepository()
app.logger.info('Получен запрос на получение списка билетов')
try:
args = self.parser.parse_args(strict=True)
except:
args = {'page': 1, 'page_size': 5}
app.logger.info(
'Номер страницы: %d; количество билетов на странице: %d' %
(args['page'], args['page_size']))
ticket_list, is_prev_page, is_next_page = repo.read_paginated(
page_number=args['page'], page_size=args['page_size'])
tickets = ''
for ticket in ticket_list:
tickets += ticket.to_json() + '\n'
dictr = {"is_prev_page": is_prev_page, "is_next_page": is_next_page}
tickets += "\n" + json.dumps(dictr)
response = app.make_response("")
response.content_type = "application/json"
response.status_code = 200
response.data = tickets
app.logger.info('Запрос на получение списка билетов успешно обработан')
return response
def test_exists_false(self):
rep = TicketRepository()
boolean = rep.exists('5bd8ad1daf')
self.assertFalse(boolean)
def test_exists_true(self):
rep = TicketRepository()
ticket_id = rep.create(seance_id='5bd897f8af13c78fe908cb98', seat_number=1)
boolean = rep.exists(ticket_id)
self.assertTrue(boolean)
rep.delete(ticket_id)
def test_delete_existed(self):
rep = TicketRepository()
id1 = rep.create('5bd897f8af13c78fe908cb98', 2)
rep.delete(id1)
self.assertFalse(rep.exists(id1))
def test_read_paginated(self):
rep = TicketRepository()
tickets = rep.read_paginated(1, 5)
self.assertLessEqual(len(tickets), 5)
def test_get_error(self):
rep = TicketRepository()
ticket = rep.get('5bd89fd9')
self.assertIsNone(ticket)