1// Copyright 2012-present Oliver Eilhard. All rights reserved. 2// Use of this source code is governed by a MIT-license. 3// See http://olivere.mit-license.org/license.txt for details. 4 5package elastic 6 7import ( 8 "encoding/json" 9 "testing" 10) 11 12func TestBucketScriptAggregation(t *testing.T) { 13 agg := NewBucketScriptAggregation(). 14 AddBucketsPath("tShirtSales", "t-shirts>sales"). 15 AddBucketsPath("totalSales", "total_sales"). 16 Script(NewScript("tShirtSales / totalSales * 100")) 17 src, err := agg.Source() 18 if err != nil { 19 t.Fatal(err) 20 } 21 data, err := json.Marshal(src) 22 if err != nil { 23 t.Fatalf("marshaling to JSON failed: %v", err) 24 } 25 got := string(data) 26 expected := `{"bucket_script":{"buckets_path":{"tShirtSales":"t-shirts\u003esales","totalSales":"total_sales"},"script":{"inline":"tShirtSales / totalSales * 100"}}}` 27 if got != expected { 28 t.Errorf("expected\n%s\n,got:\n%s", expected, got) 29 } 30} 31