Fakes use configured base classes

stevegeek · Oct 7, 2024 · f182c8d · f182c8d
1 parent fe09507
commit f182c8d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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