Тестирование методов python, вызывающих методы класса
-
20-09-2019 - |
Вопрос
У меня есть очень простой метод:
Class Team(models.Model):
def sides(self):
return SideNames.objects.filter(team=self)
SideNames - это другая модель , определенная в том же файле, что и Team,
Который, когда я пытаюсь и тестирую:
self.assertEquals(len(t.sides()), 2)
Я получаю следующую ошибку:
return SideNames.objects.filter(team=self)
Ошибка атрибута:Объект 'NoneType' не имеет атрибута 'objects'
но если я изменю тест на
self.assertEquals(len(SideNames.objects.filter(team=t)), 2)
Тогда я не получаю сообщение об ошибке.В чем разница между вызовом SideNames.objects.filter из самого теста и вызовом фактического метода?
Для справки, вот 2 класса в полном объеме.
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)
Решение
Случайно, ваш тест не делает что-то подобное:
from myapp import models
...
models.SideNames = None
поскольку это единственное объяснение, которое я могу придумать, почему побочные имена не должны быть None в контексте этого метода.
Кроме того, сам метод бессмыслен, так как обратные связи автоматически предоставляются Django, поэтому вы могли бы просто вызвать t.sidenames_set.all()
.
Другие советы
В модуле, который определяет тест, вы импортируете имя SideNames
из какого-то другого модуля.В модуле , где это sides
метод определен, имя SideNames
не определен и не импортирован.