add track_total_hits option

elastic · Nov 5, 2019 · 0b16c81 · 0b16c81
1 parent 2267889
commit 0b16c81
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/lib/elasticsearch/dsl/search/options.rb b/lib/elasticsearch/dsl/search/options.rb
@@ -21,7 +21,8 @@ class Options
           :version,
           :indices_boost,
           :track_scores,
-          :min_score
+          :min_score,
+          :track_total_hits
         ]
 
         def initialize(*args, &block)

diff --git a/test/unit/search_options_test.rb b/test/unit/search_options_test.rb
@@ -64,6 +64,14 @@ class SearchOptionsTest < ::Elasticsearch::Test::UnitTestCase
           assert_equal( { version: true }, subject.to_hash )
         end
 
+        should "encode track_total_hits" do
+          subject.track_total_hits 123
+          assert_equal( { track_total_hits: 123 }, subject.to_hash )
+
+          subject.track_total_hits true
+          assert_equal( { track_total_hits: true }, subject.to_hash )
+        end
+
         should "encode indices_boost" do
           subject.indices_boost foo: 'bar'
           assert_equal( { indices_boost: { foo: 'bar' } }, subject.to_hash )