fetch extra schema items locally

reconcile/tests/conftest.py

@@ -1,7 +1,10 @@
 import json
 import os
+import re
 import warnings
 
+from referencing.jsonschema import DRAFT7
+
 SCHEMA_DIR = os.path.join(os.path.dirname(__file__), "specs")
 SUPPORTED_API_VERSIONS = ["0.2"]
 
@@ -23,3 +26,21 @@ def get_schema(
             msg = f"Schema file {schema_path} not found"
             warnings.warn(msg)
     return schemas
+
+
+# set up jsonschema registry
+def retrieve_schema_from_filesystem(uri: str):
+    recon_schema = re.match(
+        r"https://reconciliation-api\.github\.io/specs/(.*)/schemas/(.*\.json)",
+        uri,
+    )
+    if recon_schema:
+        schema_version = recon_schema.group(1)
+        schema_file = recon_schema.group(2)
+        return DRAFT7.create_resource(
+            get_schema(schema_file, supported_api_versions=[schema_version])[
+                schema_version
+            ]
+        )
+
+    raise ValueError(f"Unknown URI {uri}")

reconcile/tests/test_company_api.py

@@ -3,10 +3,11 @@
 import os
 
 import jsonschema
+from referencing import Registry
 
 from ftc.tests import TestCase
 
-from .conftest import get_schema
+from .conftest import get_schema, retrieve_schema_from_filesystem
 
 logger = logging.getLogger(__name__)
 
@@ -22,6 +23,10 @@
 
 
 class TestCompanyReconcileAPI(TestCase):
+    def setUp(self):
+        super().setUp()
+        self.registry = Registry(retrieve=retrieve_schema_from_filesystem)
+
     # GET request to /api/v1/reconcile/company should return the service spec
     def test_get_company_service_spec(self):
         for schema_version, schema in get_schema("manifest.json").items():
@@ -47,6 +52,7 @@ def test_get_company_service_spec(self):
                     instance=data,
                     schema=schema,
                     cls=jsonschema.Draft7Validator,
+                    registry=self.registry,
                 )
 
     # POST request to /api/v1/reconcile/company should return a list of candidates
@@ -74,6 +80,7 @@ def test_company_reconcile(self):
                     instance=data,
                     schema=schema,
                     cls=jsonschema.Draft7Validator,
+                    registry=self.registry,
                 )
 
     # POST request to /api/v1/reconcile should return a list of candidates
@@ -100,4 +107,5 @@ def test_reconcile_empty(self):
                     instance=data,
                     schema=schema,
                     cls=jsonschema.Draft7Validator,
+                    registry=self.registry,
                 )

reconcile/tests/test_reconcile_all_api.py

@@ -4,10 +4,11 @@
 from typing import Tuple
 
 import jsonschema
+from referencing import Registry
 
 from ftc.tests import TestCase
 
-from .conftest import get_schema
+from .conftest import get_schema, retrieve_schema_from_filesystem
 
 logger = logging.getLogger(__name__)
 
@@ -44,6 +45,10 @@ def get_test_cases(schema_file):
 
 
 class TestReconcileAllAPI(TestCase):
+    def setUp(self):
+        super().setUp()
+        self.registry = Registry(retrieve=retrieve_schema_from_filesystem)
+
     # GET request to /api/v1/reconcile should return the service spec
     def test_get_service_spec(self):
         for base_url, schema_version, schema in get_test_cases("manifest.json"):
@@ -62,6 +67,7 @@ def test_get_service_spec(self):
                     instance=data,
                     schema=schema,
                     cls=jsonschema.Draft7Validator,
+                    registry=self.registry,
                 )
 
     # POST request to /api/v1/reconcile should return a list of candidates
@@ -89,6 +95,7 @@ def test_reconcile(self):
                     instance=data,
                     schema=schema,
                     cls=jsonschema.Draft7Validator,
+                    registry=self.registry,
                 )
 
     # POST request to /api/v1/reconcile should return a list of candidates
@@ -115,6 +122,7 @@ def test_reconcile_empty(self):
                     instance=data,
                     schema=schema,
                     cls=jsonschema.Draft7Validator,
+                    registry=self.registry,
                 )
 
     # POST request to /api/v1/reconcile/suggest/entity should return a list of candidates
@@ -149,6 +157,7 @@ def test_reconcile_suggest_entity(self):
                     instance=data,
                     schema=schema,
                     cls=jsonschema.Draft7Validator,
+                    registry=self.registry,
                 )
 
     # POST request to /api/v1/reconcile/suggest/entity should return a list of candidates
@@ -194,6 +203,7 @@ def test_reconcile_extend(self):
                     instance=data,
                     schema=schema,
                     cls=jsonschema.Draft7Validator,
+                    registry=self.registry,
                 )
 
     # GET request to /api/v1/reconcile/suggest/entity should return a list of candidates
@@ -228,4 +238,5 @@ def test_reconcile_propose_properties(self):
                     instance=data,
                     schema=schema,
                     cls=jsonschema.Draft7Validator,
+                    registry=self.registry,
                 )