Тестирование методов Python, которые вызывают методы класса - PullRequest
Новая
       10

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

0 голосов
/ 13 февраля 2010

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

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)

AttributeError: объект '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)

Ответы [ 2 ]

1 голос
/ 13 февраля 2010

Случайно, ваш тест делает что-то вроде этого: 

from myapp import models

...

models.SideNames = None

, так как это единственное объяснение, которое я могу придумать, почему SideNames должно быть None в контексте этого метода.

Кроме того, сам метод бессмысленен, так как обратные связи автоматически обеспечиваются Django, так что вы можете просто вызвать t.sidenames_set.all().

1 голос
/ 13 февраля 2010

В модуле, который определяет тест, вы импортируете имя SideNames из другого модуля. В модуле, где определен метод sides, имя SideNames не определено или не импортировано.

...