Add latest changes from gitlab-org/gitlab@master

1 files changed, 20 insertions, 0 deletions

diff --git a/spec/support/helpers/content_security_policy_helpers.rb b/spec/support/helpers/content_security_policy_helpers.rb
new file mode 100644
index 00000000000..c9f15e65c74
--- /dev/null
+++ b/spec/support/helpers/content_security_policy_helpers.rb
@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module ContentSecurityPolicyHelpers
+  # Expecting 2 calls to current_content_security_policy by default, once for
+  # the call that's being tested and once for the call in ApplicationController
+  def setup_csp_for_controller(controller_class, times = 2)
+    expect_next_instance_of(controller_class) do |controller|
+      expect(controller).to receive(:current_content_security_policy)
+                              .and_return(ActionDispatch::ContentSecurityPolicy.new).exactly(times).times
+    end
+  end
+
+  # Expecting 2 calls to current_content_security_policy by default, once for
+  # the call that's being tested and once for the call in ApplicationController
+  def setup_existing_csp_for_controller(controller_class, csp, times = 2)
+    expect_next_instance_of(controller_class) do |controller|
+      expect(controller).to receive(:current_content_security_policy).and_return(csp).exactly(times).times
+    end
+  end
+end