pytest mock的使用

简单使用return_value返回固定的数据

• 方法一，使用装饰器。

import mock
def add(a, b):
    return a + b


# 使用mock的return_value参数改变add返回的数据
@mock.patch('add', return_value=10)
def test_add():
    print(add(1, 2))  # 10

• 方法二，使用mock.path方法。

import mock
def test_add():
    with mock.patch('add', return_value=10):
        print(add(1, 2)) # 10

使用side_effect返回可变的数据

• 当一个方法被同一个测试调用多次，而自己又不想得到同一个返回值，就轮到side_effect上场了。
• 依次返回list里的数据。

# 方法一
@mock.patch('add', side_effect=[10, 20])
def test_add():
    print(add(1, 2))  # 10
    print(add(1, 2))  # 20

# 方法二
def test_add():
    with mock.patch('add', side_effect=[10, 20]):
        print(add(1, 2))  # 10
        print(add(1, 2))  # 20

• 根据传入的参数返回指定的数据。

# 方法一
@mock.patch('add', side_effect={(1, 2): 10, (2, 3): 20})
def test_add():
    print(add(2, 3))  # 20
    print(add(1, 2))  # 10

# 方法二
def test_add():
    with mock.patch('add', side_effect={(1, 2): 10, (2, 3): 20}):
        print(add(2, 3))  # 20
        print(add(1, 2))  # 10

---