#782: Global URI store for resolver

Author: nezhar

jsonschema/tests/test_validators.py

@@ -15,6 +15,7 @@
 
 from jsonschema import FormatChecker, TypeChecker, exceptions, validators
 from jsonschema.tests._helpers import bug
+from jsonschema.validators import remove_store_item
 
 
 def startswith(validator, startswith, instance, schema):
@@ -1527,6 +1528,9 @@ def handler(url):
         with resolver.resolving(ref) as resolved:
             self.assertEqual(resolved, schema)
 
+        # Remove test item from store
+        remove_store_item(ref)
+
     def test_cache_remote_on(self):
         response = [object()]
 
@@ -1545,6 +1549,9 @@ def handler(url):
         with resolver.resolving(ref):
             pass
 
+        # Remove test item from store
+        remove_store_item(ref)
+
     def test_cache_remote_off(self):
         response = [object()]
 

jsonschema/validators.py

@@ -496,6 +496,29 @@ def extend(validator, validators=(), version=None, type_checker=None):
 )
 
 _LATEST_VERSION = Draft202012Validator
+_STORE = None
+
+
+def get_store():
+    """
+    Get store or initialize if not None
+    """
+    global _STORE
+    if _STORE is None:
+        _STORE = _utils.URIDict(
+            (id, validator.META_SCHEMA)
+            for id, validator in meta_schemas.items()
+        )
+    return _STORE
+
+
+def remove_store_item(key):
+    """
+    Remove a given key from the store
+    """
+    global _STORE
+    if _STORE is not None:
+        del _STORE[key]
 
 
 class RefResolver(object):
@@ -562,10 +585,7 @@ def __init__(
         self.handlers = dict(handlers)
 
         self._scopes_stack = [base_uri]
-        self.store = _utils.URIDict(
-            (id, validator.META_SCHEMA)
-            for id, validator in meta_schemas.items()
-        )
+        self.store = get_store()
         self.store.update(store)
         self.store[base_uri] = referrer