摘要：一些常用的示例先簡(jiǎn)單定義個(gè)類，方便舉例這個(gè)類里有兩個(gè)成員方法，一個(gè)有參數(shù)，一個(gè)無(wú)參數(shù)。有個(gè)模塊方法，返回?cái)?shù)據(jù)庫(kù)中所有指定的人員，并按排序掉整個(gè)數(shù)據(jù)庫(kù)訪問(wèn)先得到一個(gè)的對(duì)象，再在中設(shè)置一個(gè)對(duì)象，此時(shí)不需要自己再創(chuàng)建

先簡(jiǎn)單定義個(gè)類，方便舉例：

class Person:
    def __init__(self):
        self.__age = 10
        
    def get_fullname(self, first_name, last_name):
        return first_name + " " + last_name
        
    def get_age(self):
        return self.__age
        
    @staticmethod
    def get_class_name():
        return Person.__name__

這個(gè)類里有兩個(gè)成員方法，一個(gè)有參數(shù)，一個(gè)無(wú)參數(shù)。還有一個(gè)靜態(tài)方法

1. 使用Mock類，返回固定值

class PersonTest(TestCase):
    def test_should_get_age(self):
        p = Person()
        
        # 不mock時(shí)，get_age應(yīng)該返回10
        self.assertEqual(p.get_age(), 10)
        
        # mock掉get_age方法，讓它返回20
        p.get_age = Mock(return_value=20)
        self.assertEqual(p.get_age(), 20)
    
    def test_should_get_fullname(self):
        p = Person()
        
        # mock掉get_fullname，讓它返回"James Harden"
        p.get_fullname = Mock(return_value="James Harden")
        self.assertEqual(p.get_fullname(), "James Harden")

2. 校驗(yàn)參數(shù)個(gè)數(shù)，再返回固定值

上面的例子你也許已經(jīng)注意到了，調(diào)用p.get_fullname時(shí)沒(méi)有給任何的參數(shù)，但是依然可以工作。
如果想校驗(yàn)參數(shù)需要用create_autospec模塊方法替代Mock類。

class PersonTest(TestCase):
    def test_should_get_fullname(self):
        p = Person()
        
        p.get_fullname = create_autospec(p.get_fullname, return_value="James Harden")
        
        # 隨便給兩個(gè)參數(shù)，依然會(huì)返回mock的值
        self.assertEqual(p.get_fullname("1", "2"), "James Harden") 
        
        # 如果參數(shù)個(gè)數(shù)不對(duì)，會(huì)報(bào)錯(cuò)TypeError: missing a required argument: "last_name"
        p.get_fullname("1")

3. 使用side_effect, 依次返回指定值

class PersonTest(TestCase):
    def test_should_get_age(self):
        p = Person()
        
        p.get_age = Mock(side_effect=[10, 11, 12])

        self.assertEqual(p.get_age(), 10)
        self.assertEqual(p.get_age(), 11)
        self.assertEqual(p.get_age(), 12)

4. 根據(jù)參數(shù)不同，返回不同的值

class PersonTest(TestCase):
    def test_should_get_fullname(self):
        p = Person()
        
        values = {("James", "Harden"): "James Harden", ("Tracy", "Grady"): "Tracy Grady"}
        p.get_fullname = Mock(side_effect=lambda x, y: values[(x, y)])
        
        self.assertEqual(p.get_fullname("James", "Harden"), "James Harden")
        self.assertEqual(p.get_fullname("Tracy", "Grady"), "Tracy Grady")

5. 拋出異常

class PersonTest(TestCase):
    def test_should_raise_exception(self):
        p = Person()
        
        p.get_age = Mock(side_effect=TypeError("integer type"))
        # 只要調(diào)就會(huì)拋出異常
        self.assertRaises(TypeError, p.get_age)

6. 檢驗(yàn)是否調(diào)用

