Added includes method on relation

Andrew Kane
1 parent 79e14153
Showing 2 changed files with 13 additions and 0 deletions Show diff stats
lib/searchkick/relation.rb
test/query_test.rb
@@ -170,6 +170,18 @@ module Searchkick
     end
  
     # experimental
+    def includes(*values)
+      clone.includes!(*values)
+    end
+
+    # experimental
+    def includes!(*values)
+      check_loaded
+      (@options[:includes] ||= []).concat(values)
+      self
+    end
+
+    # experimental
     def only(*keys)
       Relation.new(@model, @term, **@options.slice(*keys))
     end
@@ -61,6 +61,7 @@ class QueryTest &lt; Minitest::Test
  
     store_names ["Product A"]
     assert Product.search("product", includes: [:store]).first.association(:store).loaded?
+    assert Product.search("product").includes(:store).first.association(:store).loaded?
   end
  
   def test_model_includes
...	...	@@ -170,6 +170,18 @@ module Searchkick
170	170	end
171	171
172	172	# experimental
	173	+ def includes(*values)
	174	+ clone.includes!(*values)
	175	+ end
	176	+
	177	+ # experimental
	178	+ def includes!(*values)
	179	+ check_loaded
	180	+ (@options[:includes] \|\|= []).concat(values)
	181	+ self
	182	+ end
	183	+
	184	+ # experimental
173	185	def only(*keys)
174	186	Relation.new(@model, @term, *@options.slice(keys))
175	187	end
...	...
...	...	@@ -61,6 +61,7 @@ class QueryTest < Minitest::Test
61	61
62	62	store_names ["Product A"]
63	63	assert Product.search("product", includes: [:store]).first.association(:store).loaded?
	64	+ assert Product.search("product").includes(:store).first.association(:store).loaded?
64	65	end
65	66
66	67	def test_model_includes
...	...