69 lines
2.1 KiB
Ruby
69 lines
2.1 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
require 'spec_helper'
|
||
|
|
|
||
|
|
describe Gitlab::Pagination::Keyset::Pager do
|
||
|
|
let(:relation) { Project.all.order(id: :asc) }
|
||
|
|
let(:request) { double('request', page: page, apply_headers: nil) }
|
||
|
|
let(:page) { Gitlab::Pagination::Keyset::Page.new(order_by: { id: :asc }, per_page: 3) }
|
||
|
|
let(:next_page) { double('next page') }
|
||
|
|
|
||
|
|
before_all do
|
||
|
|
create_list(:project, 7)
|
||
|
|
end
|
||
|
|
|
||
|
|
describe '#paginate' do
|
||
|
|
subject { described_class.new(request).paginate(relation) }
|
||
|
|
|
||
|
|
it 'loads the result relation only once' do
|
||
|
|
expect do
|
||
|
|
subject
|
||
|
|
end.not_to exceed_query_limit(1)
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'passes information about next page to request' do
|
||
|
|
lower_bounds = relation.limit(page.per_page).last.slice(:id)
|
||
|
|
expect(page).to receive(:next).with(lower_bounds, false).and_return(next_page)
|
||
|
|
expect(request).to receive(:apply_headers).with(next_page)
|
||
|
|
|
||
|
|
subject
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'when retrieving the last page' do
|
||
|
|
let(:relation) { Project.where('id > ?', Project.maximum(:id) - page.per_page).order(id: :asc) }
|
||
|
|
|
||
|
|
it 'indicates this is the last page' do
|
||
|
|
expect(request).to receive(:apply_headers) do |next_page|
|
||
|
|
expect(next_page.end_reached?).to be_truthy
|
||
|
|
end
|
||
|
|
|
||
|
|
subject
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'when retrieving an empty page' do
|
||
|
|
let(:relation) { Project.where('id > ?', Project.maximum(:id) + 1).order(id: :asc) }
|
||
|
|
|
||
|
|
it 'indicates this is the last page' do
|
||
|
|
expect(request).to receive(:apply_headers) do |next_page|
|
||
|
|
expect(next_page.end_reached?).to be_truthy
|
||
|
|
end
|
||
|
|
|
||
|
|
subject
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'returns an array with the loaded records' do
|
||
|
|
expect(subject).to eq(relation.limit(page.per_page).to_a)
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'validating the order clause' do
|
||
|
|
let(:page) { Gitlab::Pagination::Keyset::Page.new(order_by: { created_at: :asc }, per_page: 3) }
|
||
|
|
|
||
|
|
it 'raises an error if has a different order clause than the page' do
|
||
|
|
expect { subject }.to raise_error(ArgumentError, /order_by does not match/)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|