[Python] 在 pytest 裡 monkey patch 系統內建的函式
今天在用 pytest 寫 unit-testing 程式時,遇到一個問題:
要如何 mock 掉 python 原本就提供的內建函式呢?
平常如果是像 urllib.quote() 這種函式,可以用下面的方法來 mock:
def test_mock(monkeypatch): def mock_urllib_quote(s, safe="/"): # Do something... return "mocked result" # Mock set_system_info import urllib monkeypatch.setattr(urllib, "quote", mock_urllib_quote) assert urllib.quote("abc") == "mocked result"
不過,如果我今天想要 mock 的是像 open() 這種內建函式呢?
在用 open() 前,好像不用 import 某個特定模組…
後來在網路上找到了解答,原來 open() 這類的內建函式是屬於 __builtin__ 模組的,
所以只要去 mock __builtin__ 模組就行囉~
def test_mock_open(monkeypatch): # Mock open() import __builtin__ monkeypatch.setattr(__builtin__, "open", mock_open) f = open("test") assert f.read() == "" assert f.fileno() == 0
這邊定義了一個 mock_open(),我讓它傳回一個自己定義的 MockFileObj,
然後在裡面定義我想要的檔案讀寫行為:
class MockFileObj(object): data = "" def __init__(self): pass def __enter__(self): return self def __exit__(self, exit_type, msg, traceback): pass def read(self): return MockFileObj.data def readlines(self, size=0): return [] def write(self, data): MockFileObj.data += data print "write [%s]" % (data) def fileno(self): return 0 def close(self): pass def mock_open(name, *args, **kwargs): return MockFileObj()
參考資料:
pytest: Monkeypatching/mocking modules and environments
Mock python builtin functions in pytest
(本頁面已被瀏覽過 1,318 次)