Added body_options and request params

Andrew Kane
1 parent 28007e3d
Showing 2 changed files with 24 additions and 0 deletions Show diff stats
lib/searchkick/relation.rb
test/query_test.rb
@@ -276,6 +276,24 @@ module Searchkick
       self
     end
+    def body_options(value)
+      spawn.body_options!(value)
+    end
+
+    def body_options!(value)
+      options[:body_options] = value
+      self
+    end
+
+    def request_params(value)
+      spawn.request_params!(value)
+    end
+
+    def request_params!(value)
+      options[:request_params] = value
+      self
+    end
+
     # same as Active Record
     def inspect
       entries = results.first(11).map!(&:inspect)
@@ -14,6 +14,11 @@ class QueryTest &lt; Minitest::Test
     assert_search "milk", ["Milk", "Milk2"], body_options: {min_score: 0.0001}
   end
+  def test_body_options
+    store_names ["Milk", "Milk2"]
+    assert_search_relation ["Milk", "Milk2"], Product.search("milk", relation: true).body_options(min_score: 0.0001)
+  end
+
   def test_default_timeout
     assert_equal "6s", Product.search("*", execute: false).body[:timeout]
   end
@@ -24,6 +29,7 @@ class QueryTest &lt; Minitest::Test
   def test_request_params
     assert_equal "dfs_query_then_fetch", Product.search("*", request_params: {search_type: "dfs_query_then_fetch"}, execute: false).params[:search_type]
+    assert_equal "dfs_query_then_fetch", Product.search("*", relation: true).request_params(search_type: "dfs_query_then_fetch").query.params[:search_type]
   end
   def test_debug
	@@ -276,6 +276,24 @@ module Searchkick		@@ -276,6 +276,24 @@ module Searchkick
276	self	276	self
277	end	277	end
278		278
		279	+ def body_options(value)
		280	+ spawn.body_options!(value)
		281	+ end
		282	+
		283	+ def body_options!(value)
		284	+ options[:body_options] = value
		285	+ self
		286	+ end
		287	+
		288	+ def request_params(value)
		289	+ spawn.request_params!(value)
		290	+ end
		291	+
		292	+ def request_params!(value)
		293	+ options[:request_params] = value
		294	+ self
		295	+ end
		296	+
279	# same as Active Record	297	# same as Active Record
280	def inspect	298	def inspect
281	entries = results.first(11).map!(&:inspect)	299	entries = results.first(11).map!(&:inspect)