如何在 pytest-mock 中模拟链式调用答案

【问题标题】：How to mock chained calls in pytest-mock如何在 pytest-mock 中模拟链式调用
【发布时间】：2021-11-01 01:31:06
【问题描述】：

我正在尝试使用pytest-mock 来模拟链式函数调用。

@app.route('/application', methods=['PUT'])
    def update_application():
        a = json.loads(request.data)['application']
        application = Application.objects(id=a['id']).first()
        if not application:
            return jsonify({'error': 'data not found'})
        else:
            application.update(jobTitle=a['jobTitle'],
                                companyName=a['companyName'],
                                date=a['date'],
                                status=a['status'

update_application 方法有一个链式函数调用，如下所示

Application.objects(id=a['id']).first()

我尝试使用以下方法模拟链接调用，但无法模拟。

def test_update_application(client, mocker):
    application = Application(id=4, jobTitle='fakeJob12345', companyName='fakeCompany', date=str(datetime.date(2021, 9, 22)))
    mocker.patch(
        'app.Application.update'
    )
    m = mocker.MagicMock()
    m.Application.objects().first.return_value = application
    rv = client.put('/application', json={'application':{
        'id':2, 'jobTitle':'fakeJob12345', 'companyName':'fakeCompany', 'date':str(datetime.date(2021, 9, 23)), 'status':'1'
        }})
    print(rv.data)
    jdata = json.loads(rv.data.decode("utf-8"))["jobTitle"]
    assert jdata == 'fakeJob12345'

我可以像下面那样模拟Application.objects(id=a['id'])，效果很好。

mocker.patch(
        'app.Application.objects',
        return_value = list_application
    )

谁能告诉我如何模拟链式函数？

【问题讨论】：

标签： mocking pytest pytest-mock

【解决方案1】：

您可以这样做 - 我只更改了处理模拟的 3 行代码：

def test_update_application(client, mocker):
    application = Application(id=4, jobTitle='fakeJob12345', companyName='fakeCompany', date=str(datetime.date(2021, 9, 22)))

    mock_objects = mocker.MagicMock(name='objects')
    mocker.patch('app.Application.objects', new=mock_objects)
    mock_objects.return_value.first.return_value = application

    rv = client.put('/application', json={'application':{
        'id':2, 'jobTitle':'fakeJob12345', 'companyName':'fakeCompany', 'date':str(datetime.date(2021, 9, 23)), 'status':'1'
        }})
    print(rv.data)
    jdata = json.loads(rv.data.decode("utf-8"))["jobTitle"]
    assert jdata == 'fakeJob12345'

重要的是要了解您正在模拟 Application 类的 objects 函数，如下所示：mocker.patch('app.Application.objects', new=mock_objects)。

然后，需要设置返回值：mock_objects.return_value.first.return_value = application。

第一部分mock_objects.return_value是匹配Application.objects(id=a['id'])返回的对象，然后你需要追加.first.return_value = application匹配.first()的返回值

【讨论】：