1 files changed, 5 insertions, 4 deletions
diff --git a/spec/graphql/resolvers/release_milestones_resolver_spec.rb b/spec/graphql/resolvers/release_milestones_resolver_spec.rb
index 5f66cba859d..f05069998d0 100644
--- a/spec/graphql/resolvers/release_milestones_resolver_spec.rb
+++ b/spec/graphql/resolvers/release_milestones_resolver_spec.rb
@@ -6,18 +6,19 @@ RSpec.describe Resolvers::ReleaseMilestonesResolver do
   include GraphqlHelpers
 
   let_it_be(:release) { create(:release, :with_milestones, milestones_count: 2) }
+  let_it_be(:current_user) { create(:user, developer_projects: [release.project]) }
 
   let(:resolved) do
-    resolve(described_class, obj: release)
+    resolve(described_class, obj: release, ctx: { current_user: current_user })
   end
 
   describe '#resolve' do
-    it "returns an OffsetActiveRecordRelationConnection" do
-      expect(resolved).to be_a(::Gitlab::Graphql::Pagination::OffsetActiveRecordRelationConnection)
+    it "uses offset-pagination" do
+      expect(resolved).to be_a(::Gitlab::Graphql::Pagination::OffsetPaginatedRelation)
     end
 
     it "includes the release's milestones in the returned OffsetActiveRecordRelationConnection" do
-      expect(resolved.items).to eq(release.milestones.order_by_dates_and_title)
+      expect(resolved.to_a).to eq(release.milestones.order_by_dates_and_title)
     end
   end
 end