handle resolver without current authority; tests

laws-africa · Jan 20, 2024 · d6b5960 · d6b5960
1 parent 0347667
commit d6b5960
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/peachjam/resolver.py b/peachjam/resolver.py
@@ -82,11 +82,11 @@ class RedirectResolver:
     }
 
     def __init__(self, app_name):
-        self.current_authority = self.RESOLVER_MAPPINGS[app_name.lower()]
+        self.current_authority = self.RESOLVER_MAPPINGS.get(app_name.lower())
 
     def get_domain_for_frbr_uri(self, parsed_frbr_uri):
         best_domain = self.get_best_domain(parsed_frbr_uri)
-        if best_domain != self.current_authority["domain"]:
+        if self.current_authority and best_domain != self.current_authority["domain"]:
             return best_domain
         return None
 

diff --git a/peachjam/tests/test_resolver.py b/peachjam/tests/test_resolver.py
@@ -1,7 +1,7 @@
 from cobalt import FrbrUri
 from django.test import TestCase
 
-from peachjam.views.documents import RedirectResolver
+from peachjam.resolver import RedirectResolver
 
 urls = [
     "/akn/zm/judgment/zmsc/2021/7/eng@2021-01-19",