test_http.py 文件源码

def test_get_request_kwargs_url():
    """
    Test that get_request_kwargs can manipulate the url.
    """
    class URLManipulatingAPI(HTTPEater):
        request_cls = Model
        response_cls = Model
        url = 'http://not-the-real-url.com/'

        def get_request_kwargs(self, request_model: Union[Model, None], **kwargs):
            kwargs['url'] = 'http://the-real-url.com'
            return kwargs

    expected_url = 'http://the-real-url.com'

    api = URLManipulatingAPI()

    with requests_mock.Mocker() as mock:
        mock.get(expected_url, json={}, headers=JSON_HEADERS)
        api()
        assert api.url == expected_url