Fold in code suggestion from PR#199

mozilla · Jan 24, 2024 · 9c1e81d · 9c1e81d
1 parent b00c829
commit 9c1e81d
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/csp/tests/test_utils.py b/csp/tests/test_utils.py
@@ -22,7 +22,7 @@ def literal(s):
     return s
 
 
-lazy_literal = lazy(literal)
+lazy_literal = lazy(literal, str)
 
 
 @override_settings(CSP_DEFAULT_SRC=["example.com", "example2.com"])
@@ -247,11 +247,14 @@ def test_require_trusted_types_for():
     policy_eq("default-src 'self'; require-trusted-types-for 'script'", policy)
 
 
-@override_settings(CSP_TRUSTED_TYPES=["strictPolicy", "laxPolicy", "default", "'allow-duplicates'"])
+@override_settings(
+    CSP_TRUSTED_TYPES=["strictPolicy", "laxPolicy", "default", "'allow-duplicates'"]
+)
 def test_trusted_types():
     policy = build_policy()
     policy_eq(
-        "default-src 'self'; trusted-types strictPolicy laxPolicy " + "default 'allow-duplicates'",
+        "default-src 'self'; trusted-types strictPolicy laxPolicy "
+        + "default 'allow-duplicates'",
         policy,
     )
 
@@ -277,7 +280,11 @@ def test_nonce():
 def test_nonce_include_in():
     policy = build_policy(nonce="abc123")
     policy_eq(
-        ("default-src 'self'; " "script-src 'nonce-abc123'; " "style-src 'nonce-abc123'"),
+        (
+            "default-src 'self'; "
+            "script-src 'nonce-abc123'; "
+            "style-src 'nonce-abc123'"
+        ),
         policy,
     )