Testando métodos Python que chamam métodos de classe
-
20-09-2019 - |
Pergunta
Eu tenho um método muito simples:
Class Team(models.Model):
def sides(self):
return SideNames.objects.filter(team=self)
Sidenames é outro modelo definido no mesmo arquivo que a equipe,
Que quando tento testar:
self.assertEquals(len(t.sides()), 2)
Estou tendo o erro a seguir:
return SideNames.objects.filter(team=self)
AtribuitError: o objeto 'nonType' não tem 'objetos' de atributo '
Mas se eu mudar o teste para ser
self.assertEquals(len(SideNames.objects.filter(team=t)), 2)
Então eu não recebo o erro. Qual é a diferença entre chamar sidenames.Objects.Filter do próprio teste e chamar o método real?
Para referência, aqui estão as duas classes na íntegra.
class Team(models.Model):
"""The model for a football team."""
class Admin:
pass
def __unicode__(self):
return u'%s' % self.name
def is_player(self, player):
"""Checks to see if 'player' is a member if this team. Returns True if they are, or False otherwise."""
try:
teamPlayer = TeamPlayers.objects.get(player=player, team=self)
return True
except ObjectDoesNotExist:
return False
def sides(self):
"""Return the side names for this team"""
return SideNames.objects.filter(team=self)
def updateSides(self, side_a, side_b):
"""Update the side names"""
names = SideNames.objects.filter(team=self);
a = SideNames.objects.get(name = names[0].name)
a.name = side_a
a.save()
b = SideNames.objects.get(name = names[1].name)
b.name = side_b
b.save()
name = models.CharField("Team Name", max_length=255)
organiser = models.ForeignKey(User)
class SideNames(models.Model):
"""Holds the names of the sides for each team"""
class Admin:
pass
def __unicode__(self):
"""Pretty print the SideNames object"""
return self.name
team = models.ForeignKey(Team)
name = models.CharField(max_length=128)
Solução
Por qualquer chance, seu teste faz algo assim:
from myapp import models
...
models.SideNames = None
Como essa é a única explicação que consigo pensar por que o Sidenames não deve ser nenhum no contexto desse método.
Como um aparte, o método em si é inútil, pois as relações com versões anteriores são fornecidas automaticamente pelo Django, para que você possa ligar t.sidenames_set.all()
.
Outras dicas
No módulo que define o teste, você está importando o nome SideNames
de algum outro módulo. No módulo onde isso sides
o método é definido, o nome SideNames
não é definido ou importado.