测试调用类方法的python方法

时间:2010-02-13 00:57:59

标签: python unit-testing python-unittest

我有一个非常简单的方法:

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和调用实际方法之间的区别是什么?

供参考,以下是两个完整的课程。

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 个答案:

答案 0 :(得分:1)

在定义测试的模块中,您要从其他模块导入名称SideNames。在定义sides方法的模块中,未定义或导入名称SideNames

答案 1 :(得分:1)

任何机会,你的测试都会做这样的事情:

from myapp import models

...

models.SideNames = None

因为这是我能想到的唯一解释,为什么SideNames在该方法的上下文中应该是None。

顺便说一句,该方法本身毫无意义,因为Django会自动提供向后关系,因此您只需调用t.sidenames_set.all()