def create_usuario(self, cpf, nome, telegram=''):
usuario = self.session.query(Usuario).filter(Usuario.cpf == cpf).one_or_none()
if usuario:
return usuario
usuario = Usuario()
usuario.cpf = cpf
usuario.nome = nome
usuario.telegram = telegram
self.session.add(usuario)
self.session.commit()
self.session.refresh(usuario)
return usuario
def test_cpf_telegram(self):
self.login()
self._case('GET',
'/api/get_cpf_telegram/ivan',
status_code=404,
headers=self.headers)
usuario = Usuario()
usuario.telegram = 'ivan'
usuario.cpf = 'achou'
self.db_session.add(usuario)
self.db_session.commit()
r = self._case('GET',
'/api/get_cpf_telegram/ivan',
status_code=200,
headers=self.headers)
assert r['cpf'] == usuario.cpf