如何使用unittest將返回值設置為從另一個模擬對象創建的模擬對象？

Question

from connection_util import some_engine

class RecommendConversationsUsingOldAPI(BaseResource):

    def check_entitlement(self,msg_ids,usr_id, metadata):
            conn = some_engine.connect()
            some_list = conn.execute(something)

單元測試

@patch('module1.some_engine')  
def test_some_method(self, mockEngine):
        # some code.....

我可以嘲笑some_engine

@patch('module1.some_engine')

如何為conn.execute()設置return_value ？

Answer 1

可以為connect()返回手動創建的Mock對象

 mock_connection = Mock()
 mock_connection.execute.return_value = [1,2,3,4]

 mockEngine.connect.return_value = mock_connection