[Python] 在 pytest 裡 monkey patch 系統內建的函式

2015-09-23 ephrain Comments 0 Comment

今天在用 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()