from django.test import TestCase, RequestFactory
from django.db.models.query import QuerySet
from animais.models import Animal

class IndexViewTestCAse(TestCase):
    def setUp(self):
        self.factory = RequestFactory()
        self.animal = Animal.objects.create(
            nome_animal = "calopsita",
            predador = "Não",
            venenoso = "Não",
            domestico = "sim"
        )

    def test_index_view_retorna_caracteristicas_do_animal(self):
        """ Teste que verifica se a index retorna as caracteristicas do animal pesquisado """
        response = self.client.get('/',
            {'buscar':'calopsita'}
        )
        caracteristicas_animal_pesquisado = response.context['caracteristicas']
        self.assertIs(type(response.context['caracteristicas']), QuerySet)
        self.assertEqual(caracteristicas_animal_pesquisado[0].nome_animal, 'calopsita')