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 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 test_get_none(self):
rep = KassirRepository()
kassir = rep.get('5bd8ad')
self.assertIsNone(kassir)