diff --git a/stackinabox/util/requests_mock/core.py b/stackinabox/util/requests_mock/core.py index 0dd9208..d49c706 100644 --- a/stackinabox/util/requests_mock/core.py +++ b/stackinabox/util/requests_mock/core.py @@ -17,7 +17,6 @@ from requests.adapters import HTTPAdapter from urllib3.response import HTTPResponse import requests_mock -import requests_mock.compat import requests_mock.response import six @@ -25,6 +24,11 @@ from stackinabox.util import deprecator from stackinabox.util.requests_mock import reqcallable +try: + import requests_mock.compat +except ImportError: + pass + logger = logging.getLogger(__name__) @@ -52,6 +56,9 @@ def session_registration(uri, session): reqcallable.RequestMockCallable(uri) ) + if not uri.endswith('/'): + uri += '/' + # Tell the session about the adapter and the URI session.mount('http://{0}'.format(uri), StackInABox.hold_out('adapter')) session.mount('https://{0}'.format(uri), StackInABox.hold_out('adapter'))