spec/lib/api/helpers/sse_helpers_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

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe API::Helpers::SSEHelpers do
  include Gitlab::Routing

  let_it_be(:project) { create(:project) }

  subject { Class.new.include(described_class).new }

  describe '#request_from_sse?' do
    before do
      allow(subject).to receive(:request).and_return(request)
    end

    context 'when referer is nil' do
      let(:request) { double(referer: nil)}

      it 'returns false' do
        expect(URI).not_to receive(:parse)
        expect(subject.request_from_sse?(project)).to eq false
      end
    end

    context 'when referer is not from SSE' do
      let(:request) { double(referer: 'https://gitlab.com')}

      it 'returns false' do
        expect(URI).to receive(:parse).and_call_original
        expect(subject.request_from_sse?(project)).to eq false
      end
    end

    context 'when referer is from SSE' do
      let(:request) { double(referer: project_show_sse_path(project, 'master/README.md'))}

      it 'returns true' do
        expect(URI).to receive(:parse).and_call_original
        expect(subject.request_from_sse?(project)).to eq true
      end
    end
  end
end