From f182c8d9f1a1302943b85b09d47d7b99d5e3d242 Mon Sep 17 00:00:00 2001 From: Stephen Ierodiaconou Date: Mon, 7 Oct 2024 12:06:37 +0200 Subject: [PATCH] Fakes use configured base classes --- lib/quo/fakes/collection_backed_fake.rb | 2 +- lib/quo/fakes/relation_backed_fake.rb | 2 +- test/quo/fake_query_test.rb | 2 ++ 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/lib/quo/fakes/collection_backed_fake.rb b/lib/quo/fakes/collection_backed_fake.rb index f9494f4..5c9d6d4 100644 --- a/lib/quo/fakes/collection_backed_fake.rb +++ b/lib/quo/fakes/collection_backed_fake.rb @@ -4,7 +4,7 @@ module Quo module Fakes - class CollectionBackedFake < CollectionBackedQuery + class CollectionBackedFake < Quo.collection_backed_query_base_class prop :results, _Any, reader: false prop :page_count, _Nilable(Integer), reader: false diff --git a/lib/quo/fakes/relation_backed_fake.rb b/lib/quo/fakes/relation_backed_fake.rb index 3a18cd5..e8620c0 100644 --- a/lib/quo/fakes/relation_backed_fake.rb +++ b/lib/quo/fakes/relation_backed_fake.rb @@ -4,7 +4,7 @@ module Quo module Fakes - class RelationBackedFake < RelationBackedQuery + class RelationBackedFake < Quo.relation_backed_query_base_class prop :results, _Any, reader: false prop :page_count, _Nilable(Integer), reader: false prop :total_count, _Nilable(Integer), reader: false diff --git a/test/quo/fake_query_test.rb b/test/quo/fake_query_test.rb index 9fc83f7..825a543 100644 --- a/test/quo/fake_query_test.rb +++ b/test/quo/fake_query_test.rb @@ -21,6 +21,7 @@ def collection q = NewCommentsForAuthorQuery.new(author_id: 1) assert q.results.is_a?(Quo::RelationResults) assert q.is_a?(Quo::RelationBackedQuery) + assert_equal "relation", q.hello assert_equal 2, q.results.count assert_equal 1, q.results.first assert_nothing_raised do @@ -35,6 +36,7 @@ def collection q = klass.new assert q.results.is_a?(Quo::CollectionResults) assert q.is_a?(Quo::CollectionBackedQuery) + assert_equal "collection", q.hello assert_equal 2, q.results.count assert_equal 1, q.results.first assert_nothing_raised do