Pymongo中的MongoDB Python模擬聚合函數

Question

我想為下面的代碼模擬PyMongo提供的aggregate函數：

client = MongoClient(host="localhost", port=27017,username="Harsha", password="Harsha", authSource="admin")
db_obj = client["DB name"]
mongo_result = db_obj[collection_name].aggregate(pipeline)

我想模擬aggregate函數。

誰能幫我模擬集合函數嗎？

我嘗試了以下代碼段來模擬聚合函數：

試用1：

from pymongo import collection
collection_obj = collection.Collection(client["DB name"], "collection_name")

def mock_get(self, *args):
    return "Result I want"

@mock.patch(collection_obj.Collection.aggregate, side_effect=mock_get)
def test_demo(self):
    .
    .
    .
    .

這不起作用，因為@mock.patch需要字符串完整路徑參數。

所以我也試圖給出完整的aggregation函數路徑

試用2：

class BasicTest(unittest.TestCase):

    def mock_get(self, *args):
        return "Result I want"

    @mock.patch('pymongo.collection.Collection.aggregate', side_effect=mock_get)
    def test_demo(self):
        .
        .
        .
        .

這給了我：

TypeError: test_demo() takes 1 positional argument but 2 were given

Answer 1

patch將模擬對象作為附加參數傳遞給修飾的函數，以便您可以對其進行斷言。 像這樣更改代碼：

@mock.patch('pymongo.collection.Collection.aggregate', side_effect=mock_get)
def test_demo(self, mock_object):
     .
     .