python – ページ 14 – ぷすぅ～ぷすぅ～

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

関数名・メソッドの呼び出しのアサーションで


with patch('codecs.open') as mock:
    instance = mock.return_value

    expected = [call("arg1", today + "arg2", "arg3")]
    instance.methodname.assert_has_calls(expectedLog, any_order=True)

with patch('codecs.open') as mock:

instance = mock.return_value

expected = [call("arg1", today + "arg2", "arg3")]

instance.methodname.assert_has_calls(expectedLog, any_order=True)

で、正規表現を使いたい場合
引数に処理時間が入っていて、単純に比較できない場合など。

ちょっと無理矢理だが以下のようにすると正規表現が使える


with patch('codecs.open') as mock:
    instance = mock.return_value

    # methodname メソッドの最初の呼びだしを取得
    message = instance.methodname.call_args_list[0]
    # message は、呼び出し時の引数がタプルになって入っている。以下の例は第二引数を正規表現で比べている
    self.assertRegexpMatches(message[1], ".*arg2")

with patch('codecs.open') as mock:

instance = mock.return_value

# methodname メソッドの最初の呼びだしを取得

message = instance.methodname.call_args_list[0]

# message は、呼び出し時の引数がタプルになって入っている。以下の例は第二引数を正規表現で比べている

self.assertRegexpMatches(message[1], ".*arg2")

methodname の二番目の呼び出しを取る場合は,call_args_listの二番目を参照すればよい


message = instance.methodname.call_args_list[1]

message = instance.methodname.call_args_list[1]

カテゴリー: python

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

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

[python] unittest の関数 / メソッド名の呼び出しアサーションで、正規表現を使う