Python：關於構建模擬函數的問題

Question

我正在編寫單元測試來驗證我的項目功能。我需要用模擬函數替換一些函數，我想用Python模擬庫。我使用的實現似乎沒有正常工作，但我不明白我做錯了什麼。這裏的簡化情形：

根/ connector.py

from ftp_utils.py import * 

def main(): 
    config = yaml.safe_load("vendor_sftp.yaml") 
    downloaded_files = [] 

    downloaded_files = get_files(config) 
    for f in downloaded_files: 
     #do something 

根/ utils的/ ftp_utils.py

import os 
import sys 
import pysftp 

def get_files(config): 
    sftp = pysftp.Connection(config['host'], username=config['username']) 
    sftp.chdir(config['remote_dir']) 
    down_files = sftp.listdir() 
    if down_files is not None: 
    for f in down_files: 
     sftp.get(f, os.path.join(config['local_dir'], f), preserve_mtime=True) 

    return down_files 

根/測試/ connector_tester.py

import unittest 
import mock 
import ftp_utils 
import connector 

def get_mock_files(): 
    return ['digital_spend.csv', 'tv_spend.csv'] 

class ConnectorTester(unittest.TestCase) 
    @mock.patch('ftp_utils.get_files', side_effect=get_mock_files) 
    def test_main_process(self, get_mock_files_function): 
    # I want to use a mock version of the get_files function 
    connector.main() 

當我調試我的測試時，我期望在connector.py的main中調用的get_files函數是get_mock_files（），而是ftp_utils.get_files（）。我在這裏做錯了什麼？我應該更改我的代碼以正確調用get_mock_file（）模擬？

感謝， 阿萊西奧

Answer 1

我覺得有幾個問題，您的方案：

• connector.py不能從ftp_utils.py這樣
• 也不能connector_tester.py
• 作爲一種習慣，它是進口更好地讓您的測試文件的形式test_xxx.py
• 使用unittest與修補，見this example

一般情況下，儘量提供工作最少例子所以這是大家來運行你的代碼更容易。

我修改，而大量的例子，使其工作，但基本上，問題是，你修補'ftp_utils.get_files'雖然它不是實際上是所謂的內部connector.main()但可能寧願'connector.get_files'參考。

下面是修改的例子的目錄：

test_connector.py 
ftp_utils.py 
connector.py 

test_connector.py：

import unittest 
import sys 
import mock 
import connector 

def get_mock_files(*args, **kwargs): 
    return ['digital_spend.csv', 'tv_spend.csv'] 

class ConnectorTester(unittest.TestCase): 
    def setUp(self): 
     self.patcher = mock.patch('connector.get_files', side_effect=get_mock_files) 
     self.patcher.start() 

    def test_main_process(self): 
     # I want to use a mock version of the get_files function 
     connector.main() 

suite = unittest.TestLoader().loadTestsFromTestCase(ConnectorTester) 

if __name__ == "__main__": 
    unittest.main() 

NB：運行connector.main()當所謂的'connector.get_files'

connector.py：

from ftp_utils import * 

def main(): 
    config = None 
    downloaded_files = [] 
    downloaded_files = get_files(config) 
    for f in downloaded_files: 
     print(f) 

connector/ftp_utils.py不變。