Skip to content

Instantly share code, notes, and snippets.

@YanhaoYang
Last active April 18, 2018 07:20
Show Gist options
  • Save YanhaoYang/0c35ed968e6b0f1ab247fd9eff33eb17 to your computer and use it in GitHub Desktop.
Save YanhaoYang/0c35ed968e6b0f1ab247fd9eff33eb17 to your computer and use it in GitHub Desktop.
Mock function indirectly with pytest
def test_something(mocker):
session = requests.Session()
resp = requests.Response()
resp._content = "response_body"
resp.status_code = 200
resp2 = requests.Response()
resp2._content = '{"results":[]}'.encode('utf-8')
resp2.status_code = 200
mocker.patch.object(session, 'get', side_effect=[resp, resp2])
mocker.patch('requests.Session', return_value=session)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment