python unittest 之mock学习笔记(续二)

假设有文件class1和class2，class2类中的函数调用了class1类中的函数。利用mock方法生成class1类中函数的一个fake返回值，在此基础上，对class2类中的函数进行单元测试。
代码如下：
class1文件

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# class1.py
class Class1():
    def flag(self):
        return 1

classs2文件

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#  class2.py
from class1 import Class1
class Class2(object):
    def adds(self,a,b,c):
        if Class1.flag():
            return (a+b+c)
        else:
            return (a+b)

对class2利用mock测试文件：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# ClassStudy.py

from class2 import Class2

import mock
import unittest

class Class2TestCase(unittest.TestCase):

    @mock.patch('class2.Class1')
    def test_adds(self, mock_class1):
        # 实例化Class2
        c2 = Class2()
        # 设置class2类中的函数调用class1类中的flag为0
        # 则调用class2类中的函数adds返回a+b
        print ("设置class2类中的函数调用class1类中的flag为0")
        print ("则调用class2类中的函数adds返回a+b")
        mock_class1.flag.return_value = 0
        num = c2.adds(1,2,3)
        self.assertEqual(num,3)
        # 设置class2类中的函数调用class1类中的flag为1
        # 则调用class2类中的函数adds返回a+b+c
        print ("设置class2类中的函数调用class1类中的flag为1")
        print ("则调用class2类中的函数adds返回a+b+c")
        mock_class1.flag.return_value = 1
        num = c2.adds(1, 2, 3)
        self.assertEqual(num, 6)