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_update_passaro(self):
newPas = ('Bentivinus Bolotoide', 'Bem-não-te-viu', 'Passarinho')
oldPas = ('Bentivinus Bolotoide', 'Bem-te-vi', 'Passarinho')
conn = self.__class__.connection
pas = Passaro(conn)
pas.adiciona(*oldPas)
pas.atualiza(*newPas)
res = pas.lista()
self.assertSequenceEqual(res, (newPas, ))
pas.remove('Bentivinus Bolotoide')
res = pas.lista()
self.assertFalse(res)
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)
def test_usuario_remove_preferencia_passaro(self):
conn = self.__class__.connection
pas = Passaro(conn)
cid = Cidade(conn)
user = Usuario(conn)
# Pega todas as cidades
cids = cid.lista()
oldPas = ('Bentivinus Bolotoide', 'Bem-te-vi', 'Passarinho')
oldUser = ('david', "*****@*****.**", "David Fogelman", cids[0][0])
oldPref = (oldUser[0], oldPas[0])
user.adiciona(*oldUser)
res = user.acha(oldUser[0])
self.assertSequenceEqual(res, oldUser)
pas.adiciona(*oldPas)
res = pas.lista()
self.assertCountEqual(res, (oldPas, ))
res = user.lista_pref(oldUser[0])
self.assertFalse(res)
user.adiciona_pref(*oldPref)
res = user.lista_pref(oldUser[0])
self.assertIsNotNone(res)
self.assertSequenceEqual(res, (oldPref, ))
def get_passaro():
pas = Passaro(connection)
passaros = pas.lista()
if passaros:
return [{
"tag": p[0],
"especie": p[1],
"nome_pop": p[2]
} for p in passaros]
return {}