spec/lib/gitlab/middleware/webhook_recursion_detection_spec.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

# frozen_string_literal: true

require 'fast_spec_helper'
require 'action_dispatch'
require 'rack'
require 'request_store'

RSpec.describe Gitlab::Middleware::WebhookRecursionDetection do
  let(:app) { double(:app) }
  let(:middleware) { described_class.new(app) }
  let(:env) { Rack::MockRequest.env_for("/").merge(headers) }

  around do |example|
    Gitlab::WithRequestStore.with_request_store { example.run }
  end

  describe '#call' do
    subject(:call) { described_class.new(app).call(env) }

    context 'when the recursion detection header is present' do
      let(:new_uuid) { SecureRandom.uuid }
      let(:headers) { { 'HTTP_X_GITLAB_EVENT_UUID' => new_uuid } }

      it 'sets the request UUID from the header' do
        expect(app).to receive(:call)
        expect { call }.to change { Gitlab::WebHooks::RecursionDetection::UUID.instance.request_uuid }.to(new_uuid)
      end
    end

    context 'when recursion headers are not present' do
      let(:headers) { {} }

      it 'works without errors' do
        expect(app).to receive(:call)

        call

        expect(Gitlab::WebHooks::RecursionDetection::UUID.instance.request_uuid).to be_nil
      end
    end
  end
end