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