From 9c1e81d606a01a586c18dc80b80d63078d65c175 Mon Sep 17 00:00:00 2001
From: Steve Jalim <stevejalim@mozilla.com>
Date: Wed, 24 Jan 2024 17:05:55 +0000
Subject: [PATCH] Fold in code suggestion from PR#199

---
 csp/tests/test_utils.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/csp/tests/test_utils.py b/csp/tests/test_utils.py
index 00760d5..c67e0e6 100644
--- 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,
     )