Тестирование методов python, вызывающих методы класса

StackOverflow https://stackoverflow.com/questions/2256063

Вопрос

У меня есть очень простой метод:

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 не определен и не импортирован.

Лицензировано под: CC-BY-SA с атрибуция
Не связан с StackOverflow
scroll top