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

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

? 場景：測試一個讀取配置文件的函數(shù)

假設(shè)你有一個函數(shù)，用於讀取文本文件並返回內(nèi)容：

 # 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

# 假設(shè)上面的函數(shù)在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')

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

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

? 關(guān)鍵點說明

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

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

  

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

? 模擬寫入文件的例子

假設(shè)你有一個寫日誌的函數(shù)：

 # 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 是否被調(diào)用handle = mock_file()
    handle.write.assert_called_once_with('Error occurred')

? 小技巧：模擬多次調(diào)用不同文件

@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')

? 總結(jié)：關(guān)鍵步驟

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

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

以上是Python模擬打開文件示例的詳細內(nèi)容。更多資訊請關(guān)注PHP中文網(wǎng)其他相關(guān)文章！