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

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

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

コメントを残す コメントをキャンセル

コメントを残すコメントをキャンセル