2015年10月 – ぷすぅ～ぷすぅ～

mock_openを使えばよい。


# mock 関連import 
from mock import Mock, MagicMock, patch, call, mock_open


def test_something(self):
	# open の MagicMockインスタンス
    m = mock_open()

    # openを使う場所を指定する。 somemoduleはテスト対象のモジュール
    # somemodule 内部で呼ばれる open は、 m (mock_open() すなわち MagicMock )に置き換わる
    with patch('somemodule.open', m, create=True):
    	# テストを実行する
    	somemodule.action()

    	# モック化された openの実行
        mock = m()
        # write で、"write something" が呼ばれたかどうか確認
        mock.write.assert_any_call("write something")

# mock 関連import

from mock import Mock, MagicMock, patch, call, mock_open

def test_something(self):

# open の MagicMockインスタンス

m = mock_open()

# openを使う場所を指定する。 somemoduleはテスト対象のモジュール

# somemodule 内部で呼ばれる open は、 m (mock_open() すなわち MagicMock )に置き換わる

with patch('somemodule.open', m, create=True):

# テストを実行する

somemodule.action()

# モック化された openの実行

mock = m()

# write で、"write something" が呼ばれたかどうか確認

mock.write.assert_any_call("write something")

requests.session をモック化して、get と、deleteメソッドをモック化する例

まず置き換える関数を定義して


# 返還するリクエスト
class DummyRequest :
    def __init__(self):
        self.status_code = 200
        self.text = ""

# get の置き換え
def mock_get(url, headers, verify):
    # getの結果を返す
    res = DummyRequest
    res.text = '{"result": true}'
    return res

# deleteの置き換え
def dummy_delete(url, headers, verify):
    # deleteの結果を返す
    res = DummyRequest
    res.text = '{"result": true}'
    return res

# 返還するリクエスト

class DummyRequest :

def __init__(self):

self.status_code = 200

self.text = ""

# get の置き換え

def mock_get(url, headers, verify):

# getの結果を返す

res = DummyRequest

res.text = '{"result": true}'

return res

# deleteの置き換え

def dummy_delete(url, headers, verify):

# deleteの結果を返す

res = DummyRequest

res.text = '{"result": true}'

return res

実際に置き換えるには以下のようにする


def test_something(self):

    with patch('requests.session') as request_mock:
        instance = request_mock.return_value
        instance.get = mock_get
        instance.delete = dummy_delete

def test_something(self):

with patch('requests.session') as request_mock:

instance = request_mock.return_value

instance.get = mock_get

instance.delete = dummy_delete

月: 2015年10月

[python] open をモック化する

[python] mock でインスタンスの複数のメソッドを置き換える

requests.session をモック化して、get と、deleteメソッドをモック化する例