diff options
Diffstat (limited to 'spec/support/helpers/content_security_policy_helpers.rb')
-rw-r--r-- | spec/support/helpers/content_security_policy_helpers.rb | 20 |
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 |