def get_passaro(passaro_id: str):
pas = Passaro(connection)
try:
p = pas.acha(passaro_id.lower())
return {"tag": p[0], "especie": p[1], "nome_pop": p[2]}
except Exception as e:
return {"error": "Não foi possivel adicionar passaro"}
def test_lista_passaros(self):
conn = self.__class__.connection
pas = Passaro(conn)
allPas = [('Bentivinus Bolotoide', 'Bem-não-te-viu', 'Passarinho'),
('Quero Queroides', 'QueroQuerQuero', 'Monstrinho')]
# Verifica que ainda não tem pássaros no sistema.
res = pas.lista()
self.assertFalse(res)
# Adiciona alguns perigos.
passaros_id = []
for p in allPas:
pas.adiciona(*p)
passaros_id.append(pas.acha(p[0]))
# Verifica se os perigos foram adicionados corretamente.
res = pas.lista()
self.assertCountEqual(res, passaros_id)
# Remove os perigos.
for p in passaros_id:
pas.remove(p[0])
# Verifica que todos os perigos foram removidos.
res = pas.lista()
self.assertFalse(res)
def test_adiciona_passaro(self):
oldPas = ('Bentivinus Bolotoide', 'Bem-te-vi', 'Passarinho')
conn = self.__class__.connection
pas = Passaro(conn)
pas.adiciona(*oldPas)
# tenta adicionar duas vezes o passaro
try:
pas.adiciona(*oldPas)
self.fail('Nao deveria ter adicionado o mesmo passaro duas vezes.')
except ValueError as e:
pass
id = pas.acha(oldPas[0])
self.assertIsNotNone(id)
# Tenta achar um perigo inexistente.
id = pas.acha('Bentivinus NonExistum')
self.assertIsNone(id)
def test_remove_passaro(self):
oldPas = ('Bentivinus Bolotoide', 'Bem-te-vi', 'Passarinho')
conn = self.__class__.connection
pas = Passaro(conn)
pas.adiciona(*oldPas)
id = pas.acha(oldPas[0])
res = pas.lista()
self.assertCountEqual(res, (id, ))
pas.remove(oldPas[0])
res = pas.lista()
self.assertFalse(res)