def test_minhas_fichas(self):
self.unauthorized('/api/minhas_fichas/ghost')
self.login()
self._case('GET',
'/api/minhas_fichas/ghost',
status_code=404,
headers=self.headers)
usuario = Usuario()
usuario.cpf = 'ficha_test'
ovr = OVR()
ovr.responsavel_cpf = usuario.cpf
self.db_session.add(usuario)
self.db_session.add(ovr)
self.db_session.commit()
self._case('GET',
'/api/minhas_fichas/ficha_test',
status_code=400,
headers=self.headers)
ovr.datahora = datetime.today()
self.db_session.add(ovr)
self.db_session.commit()
self._case('GET',
'/api/minhas_fichas/ficha_test',
status_code=200,
headers=self.headers)
def test_Setores_Usuarios(self):
setorpai = Setor()
setorpai.id = 11
setorpai.nome = 'Pai de Todos'
session.add(setorpai)
setor1 = Setor()
setor1.id = 12
setor1.pai_id = 11
setor1.nome = 'Filho 1'
session.add(setor1)
setor2 = Setor()
setor2.id = 13
setor2.pai_id = 11
setor2.nome = 'Filho 2'
session.add(setor2)
usuario_setor = Usuario()
usuario_setor.cpf = 'chefe'
usuario_setor.nome = 'user_setorpai'
usuario_setor.setor_id = setorpai.id
session.add(usuario_setor)
usuario_setor1 = Usuario()
usuario_setor1.cpf = '1'
usuario_setor1.nome = 'user_setor1'
usuario_setor1.setor_id = setor1.id
session.add(usuario_setor1)
usuario_setor2 = Usuario()
usuario_setor2.cpf = '2'
usuario_setor2.nome = 'user_setor2'
usuario_setor2.setor_id = setor2.id
session.add(usuario_setor2)
session.commit()
setores = get_setores_choice(session)
assert isinstance(setores, list)
setores = get_setores_cpf(session, '1')
assert len(setores) == 1
assert isinstance(setores[0], Setor)
usuario_vazio = get_setores_cpf(session, '123456')
assert len(usuario_vazio) == 0
assert isinstance(usuario_vazio, list)
setores = get_setores_usuario(session, usuario_setor)
assert len(setores) == 3
assert isinstance(setores[0], Setor)
def create_usuarios(session):
usuarios = [('mycroft', 'm5', 1), ('holmes', 'sherlock', 2),
('watson', 'dr', 2), ('adler', 'irene', 2),
('lestrade', 'inspetor', 3), ('macdonald', 'inspetor', 3),
('ivan', 'ivan', 3)]
for linha in usuarios:
usuario = Usuario()
usuario.cpf = linha[0]
usuario.nome = linha[0]
usuario.setor_id = linha[2]
session.add(usuario)
session.commit()
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 create_usuarios(session):
usuarios = [('carlos', 'carlos', 40), ('erika', 'erika', 50),
('usuarioA1', 'usuarioA1', 50), ('usuarioA2', 'usuarioA2', 50),
('kanoo', 'kanoo', 60), ('usuarioB3', 'usuarioB3', 60),
('usuarioB4', 'usuarioB4', 60)]
for linha in usuarios:
usuario = Usuario()
usuario.cpf = linha[0]
usuario.nome = linha[1]
usuario.setor_id = linha[2]
# usuario.perfis = linha[3]
session.add(usuario)
session.commit()
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
def create_usuario(self, cpf, nome, setor: Setor = None):
usuario = self.session.query(Usuario).filter(Usuario.cpf == cpf).one_or_none()
if usuario:
return usuario
usuarios = get_usuarios(self.session)
numeroatual = len(usuarios)
usuario = Usuario()
usuario.cpf = cpf
usuario.nome = nome
if setor:
usuario.setor_id = setor.id
self.session.add(usuario)
self.session.commit()
usuarios = get_usuarios(self.session)
# assert usuarios[numeroatual][0] == usuario.cpf
return usuario
from bhadrasana.models import ENaoAutorizado, Usuario, ESomenteMesmoUsuario, ESomenteUsuarioResponsavel
from bhadrasana.models import ovr
from bhadrasana.models.rvfmanager import get_rvfs_filtro, cadastra_rvf, programa_rvf_container
from .test_base import BaseTestCase
sys.path.append('.')
from bhadrasana.models.rvf import metadata, RVF, ImagemRVF
engine = create_engine('sqlite://')
Session = sessionmaker(bind=engine)
session = Session()
metadata.create_all(engine)
ovr.create_tiposevento(session)
usuario = Usuario()
usuario.cpf = 'teste'
session.add(usuario)
usuario2 = Usuario()
usuario2.cpf = 'teste2'
session.add(usuario2)
session.commit()
class RVFTestCase(BaseTestCase):
def setUp(self) -> None:
super().setUp(session)
metadata.create_all(engine,
[
metadata.tables['ovr_marcasencontradas'],
metadata.tables['ovr_lacres'],