class PersonTest(TestCase):
    def test_should_validate_method_calling(self):
        p.get_fullname = Mock(return_value="James Harden")

        # 沒(méi)調(diào)用過(guò)
        p.get_fullname.assert_not_called()  # Python 3.5

        p.get_fullname("1", "2")

        # 調(diào)用過(guò)任意次數(shù)
        p.get_fullname.assert_called() # Python 3.6
        # 只調(diào)用過(guò)一次, 不管參數(shù)
        p.get_fullname.assert_called_once() # Python 3.6
        # 只調(diào)用過(guò)一次，并且符合指定的參數(shù)
        p.get_fullname.assert_called_once_with("1", "2")

        p.get_fullname("3", "4")
        # 只要調(diào)用過(guò)即可，必須指定參數(shù)
        p.get_fullname.assert_any_call("1", "2")

        # 重置mock，重置之后相當(dāng)于沒(méi)有調(diào)用過(guò)
        p.get_fullname.reset_mock()
        p.get_fullname.assert_not_called()
        
        # Mock對(duì)象里除了return_value, side_effect屬性外，
        # called表示是否調(diào)用過(guò)，call_count可以返回調(diào)用的次數(shù)
        self.assertEqual(p.get_fullname.called, False)
        self.assertEqual(p.get_fullname.call_count, 0)
        
        p.get_fullname("1", "2")
        p.get_fullname("3", "4")
        self.assertEqual(p.get_fullname.called, True)
        self.assertEqual(p.get_fullname.call_count, 2)

靜態(tài)方法和模塊方法需要使用patch來(lái)mock。

1. 在測(cè)試方法參數(shù)中得到Mock對(duì)象

class PersonTest(TestCase):
    # 以字符串的形式列出靜態(tài)方法的路徑，在測(cè)試的參數(shù)里會(huì)自動(dòng)得到一個(gè)Mock對(duì)象
    @patch("your.package.module.Person.get_class_name")
    def test_should_get_class_name(self, mock_get_class_name):
        mock_get_class_name.return_value = "Guy"

        self.assertEqual(Person.get_class_name(), "Guy")

2. 在patch中設(shè)置Mock對(duì)象

class PersonTest(TestCase):
    mock_get_class_name = Mock(return_value="Guy")

    # 在patch中給出定義好的Mock的對(duì)象，好處是定義好的對(duì)象可以復(fù)用
    @patch("your.package.module.Person.get_class_name", mock_get_class_name)
    def test_should_get_class_name(self):
        self.assertEqual(Person.get_class_name(), "Guy")

3. 使用patch.object

class PersonTest(TestCase):
    mock_get_class_name = Mock(return_value="Guy")

    # 使用patch.object來(lái)mock，好處是Person類不是以字符串形式給出的
    @patch.object(Person, "get_class_name", mock_get_class_name)
    def test_should_get_class_name(self, ):
        self.assertEqual(Person.get_class_name(), "Guy")

4. 使用with控制作用域

class PersonTest(TestCase):
    # 作用域之外，依然返回真實(shí)值
    def test_should_get_class_name(self, ):
        mock_get_class_name = Mock(return_value="Guy")
        with patch("your.package.module.Person.get_class_name", mock_get_class_name):
            self.assertEqual(Person.get_class_name(), "Guy")

        self.assertEqual(Person.get_class_name(), "Person")

在django里，我們經(jīng)常需要mock數(shù)據(jù)庫(kù)，而訪問(wèn)數(shù)據(jù)庫(kù)時(shí)經(jīng)常是鏈?zhǔn)秸{(diào)用，看個(gè)例子。

def get_person(name):
    return Person.objects.filter(name=name).order_by("age")

有個(gè)模塊方法，返回?cái)?shù)據(jù)庫(kù)中所有指定name的人員，并按age排序

mock掉整個(gè)數(shù)據(jù)庫(kù)訪問(wèn)

@patch("your.package.module.Person.objects.filter")
def test_should_get_person(self, mock_filter):
    # 先得到一個(gè)filter的Mock對(duì)象，再在return_value中設(shè)置一個(gè)Mock對(duì)象，此時(shí)不需要自己再創(chuàng)建
    mock_filter.return_value.order_by.return_value = None
    
    self.assertIsNone(get_person())