spec/controllers/projects/alerting/notifications_controller_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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

# frozen_string_literal: true

require 'spec_helper'

describe Projects::Alerting::NotificationsController do
  let_it_be(:project) { create(:project) }
  let_it_be(:environment) { create(:environment, project: project) }

  describe 'POST #create' do
    let(:service_response) { ServiceResponse.success }
    let(:notify_service) { instance_double(Projects::Alerting::NotifyService, execute: service_response) }

    around do |example|
      ForgeryProtection.with_forgery_protection { example.run }
    end

    before do
      allow(Projects::Alerting::NotifyService).to receive(:new).and_return(notify_service)
    end

    def make_request(body = {})
      post :create, params: project_params, body: body.to_json, as: :json
    end

    context 'when notification service succeeds' do
      let(:payload) do
        {
          title: 'Alert title',
          hosts: 'https://gitlab.com'
        }
      end

      let(:permitted_params) { ActionController::Parameters.new(payload).permit! }

      it 'responds with ok' do
        make_request

        expect(response).to have_gitlab_http_status(:ok)
      end

      it 'does not pass excluded parameters to the notify service' do
        make_request(payload)

        expect(Projects::Alerting::NotifyService)
          .to have_received(:new)
          .with(project, nil, permitted_params)
      end
    end

    context 'when notification service fails' do
      let(:service_response) { ServiceResponse.error(message: 'Unauthorized', http_status: :unauthorized) }

      it 'responds with the service response' do
        make_request

        expect(response).to have_gitlab_http_status(:unauthorized)
      end
    end

    context 'bearer token' do
      context 'when set' do
        it 'extracts bearer token' do
          request.headers['HTTP_AUTHORIZATION'] = 'Bearer some token'

          expect(notify_service).to receive(:execute).with('some token')

          make_request
        end

        it 'pass nil if cannot extract a non-bearer token' do
          request.headers['HTTP_AUTHORIZATION'] = 'some token'

          expect(notify_service).to receive(:execute).with(nil)

          make_request
        end
      end

      context 'when missing' do
        it 'passes nil' do
          expect(notify_service).to receive(:execute).with(nil)

          make_request
        end
      end
    end
  end

  def project_params(opts = {})
    opts.reverse_merge(namespace_id: project.namespace, project_id: project)
  end
end