python mox使用方法(2)

  上文讲了如何架空一个普通的函数，但是在python代码中，大部分都是以类的形式存在的，那现在就开始讲如何架空一个类。

  首先，我们要声明一个叫做Zoo的类，并把之前的函数移入这个类中。

 class Zoo(object):
    def __init__(self):
        self.animals = ['penguin', 'swan', 'tiger', 'monkey']

    def get_most_popular_animal(self):
        month = self.get_month()
        if month in [11,12,1]:
            return 'penguin'
        elif month in [6,7,8]:
            return 'swan'
        else:
            return 'tiger'

    def get_month(self):
        today = date.today()
        return today.month

  那现在要测试函数get_most_popular_animal的时候，就需要对类之内的函数get_month进行处理了。

    def test_get_most_popular_animal(self):
        z = zoo.Zoo()
        self.m.StubOutWithMock(z, 'get_month')
        z.get_month().AndReturn(10)

        self.m.ReplayAll()
        self.assertEqual('tiger', z.get_most_popular_animal())
        self.m.VerifyAll()
      

  这里可以看出，我们需要对产生的对象进行mock，修改相应的函数来达成目的。

  还有另一种情况是,在某个函数执行的过程中,会调用其他对象的方法。这时，就需要对这个对象做处理了。

  现在修改下Zoo函数，引入一个管理员的脚色。

class Zoo(object):
    def __init__(self, zookeeper):
        self.animals = ['penguin', 'swan', 'tiger', 'monkey']
        self.zookeeper = zookeeper


    def feed(self):
        for animal in self.animals:
            self.zookeeper.feed(animal)
        return True


class ZooKeeper(object):
    def __init__(self, name):
        self.name = name

    def feed(self, animal):
        print "feed animal %s" % animal

   并添加一个简单的函数feed()，饲养员喂养每种动物，主函数永远返回True。表示成功的喂养。

   那现在写一个测试函数来测试feed，这里zookeeper的feed函数其实每次返回的都是None，并不会影响主测试函数。现在只需要将这个函数给打个桩即可保证主函数运行的成 功了。

    def test_feed(self):
        zookeeper = self.m.CreateMock(zoo.ZooKeeper)
        zookeeper.feed(mox.IgnoreArg())
        zookeeper.feed(mox.IgnoreArg())
        zookeeper.feed(mox.IgnoreArg())
        zookeeper.feed(mox.IgnoreArg())

        z = zoo.Zoo(zookeeper)

        self.m.ReplayAll()
        self.assertEqual(True, z.feed())
        self.m.VerifyAll()

  测试用例中，首先为zookeeper这个对象创建mock，然后声明feed函数的行为，因为函数不需要返回值，所以这里也就不需要使用AndReturn。

  mox.IgnoreArg()表示忽略传入参数检查。如果需要，则可以使用IsA，StrContains,In, ContainsKeyValue, IsAlmost等等函数。如果这里将

    zookeeper.feed(mox.IgnoreArg())

  替换为

    zookeeper.feed(mox.IsA(dict))

  就会报错：

UnexpectedMethodCallError: Unexpected method call.  unexpected:-  expected:+
- ZooKeeper.feed('monkey') -> None
?                 ^^^^^^

+ ZooKeeper.feed(<type 'dict'>) -> None
?                ++++++ ^^^^ +