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