Python模擬打開文件示例

使用@patch('mymodule.open', new_callable=mock_open)模擬文件操作，避免真實讀寫文件；2. 通過mock_open的read_data參數設置模擬文件內容，如read_data='debug=True'；3. 測試讀取函數時驗證返回結果及open調用參數；4. 測試寫入函數時通過mock_file()獲取句柄並驗證write方法調用；5. 模擬多次不同文件讀取時使用side_effect返回多個mock_open實例；6. 關鍵是patch路徑必須指向使用open的模塊內，而非builtins.open，否則mock無效。

在Python 中使用unittest.mock來模擬文件操作（如open() ）是一個常見的測試技巧，尤其當你想避免真實讀寫文件時。下面是一個實用的示例，展示如何用mock模擬open()函數並測試讀取或寫入文件的函數。

✅ 場景：測試一個讀取配置文件的函數

假設你有一個函數，用於讀取文本文件並返回內容：

 # mymodule.py
def read_config(filename):
    with open(filename, 'r') as f:
        return f.read().strip()

我們不希望測試時真的去讀一個文件，而是用mock模擬open() 。

✅ 使用patch模擬open()

 # test_mymodule.py
from unittest.mock import mock_open, patch
import unittest

# 假設上面的函數在mymodule 模塊中from mymodule import read_config

class TestReadConfig(unittest.TestCase):

    @patch('mymodule.open', new_callable=mock_open, read_data=' debug=True ')
    def test_read_config(self, mock_file):
        result = read_config('config.txt')

        # 驗證結果self.assertEqual(result, 'debug=True')

        # 驗證open 被正確調用mock_file.assert_called_once_with('config.txt', 'r')

? 關鍵點說明

• @patch('mymodule.open', ...) ：
  注意路徑是'mymodule.open' ，因為你要patch 的open是在mymodule模塊中使用的內置函數。不能直接patch 內置的open全局函數，必須patch 使用它的模塊中的open 。

• mock_open(read_data='...') ：
  創建一個模擬的open()返回值， read_data指定文件讀取時返回的內容。

  

• with open(...) as f:被自動模擬， f.read()會返回read_data的內容。

✅ 模擬寫入文件的例子

假設你有一個寫日誌的函數：

 # mymodule.py
def write_log(filename, message):
    with open(filename, 'w') as f:
        f.write(message)

測試寫入操作：

 @patch('mymodule.open', new_callable=mock_open)
def test_write_log(self, mock_file):
    write_log('log.txt', 'Error occurred')

    # 檢查open 是否以'w' 打開mock_file.assert_called_once_with('log.txt', 'w')

    # 檢查write 是否被調用handle = mock_file()
    handle.write.assert_called_once_with('Error occurred')

✅ 小技巧：模擬多次調用不同文件

@patch('mymodule.open', new_callable=mock_open)
def test_multiple_files(self, mock_file):
    mock_file.side_effect = [
        unittest.mock.mock_open(read_data='file1').return_value,
        unittest.mock.mock_open(read_data='file2').return_value,
    ]

    data1 = read_config('a.txt')
    data2 = read_config('b.txt')

    self.assertEqual(data1, 'file1')
    self.assertEqual(data2, 'file2')

✅ 總結：關鍵步驟

• 使用@patch('module.open', new_callable=mock_open) 。
• 用read_data設置模擬文件內容。
• 測試後可驗證open()和write()是否被正確調用。
• 注意路徑要正確（是使用open的模塊，而不是builtins.open ）。

基本上就這些。 mock 文件操作不復雜，但容易因為patch 路徑寫錯而失敗，記得檢查模塊名和導入方式。

以上是Python模擬打開文件示例的詳細內容。更多資訊請關注PHP中文網其他相關文章！