Fix jinja2 import after version upgrade

hypothesis · Jan 25, 2024 · a00811c · a00811c
1 parent 8c1f5e9
commit a00811c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/bouncer/util.py b/bouncer/util.py
@@ -1,6 +1,6 @@
 from urllib import parse
 
-import jinja2
+from markupsafe import Markup
 from pyramid import i18n
 
 _ = i18n.TranslationStringFactory(__package__)
@@ -148,8 +148,8 @@ def get_pretty_url(url):
         return None
 
     pretty_url = parsed_url.netloc[:NETLOC_MAX_LENGTH]
-    if len(parsed_url.netloc) > NETLOC_MAX_LENGTH:  # pragma: nocover
-        pretty_url += jinja2.Markup("&hellip;")
+    if len(parsed_url.netloc) > NETLOC_MAX_LENGTH:
+        pretty_url += Markup("&hellip;")
     return pretty_url
 
 

diff --git a/tests/unit/bouncer/util_test.py b/tests/unit/bouncer/util_test.py
@@ -110,6 +110,12 @@ def test_parse_document_returns_authority(es_annotation_doc):
     assert authority == "hypothes.is"
 
 
+def test_get_pretty_url_for_long_url():
+    long_netloc = "https://www.verylongdomainthatkeepsgoingandgoing.com"
+
+    assert "www.verylongdomainthatkeepsgoi&hellip;" == util.get_pretty_url(long_netloc)
+
+
 @pytest.fixture
 def es_annotation_doc():
     """