mock_injection_test2.py

import socket
import inject_exception2
import requests
from mock import patch
import nose.tools
 
# ...snippet
 
def test_api_call():
    with patch.object(requests, "get") as get_mock:
        with nose.tools.assert_raises(socket.error) and nose.tools.assert_raises(inject_exception2.ExceededRetries):
            get_mock.side_effect = socket.error
            inject_exception2.retry(100, call_api, "http://example.com")