summaryrefslogtreecommitdiff
path: root/spec/lib/bitbucket_server/paginator_spec.rb
blob: 66bf5bf3f0252236752f0ff27f0f489a28e65cf3 (plain)
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
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe BitbucketServer::Paginator do
  let(:last_page) { double(:page, next?: false, items: ['item_2']) }
  let(:first_page) { double(:page, next?: true, next: last_page, items: ['item_1']) }
  let(:connection) { instance_double(BitbucketServer::Connection) }

  describe '#items' do
    let(:paginator) { described_class.new(connection, 'http://more-data', :pull_request) }
    let(:page_attrs) { { 'isLastPage' => false, 'nextPageStart' => 1 } }

    it 'returns items and raises StopIteration in the end' do
      allow(paginator).to receive(:fetch_next_page).and_return(first_page)
      expect(paginator.items).to match(['item_1'])

      allow(paginator).to receive(:fetch_next_page).and_return(last_page)
      expect(paginator.items).to match(['item_2'])

      allow(paginator).to receive(:fetch_next_page).and_return(nil)
      expect { paginator.items }.to raise_error(StopIteration)
    end

    it 'obeys limits' do
      limited = described_class.new(connection, 'http://more-data', :pull_request, page_offset: 0, limit: 1)
      allow(limited).to receive(:fetch_next_page).and_return(first_page)

      expect(limited.has_next_page?).to be_truthy
      expect(limited.items).to match(['item_1'])
      expect(limited.has_next_page?).to be_truthy
      expect { limited.items }.to raise_error(StopIteration)
    end

    it 'does not stop if limit is unspecified' do
      stub_const("BitbucketServer::Paginator::PAGE_LENGTH", 1)
      paginator = described_class.new(connection, 'http://more-data', :pull_request, page_offset: 0, limit: nil)
      allow(paginator).to receive(:fetch_next_page).and_return(first_page, last_page)

      expect(paginator.has_next_page?).to be_truthy
      expect(paginator.items).to match(['item_1'])
      expect(paginator.has_next_page?).to be_truthy
      expect(paginator.items).to match(['item_2'])
    end

    it 'calls the connection with different offsets' do
      expect(connection).to receive(:get).with('http://more-data', start: 0, limit: BitbucketServer::Paginator::PAGE_LENGTH).and_return(page_attrs)

      expect(paginator.items).to eq([])

      expect(connection).to receive(:get).with('http://more-data', start: 1, limit: BitbucketServer::Paginator::PAGE_LENGTH).and_return({})

      expect(paginator.items).to eq([])

      expect { paginator.items }.to raise_error(StopIteration)
    end
  end
end