diff --git a/lib/searchkick/relation.rb b/lib/searchkick/relation.rb index 354e6d7..8bac8f9 100644 --- a/lib/searchkick/relation.rb +++ b/lib/searchkick/relation.rb @@ -232,7 +232,7 @@ module Searchkick # TODO merge options def aggs!(*args) - options[:aggs] = args + options[:aggs] = args.size == 1 ? args.first : args self end diff --git a/test/aggs_test.rb b/test/aggs_test.rb index 32ee7f3..ecfcbb8 100644 --- a/test/aggs_test.rb +++ b/test/aggs_test.rb @@ -18,6 +18,7 @@ class AggsTest < Minitest::Test def test_where assert_equal ({1 => 1}), store_agg(aggs: {store_id: {where: {in_stock: true}}}) + assert_equal ({1 => 1}), buckets_as_hash(Product.search("Product", relation: true).aggs(store_id: {where: {in_stock: true}}).aggs["store_id"]) end def test_order -- libgit2 0.21.0