62 lines
1.9 KiB
Ruby
62 lines
1.9 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
require 'fast_spec_helper'
|
||
|
|
|
||
|
|
RSpec.describe Gitlab::Pagination::OffsetHeaderBuilder do
|
||
|
|
let(:request) { double(url: 'http://localhost') }
|
||
|
|
let(:request_context) { double(header: nil, params: { per_page: 5 }, request: request) }
|
||
|
|
|
||
|
|
subject do
|
||
|
|
described_class.new(
|
||
|
|
request_context: request_context, per_page: 5, page: 2,
|
||
|
|
next_page: 3, prev_page: 1, total: 10, total_pages: 3
|
||
|
|
)
|
||
|
|
end
|
||
|
|
|
||
|
|
describe '#execute' do
|
||
|
|
let(:basic_links) do
|
||
|
|
%{<http://localhost?page=1&per_page=5>; rel="prev", <http://localhost?page=3&per_page=5>; rel="next", <http://localhost?page=1&per_page=5>; rel="first"}
|
||
|
|
end
|
||
|
|
|
||
|
|
let(:last_link) do
|
||
|
|
%{, <http://localhost?page=3&per_page=5>; rel="last"}
|
||
|
|
end
|
||
|
|
|
||
|
|
def expect_basic_headers
|
||
|
|
expect(request_context).to receive(:header).with('X-Per-Page', '5')
|
||
|
|
expect(request_context).to receive(:header).with('X-Page', '2')
|
||
|
|
expect(request_context).to receive(:header).with('X-Next-Page', '3')
|
||
|
|
expect(request_context).to receive(:header).with('X-Prev-Page', '1')
|
||
|
|
expect(request_context).to receive(:header).with('Link', basic_links + last_link)
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'sets headers to request context' do
|
||
|
|
expect_basic_headers
|
||
|
|
expect(request_context).to receive(:header).with('X-Total', '10')
|
||
|
|
expect(request_context).to receive(:header).with('X-Total-Pages', '3')
|
||
|
|
|
||
|
|
subject.execute
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'exclude total headers' do
|
||
|
|
it 'does not set total headers to request context' do
|
||
|
|
expect_basic_headers
|
||
|
|
expect(request_context).not_to receive(:header)
|
||
|
|
|
||
|
|
subject.execute(exclude_total_headers: true)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'pass data without counts' do
|
||
|
|
let(:last_link) { '' }
|
||
|
|
|
||
|
|
it 'does not set total headers to request context' do
|
||
|
|
expect_basic_headers
|
||
|
|
expect(request_context).not_to receive(:header)
|
||
|
|
|
||
|
|
subject.execute(data_without_counts: true)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|