Move requests import to runtime

jsonschema/tests/test_validators.py

@@ -1013,6 +1013,26 @@ def test_schema_error_message(self):
         )
 
 
+class MockImport(object):
+
+    def __init__(self, module, _mock):
+        self._module = module
+        self._mock = _mock
+        self._orig_import = None
+
+    def __enter__(self):
+        self._orig_import = sys.modules.get(self._module, None)
+        sys.modules[self._module] = self._mock
+        return self._mock
+
+    def __exit__(self, *args):
+        if self._orig_import is None:
+            del sys.modules[self._module]
+        else:
+            sys.modules[self._module] = self._orig_import
+        return True
+
+
 class TestRefResolver(TestCase):
 
     base_uri = ""
@@ -1062,7 +1082,7 @@ def test_it_retrieves_unstored_refs_via_requests(self):
         ref = "http://bar#baz"
         schema = {"baz": 12}
 
-        with mock.patch("jsonschema.validators.requests") as requests:
+        with MockImport("requests", mock.Mock()) as requests:
             requests.get.return_value.json.return_value = schema
             with self.resolver.resolving(ref) as resolved:
                 self.assertEqual(resolved, 12)
@@ -1072,7 +1092,7 @@ def test_it_retrieves_unstored_refs_via_urlopen(self):
         ref = "http://bar#baz"
         schema = {"baz": 12}
 
-        with mock.patch("jsonschema.validators.requests", None):
+        with MockImport("requests", None):
             with mock.patch("jsonschema.validators.urlopen") as urlopen:
                 urlopen.return_value.read.return_value = (
                     json.dumps(schema).encode("utf8"))

jsonschema/validators.py

@@ -7,11 +7,6 @@
 
 from six import add_metaclass
 
-try:
-    import requests
-except ImportError:
-    requests = None
-
 from jsonschema import _utils, _validators, _types
 from jsonschema.compat import (
     Sequence, urljoin, urlsplit, urldefrag, unquote, urlopen,
@@ -671,6 +666,10 @@ def resolve_remote(self, uri):
         .. _requests: http://pypi.python.org/pypi/requests/
 
         """
+        try:
+            import requests
+        except ImportError:
+            requests = None
 
         scheme = urlsplit(uri).scheme