如何模拟Pytest和MagicMock中sqlalchemy行的MagicMock()方法？

Question

下面是我的实际代码

    results = session.execute(query)
    for row in results:
        final.append(row._asdict())
    return final

我能够模拟session.execute来返回元组列表。但是在代码中，当调用_asdict()时，我得到一个命令，表示tuple为no属性_asdict()。下面是我用来模拟pytest的代码。我想知道如何在这里模拟_asdict()。

mock_session.execute.return_value = [tup1,
                                     tup2]

Answer 1

最后，我找到了一种使用CursorResult()来模拟它的方法，我正在模拟session.execute()的返回类型，即，如下面所示的session.execute()对象。

CursorResult = MagicMock()
mock_session.execute.return_value = [CursorResult(tup1),
                                     CursorResult(tup2)]
CursorResult.return_value._asdict.side_effect = [tup1, tup2]

Answer 2

对注释的后续操作:您可以创建一个具有_asdict方法的小类，并将其用作sqlalchemy行的替补，例如：

class FakeRow:
    def _asdict(self):
        # Put whatever test data you want in the dict
        return { 'some_data': 1, 'some_more_data': 2 }

那么你的模拟应该是这样的：

mock_session.execute.return_value = [FakeRow(), FakeRow()]