def test_colaborador_get(client):
app, db = client[0], client[1]
colaborador = Colaborador(nome="Lucas Farias de Oliveira",
cpf="02142020210",
email="*****@*****.**",
celular="92999999999",
empresa="Pontotel",
cargo="Desenvolvedor Jr",
endereco="Rua 123, n 123, bairro ABCDE",
cep="69099999",
ativo=True)
db.session.add(colaborador)
db.session.flush()
db.session.commit()
print('---------------------------------------------')
print("Get Colaborador iniciando...")
response = app.get('/colaborador/' + str(colaborador.id))
print("Response: ", response.json)
assert response.status_code == 200
print('Get Colaborador finalizado...')
print('---------------------------------------------')
def test_colaborador_put(client):
app, db = client[0], client[1]
colaborador = Colaborador(nome="Lucas Farias de Oliveira",
cpf="02142020210",
email="*****@*****.**",
celular="92999999999",
empresa="Pontotel",
cargo="Desenvolvedor Jr",
endereco="Rua 123, n 123, bairro ABCDE",
cep="69099999",
ativo=True)
db.session.add(colaborador)
db.session.flush()
db.session.commit()
print('---------------------------------------------')
print("Put Colaborador iniciando...")
url = '/colaborador/' + str(colaborador.id)
data = {"nome": "João Farias de Oliveira", "cpf": "02142020211"}
headers = {
'Content-Type': 'application/json',
'Accept': 'application/json'
}
response = app.put(url, data=json.dumps(data), headers=headers)
assert response.status_code == 200
print("Put Colaborador finalizado...")
print('---------------------------------------------')
def test_ponto_out_delete(client):
app, db = client[0], client[1]
colaborador = Colaborador(nome="Lucas Farias de Oliveira",
cpf="02142020210",
email="*****@*****.**",
celular="92999999999",
empresa="Pontotel",
cargo="Desenvolvedor Jr",
endereco="Rua 123, n 123, bairro ABCDE",
cep="69099999",
ativo=True)
db.session.add(colaborador)
db.session.flush()
db.session.commit()
ponto_out = Ponto(tipo_de_ponto="out", colaborador_id=colaborador.id)
db.session.add(ponto_out)
db.session.flush()
db.session.commit()
print('---------------------------------------------')
print("Delete Ponto Out iniciando...")
response = app.delete('/ponto/' + str(ponto_out.id))
assert response.status_code == 200
print("Delete Ponto Out finalizado...")
print('---------------------------------------------')
def save(self, data):
colaborador = Colaborador.query.filter_by(cpf=data['cpf'], ativo=True).first()
if not colaborador:
colaborador = Colaborador(
nome=data['nome'],
cpf=data['cpf'],
email=data['email'],
celular=data['celular'],
empresa=data['empresa'],
cargo=data['cargo'],
endereco=data['endereco'],
cep=data['cep'],
ativo=True
)
db.session.add(colaborador)
try:
db.session.flush()
db.session.commit()
return {'status': 'success', 'message': 'Colaborador adicionado com sucesso', 'id': colaborador.id}, 200
except Exception as err:
db.session.rollback()
return {'status': 'fail', 'message': 'Não foi possível adicionar o colaborador no momento'}, 500
finally:
db.session.close()
else:
return {'status': 'fail', 'message': 'Já existe um colaborador ativo com este CPF'}, 400
def test_ponto_out_put(client):
app, db = client[0], client[1]
colaborador = Colaborador(nome="Lucas Farias de Oliveira",
cpf="02142020210",
email="*****@*****.**",
celular="92999999999",
empresa="Pontotel",
cargo="Desenvolvedor Jr",
endereco="Rua 123, n 123, bairro ABCDE",
cep="69099999",
ativo=True)
db.session.add(colaborador)
db.session.flush()
db.session.commit()
ponto_out = Ponto(tipo_de_ponto="out", colaborador_id=colaborador.id)
db.session.add(ponto_out)
db.session.flush()
db.session.commit()
print('---------------------------------------------')
print("Put Ponto Out iniciando...")
url = '/ponto/' + str(ponto_out.id)
data = {
"created_at": "2019-09-16T20:38:43.563Z",
"last_modified_at": "2019-09-16T20:38:43.563Z"
}
headers = {
'Content-Type': 'application/json',
'Accept': 'application/json'
}
response = app.put(url, data=json.dumps(data), headers=headers)
assert response.status_code == 200
print("Put Ponto Out finalizado...")
print('---------------------------------------------')