def test_create(self):
rep = KassirRepository()
id1 = rep.create('a', 'a', 10)
id2 = rep.create('a', 'a', 10)
self.assertNotEqual(id1, id2)
rep.delete(id1)
rep.delete(id2)
def delete(self, kassir_id):
repo = KassirRepository()
app.logger.info(
'Получен запрос на увольнение кассира с идентификатором %s' %
kassir_id)
abort_if_kassir_doesnt_exist(kassir_id, repo)
repo.delete(kassir_id)
response = app.make_response("Kassir %s deleted successfully" %
kassir_id)
response.status_code = 204
app.logger.info('Кассир с идентификатором %s успешно уволен' %
kassir_id)
return response
def get(self, kassir_id):
repo = KassirRepository()
app.logger.info(
'Получен запрос на получение информации о кассире с идентификатором %s'
% kassir_id)
abort_if_kassir_doesnt_exist(kassir_id, repo)
kassir = repo.get(kassir_id)
response = app.make_response("")
response.status_code = 200
response.content_type = "application/json"
response.data = kassir.to_json()
app.logger.info(
'Запрос на получение информации о кассире с идентификатором %s успешно обработан'
% kassir_id)
return response
def post(self):
repo = KassirRepository()
app.logger.info('Получен запрос по найму кассира')
try:
payload = jsonpickle.decode(flask.request.data)
except:
payload = {'name': '1', 'stage': '2', 'year': 60}
kassir_id = repo.create(payload["name"], payload["stage"],
payload["year"])
kassir = repo.get(kassir_id)
response = app.make_response("")
response.status_code = 201
response.content_type = "application/json"
response.data = kassir.to_json() #jsonpickle.encode(kassir)
app.logger.info('Кассир с идентификатором %s успешно нанят' %
kassir_id)
return response
def get(self):
repo = KassirRepository()
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']))
kassirs_list, is_prev_page, is_next_page = repo.read_paginated(
page_number=args['page'], page_size=args['page_size'])
kassirs = ''
for kassir in kassirs_list:
kassirs += "\n" + kassir.to_json()
dictr = {"is_prev_page": is_prev_page, "is_next_page": is_next_page}
kassirs += "\n" + json.dumps(dictr)
response = app.make_response("")
response.status_code = 200
response.content_type = "application/json"
response.data = kassirs
app.logger.info(
'Запрос на получение списка кассиров успешно обработан')
return response
def test_get_right(self):
rep = KassirRepository()
kassir_id = rep.create('a', 'a', 100)
kassir1 = rep.get(kassir_id)
kassir2 = Kassir(kassir_id=fields.ObjectId(kassir_id),
name='a',
razryad='a',
year=100)
self.assertEqual(kassir1, kassir2)
rep.delete(kassir_id)
def test_delete_existed(self):
rep = KassirRepository()
id1 = rep.create('a', 'a', 10)
rep.delete(id1)
self.assertFalse(rep.exists(id1))
def test_read_paginated(self):
rep = KassirRepository()
kassirs = rep.read_paginated(1, 5)
self.assertLessEqual(len(kassirs), 5)
def test_exists_false(self):
rep = KassirRepository()
boolean = rep.exists('5bd8ad1daf')
self.assertFalse(boolean)
def test_exists_true(self):
rep = KassirRepository()
kassir_id = rep.create('a', 'a', 100)
boolean = rep.exists(kassir_id)
self.assertTrue(boolean)
rep.delete(kassir_id)
def test_get_none(self):
rep = KassirRepository()
kassir = rep.get('5bd8ad')
self.assertIsNone(kassir)