Added with_nonce method [skip ci]

ankane · Feb 3, 2022 · fe74fd1 · fe74fd1
1 parent b3f72de
commit fe74fd1
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/test/helper_test.rb b/test/helper_test.rb
@@ -12,13 +12,14 @@ def test_spec
   end
 
   def test_nonce_chart
-    stub(:content_security_policy_nonce, "test123") do
+    with_nonce do
       assert_match '<script nonce="test123">', vega_chart(Vega.lite)
     end
+    refute_match "nonce", vega_chart(Vega.lite, nonce: false)
   end
 
   def test_nonce_spec
-    stub(:content_security_policy_nonce, "test123") do
+    with_nonce do
       assert_match '<script nonce="test123">', vega_chart(Vega.lite.spec)
     end
   end
@@ -35,6 +36,12 @@ def test_bad_type
     assert_equal "expected Vega chart or spec", error.message
   end
 
+  def with_nonce
+    stub(:content_security_policy_nonce, "test123") do
+      yield
+    end
+  end
+
   # for stubbing
   def content_security_policy_nonce
     nil