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_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 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