public
/
bilibili-go-common
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1 Commit
1 Branch
0 Tags
34 MiB
 
 
 
Branch: master
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'master'
${ noResults }
bilibili-go-common/vendor/gopkg.in/olivere/elastic.v5/search_aggs_pipeline_test.go

1002 lines
28 KiB
Raw Permalink Blame History

// Copyright 2012-present Oliver Eilhard. All rights reserved.
// Use of this source code is governed by a MIT-license.
// See http://olivere.mit-license.org/license.txt for details.
package elastic
import (
"context"
"testing"
)
func TestAggsIntegrationAvgBucket(t *testing.T) {
//client := setupTestClientAndCreateIndexAndAddDocs(t, SetTraceLog(log.New(os.Stdout, "", log.LstdFlags)))
client := setupTestClientAndCreateIndexAndAddDocs(t)
esversion, err := client.ElasticsearchVersion(DefaultURL)
if err != nil {
t.Fatal(err)
}
if esversion < "2.0" {
t.Skipf("Elasticsearch %s does not have pipeline aggregations.", esversion)
return
}
// Match all should return all documents
builder := client.Search().
Index(testIndexName).
Type("order").
Query(NewMatchAllQuery()).
Pretty(true)
h := NewDateHistogramAggregation().Field("time").Interval("month")
h = h.SubAggregation("sales", NewSumAggregation().Field("price"))
builder = builder.Aggregation("sales_per_month", h)
builder = builder.Aggregation("avg_monthly_sales", NewAvgBucketAggregation().BucketsPath("sales_per_month>sales"))
res, err := builder.Do(context.TODO())
if err != nil {
t.Fatal(err)
}
if res.Hits == nil {
t.Errorf("expected Hits != nil; got: nil")
}
aggs := res.Aggregations
if aggs == nil {
t.Fatal("expected aggregations != nil; got: nil")
}
agg, found := aggs.AvgBucket("avg_monthly_sales")
if !found {
t.Fatal("expected avg_monthly_sales aggregation")
}
if agg == nil {
t.Fatal("expected avg_monthly_sales aggregation")
}
if agg.Value == nil {
t.Fatal("expected avg_monthly_sales.value != nil")
}
if got, want := *agg.Value, float64(939.2); got != want {
t.Fatalf("expected avg_monthly_sales.value=%v; got: %v", want, got)
}
}
func TestAggsIntegrationDerivative(t *testing.T) {
//client := setupTestClientAndCreateIndexAndAddDocs(t, SetTraceLog(log.New(os.Stdout, "", log.LstdFlags)))
client := setupTestClientAndCreateIndexAndAddDocs(t)
esversion, err := client.ElasticsearchVersion(DefaultURL)
if err != nil {
t.Fatal(err)
}
if esversion < "2.0" {
t.Skipf("Elasticsearch %s does not have pipeline aggregations.", esversion)
return
}
// Match all should return all documents
builder := client.Search().
Index(testIndexName).
Type("order").
Query(NewMatchAllQuery()).
Pretty(true)
h := NewDateHistogramAggregation().Field("time").Interval("month")
h = h.SubAggregation("sales", NewSumAggregation().Field("price"))
h = h.SubAggregation("sales_deriv", NewDerivativeAggregation().BucketsPath("sales"))
builder = builder.Aggregation("sales_per_month", h)
res, err := builder.Do(context.TODO())
if err != nil {
t.Fatal(err)
}
if res.Hits == nil {
t.Errorf("expected Hits != nil; got: nil")
}
aggs := res.Aggregations
if aggs == nil {
t.Fatal("expected aggregations != nil; got: nil")
}
agg, found := aggs.DateHistogram("sales_per_month")
if !found {
t.Fatal("expected sales_per_month aggregation")
}
if agg == nil {
t.Fatal("expected sales_per_month aggregation")
}
if got, want := len(agg.Buckets), 6; got != want {
t.Fatalf("expected %d buckets; got: %d", want, got)
}
if got, want := agg.Buckets[0].DocCount, int64(1); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[1].DocCount, int64(0); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[2].DocCount, int64(1); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[3].DocCount, int64(3); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[4].DocCount, int64(1); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[5].DocCount, int64(2); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
d, found := agg.Buckets[0].Derivative("sales_deriv")
if found {
t.Fatal("expected no sales_deriv aggregation")
}
if d != nil {
t.Fatal("expected no sales_deriv aggregation")
}
d, found = agg.Buckets[1].Derivative("sales_deriv")
if !found {
t.Fatal("expected sales_deriv aggregation")
}
if d == nil {
t.Fatal("expected sales_deriv aggregation")
}
if d.Value != nil {
t.Fatal("expected sales_deriv value == nil")
}
d, found = agg.Buckets[2].Derivative("sales_deriv")
if !found {
t.Fatal("expected sales_deriv aggregation")
}
if d == nil {
t.Fatal("expected sales_deriv aggregation")
}
if d.Value != nil {
t.Fatal("expected sales_deriv value == nil")
}
d, found = agg.Buckets[3].Derivative("sales_deriv")
if !found {
t.Fatal("expected sales_deriv aggregation")
}
if d == nil {
t.Fatal("expected sales_deriv aggregation")
}
if d.Value == nil {
t.Fatal("expected sales_deriv value != nil")
}
if got, want := *d.Value, float64(2348.0); got != want {
t.Fatalf("expected sales_deriv.value=%v; got: %v", want, got)
}
d, found = agg.Buckets[4].Derivative("sales_deriv")
if !found {
t.Fatal("expected sales_deriv aggregation")
}
if d == nil {
t.Fatal("expected sales_deriv aggregation")
}
if d.Value == nil {
t.Fatal("expected sales_deriv value != nil")
}
if got, want := *d.Value, float64(-1658.0); got != want {
t.Fatalf("expected sales_deriv.value=%v; got: %v", want, got)
}
d, found = agg.Buckets[5].Derivative("sales_deriv")
if !found {
t.Fatal("expected sales_deriv aggregation")
}
if d == nil {
t.Fatal("expected sales_deriv aggregation")
}
if d.Value == nil {
t.Fatal("expected sales_deriv value != nil")
}
if got, want := *d.Value, float64(-722.0); got != want {
t.Fatalf("expected sales_deriv.value=%v; got: %v", want, got)
}
}
func TestAggsIntegrationMaxBucket(t *testing.T) {
//client := setupTestClientAndCreateIndexAndAddDocs(t, SetTraceLog(log.New(os.Stdout, "", log.LstdFlags)))
client := setupTestClientAndCreateIndexAndAddDocs(t)
esversion, err := client.ElasticsearchVersion(DefaultURL)
if err != nil {
t.Fatal(err)
}
if esversion < "2.0" {
t.Skipf("Elasticsearch %s does not have pipeline aggregations.", esversion)
return
}
// Match all should return all documents
builder := client.Search().
Index(testIndexName).
Type("order").
Query(NewMatchAllQuery()).
Pretty(true)
h := NewDateHistogramAggregation().Field("time").Interval("month")
h = h.SubAggregation("sales", NewSumAggregation().Field("price"))
builder = builder.Aggregation("sales_per_month", h)
builder = builder.Aggregation("max_monthly_sales", NewMaxBucketAggregation().BucketsPath("sales_per_month>sales"))
res, err := builder.Do(context.TODO())
if err != nil {
t.Fatal(err)
}
if res.Hits == nil {
t.Errorf("expected Hits != nil; got: nil")
}
aggs := res.Aggregations
if aggs == nil {
t.Fatal("expected aggregations != nil; got: nil")
}
agg, found := aggs.MaxBucket("max_monthly_sales")
if !found {
t.Fatal("expected max_monthly_sales aggregation")
}
if agg == nil {
t.Fatal("expected max_monthly_sales aggregation")
}
if got, want := len(agg.Keys), 1; got != want {
t.Fatalf("expected len(max_monthly_sales.keys)=%d; got: %d", want, got)
}
if got, want := agg.Keys[0], "2015-04-01"; got != want {
t.Fatalf("expected max_monthly_sales.keys[0]=%v; got: %v", want, got)
}
if agg.Value == nil {
t.Fatal("expected max_monthly_sales.value != nil")
}
if got, want := *agg.Value, float64(2448); got != want {
t.Fatalf("expected max_monthly_sales.value=%v; got: %v", want, got)
}
}
func TestAggsIntegrationMinBucket(t *testing.T) {
//client := setupTestClientAndCreateIndexAndAddDocs(t, SetTraceLog(log.New(os.Stdout, "", log.LstdFlags)))
client := setupTestClientAndCreateIndexAndAddDocs(t)
esversion, err := client.ElasticsearchVersion(DefaultURL)
if err != nil {
t.Fatal(err)
}
if esversion < "2.0" {
t.Skipf("Elasticsearch %s does not have pipeline aggregations.", esversion)
return
}
// Match all should return all documents
builder := client.Search().
Index(testIndexName).
Type("order").
Query(NewMatchAllQuery()).
Pretty(true)
h := NewDateHistogramAggregation().Field("time").Interval("month")
h = h.SubAggregation("sales", NewSumAggregation().Field("price"))
builder = builder.Aggregation("sales_per_month", h)
builder = builder.Aggregation("min_monthly_sales", NewMinBucketAggregation().BucketsPath("sales_per_month>sales"))
res, err := builder.Do(context.TODO())
if err != nil {
t.Fatal(err)
}
if res.Hits == nil {
t.Errorf("expected Hits != nil; got: nil")
}
aggs := res.Aggregations
if aggs == nil {
t.Fatal("expected aggregations != nil; got: nil")
}
agg, found := aggs.MinBucket("min_monthly_sales")
if !found {
t.Fatal("expected min_monthly_sales aggregation")
}
if agg == nil {
t.Fatal("expected min_monthly_sales aggregation")
}
if got, want := len(agg.Keys), 1; got != want {
t.Fatalf("expected len(min_monthly_sales.keys)=%d; got: %d", want, got)
}
if got, want := agg.Keys[0], "2015-06-01"; got != want {
t.Fatalf("expected min_monthly_sales.keys[0]=%v; got: %v", want, got)
}
if agg.Value == nil {
t.Fatal("expected min_monthly_sales.value != nil")
}
if got, want := *agg.Value, float64(68); got != want {
t.Fatalf("expected min_monthly_sales.value=%v; got: %v", want, got)
}
}
func TestAggsIntegrationSumBucket(t *testing.T) {
//client := setupTestClientAndCreateIndexAndAddDocs(t, SetTraceLog(log.New(os.Stdout, "", log.LstdFlags)))
client := setupTestClientAndCreateIndexAndAddDocs(t)
esversion, err := client.ElasticsearchVersion(DefaultURL)
if err != nil {
t.Fatal(err)
}
if esversion < "2.0" {
t.Skipf("Elasticsearch %s does not have pipeline aggregations.", esversion)
return
}
// Match all should return all documents
builder := client.Search().
Index(testIndexName).
Type("order").
Query(NewMatchAllQuery()).
Pretty(true)
h := NewDateHistogramAggregation().Field("time").Interval("month")
h = h.SubAggregation("sales", NewSumAggregation().Field("price"))
builder = builder.Aggregation("sales_per_month", h)
builder = builder.Aggregation("sum_monthly_sales", NewSumBucketAggregation().BucketsPath("sales_per_month>sales"))
res, err := builder.Do(context.TODO())
if err != nil {
t.Fatal(err)
}
if res.Hits == nil {
t.Errorf("expected Hits != nil; got: nil")
}
aggs := res.Aggregations
if aggs == nil {
t.Fatal("expected aggregations != nil; got: nil")
}
agg, found := aggs.SumBucket("sum_monthly_sales")
if !found {
t.Fatal("expected sum_monthly_sales aggregation")
}
if agg == nil {
t.Fatal("expected sum_monthly_sales aggregation")
}
if agg.Value == nil {
t.Fatal("expected sum_monthly_sales.value != nil")
}
if got, want := *agg.Value, float64(4696.0); got != want {
t.Fatalf("expected sum_monthly_sales.value=%v; got: %v", want, got)
}
}
func TestAggsIntegrationMovAvg(t *testing.T) {
//client := setupTestClientAndCreateIndexAndAddDocs(t, SetTraceLog(log.New(os.Stdout, "", log.LstdFlags)))
client := setupTestClientAndCreateIndexAndAddDocs(t)
esversion, err := client.ElasticsearchVersion(DefaultURL)
if err != nil {
t.Fatal(err)
}
if esversion < "2.0" {
t.Skipf("Elasticsearch %s does not have pipeline aggregations.", esversion)
return
}
// Match all should return all documents
builder := client.Search().
Index(testIndexName).
Type("order").
Query(NewMatchAllQuery()).
Pretty(true)
h := NewDateHistogramAggregation().Field("time").Interval("month")
h = h.SubAggregation("the_sum", NewSumAggregation().Field("price"))
h = h.SubAggregation("the_movavg", NewMovAvgAggregation().BucketsPath("the_sum"))
builder = builder.Aggregation("my_date_histo", h)
res, err := builder.Do(context.TODO())
if err != nil {
t.Fatal(err)
}
if res.Hits == nil {
t.Errorf("expected Hits != nil; got: nil")
}
aggs := res.Aggregations
if aggs == nil {
t.Fatal("expected aggregations != nil; got: nil")
}
agg, found := aggs.DateHistogram("my_date_histo")
if !found {
t.Fatal("expected sum_monthly_sales aggregation")
}
if agg == nil {
t.Fatal("expected sum_monthly_sales aggregation")
}
if got, want := len(agg.Buckets), 6; got != want {
t.Fatalf("expected %d buckets; got: %d", want, got)
}
d, found := agg.Buckets[0].MovAvg("the_movavg")
if found {
t.Fatal("expected no the_movavg aggregation")
}
if d != nil {
t.Fatal("expected no the_movavg aggregation")
}
d, found = agg.Buckets[1].MovAvg("the_movavg")
if found {
t.Fatal("expected no the_movavg aggregation")
}
if d != nil {
t.Fatal("expected no the_movavg aggregation")
}
d, found = agg.Buckets[2].MovAvg("the_movavg")
if !found {
t.Fatal("expected the_movavg aggregation")
}
if d == nil {
t.Fatal("expected the_movavg aggregation")
}
if d.Value == nil {
t.Fatal("expected the_movavg value")
}
if got, want := *d.Value, float64(1290.0); got != want {
t.Fatalf("expected %v buckets; got: %v", want, got)
}
d, found = agg.Buckets[3].MovAvg("the_movavg")
if !found {
t.Fatal("expected the_movavg aggregation")
}
if d == nil {
t.Fatal("expected the_movavg aggregation")
}
if d.Value == nil {
t.Fatal("expected the_movavg value")
}
if got, want := *d.Value, float64(695.0); got != want {
t.Fatalf("expected %v buckets; got: %v", want, got)
}
d, found = agg.Buckets[4].MovAvg("the_movavg")
if !found {
t.Fatal("expected the_movavg aggregation")
}
if d == nil {
t.Fatal("expected the_movavg aggregation")
}
if d.Value == nil {
t.Fatal("expected the_movavg value")
}
if got, want := *d.Value, float64(1279.3333333333333); got != want {
t.Fatalf("expected %v buckets; got: %v", want, got)
}
d, found = agg.Buckets[5].MovAvg("the_movavg")
if !found {
t.Fatal("expected the_movavg aggregation")
}
if d == nil {
t.Fatal("expected the_movavg aggregation")
}
if d.Value == nil {
t.Fatal("expected the_movavg value")
}
if got, want := *d.Value, float64(1157.0); got != want {
t.Fatalf("expected %v buckets; got: %v", want, got)
}
}
func TestAggsIntegrationCumulativeSum(t *testing.T) {
//client := setupTestClientAndCreateIndexAndAddDocs(t, SetTraceLog(log.New(os.Stdout, "", log.LstdFlags)))
client := setupTestClientAndCreateIndexAndAddDocs(t)
esversion, err := client.ElasticsearchVersion(DefaultURL)
if err != nil {
t.Fatal(err)
}
if esversion < "2.0" {
t.Skipf("Elasticsearch %s does not have pipeline aggregations.", esversion)
return
}
// Match all should return all documents
builder := client.Search().
Index(testIndexName).
Type("order").
Query(NewMatchAllQuery()).
Pretty(true)
h := NewDateHistogramAggregation().Field("time").Interval("month")
h = h.SubAggregation("sales", NewSumAggregation().Field("price"))
h = h.SubAggregation("cumulative_sales", NewCumulativeSumAggregation().BucketsPath("sales"))
builder = builder.Aggregation("sales_per_month", h)
res, err := builder.Do(context.TODO())
if err != nil {
t.Fatal(err)
}
if res.Hits == nil {
t.Errorf("expected Hits != nil; got: nil")
}
aggs := res.Aggregations
if aggs == nil {
t.Fatal("expected aggregations != nil; got: nil")
}
agg, found := aggs.DateHistogram("sales_per_month")
if !found {
t.Fatal("expected sales_per_month aggregation")
}
if agg == nil {
t.Fatal("expected sales_per_month aggregation")
}
if got, want := len(agg.Buckets), 6; got != want {
t.Fatalf("expected %d buckets; got: %d", want, got)
}
if got, want := agg.Buckets[0].DocCount, int64(1); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[1].DocCount, int64(0); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[2].DocCount, int64(1); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[3].DocCount, int64(3); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[4].DocCount, int64(1); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[5].DocCount, int64(2); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
d, found := agg.Buckets[0].CumulativeSum("cumulative_sales")
if !found {
t.Fatal("expected cumulative_sales aggregation")
}
if d == nil {
t.Fatal("expected cumulative_sales aggregation")
}
if d.Value == nil {
t.Fatal("expected cumulative_sales value != nil")
}
if got, want := *d.Value, float64(1290.0); got != want {
t.Fatalf("expected cumulative_sales.value=%v; got: %v", want, got)
}
d, found = agg.Buckets[1].CumulativeSum("cumulative_sales")
if !found {
t.Fatal("expected cumulative_sales aggregation")
}
if d == nil {
t.Fatal("expected cumulative_sales aggregation")
}
if d.Value == nil {
t.Fatal("expected cumulative_sales value != nil")
}
if got, want := *d.Value, float64(1290.0); got != want {
t.Fatalf("expected cumulative_sales.value=%v; got: %v", want, got)
}
d, found = agg.Buckets[2].CumulativeSum("cumulative_sales")
if !found {
t.Fatal("expected cumulative_sales aggregation")
}
if d == nil {
t.Fatal("expected cumulative_sales aggregation")
}
if d.Value == nil {
t.Fatal("expected cumulative_sales value != nil")
}
if got, want := *d.Value, float64(1390.0); got != want {
t.Fatalf("expected cumulative_sales.value=%v; got: %v", want, got)
}
d, found = agg.Buckets[3].CumulativeSum("cumulative_sales")
if !found {
t.Fatal("expected cumulative_sales aggregation")
}
if d == nil {
t.Fatal("expected cumulative_sales aggregation")
}
if d.Value == nil {
t.Fatal("expected cumulative_sales value != nil")
}
if got, want := *d.Value, float64(3838.0); got != want {
t.Fatalf("expected cumulative_sales.value=%v; got: %v", want, got)
}
d, found = agg.Buckets[4].CumulativeSum("cumulative_sales")
if !found {
t.Fatal("expected cumulative_sales aggregation")
}
if d == nil {
t.Fatal("expected cumulative_sales aggregation")
}
if d.Value == nil {
t.Fatal("expected cumulative_sales value != nil")
}
if got, want := *d.Value, float64(4628.0); got != want {
t.Fatalf("expected cumulative_sales.value=%v; got: %v", want, got)
}
d, found = agg.Buckets[5].CumulativeSum("cumulative_sales")
if !found {
t.Fatal("expected cumulative_sales aggregation")
}
if d == nil {
t.Fatal("expected cumulative_sales aggregation")
}
if d.Value == nil {
t.Fatal("expected cumulative_sales value != nil")
}
if got, want := *d.Value, float64(4696.0); got != want {
t.Fatalf("expected cumulative_sales.value=%v; got: %v", want, got)
}
}
func TestAggsIntegrationBucketScript(t *testing.T) {
client := setupTestClientAndCreateIndexAndAddDocs(t) //, SetTraceLog(log.New(os.Stdout, "", log.LstdFlags)))
esversion, err := client.ElasticsearchVersion(DefaultURL)
if err != nil {
t.Fatal(err)
}
if esversion < "2.0" {
t.Skipf("Elasticsearch %s does not have pipeline aggregations.", esversion)
return
}
// Match all should return all documents
builder := client.Search().
Index(testIndexName).
Type("order").
Query(NewMatchAllQuery()).
Pretty(true)
h := NewDateHistogramAggregation().Field("time").Interval("month")
h = h.SubAggregation("total_sales", NewSumAggregation().Field("price"))
appleFilter := NewFilterAggregation().Filter(NewTermQuery("manufacturer", "Apple"))
appleFilter = appleFilter.SubAggregation("sales", NewSumAggregation().Field("price"))
h = h.SubAggregation("apple_sales", appleFilter)
h = h.SubAggregation("apple_percentage",
NewBucketScriptAggregation().
GapPolicy("insert_zeros").
AddBucketsPath("appleSales", "apple_sales>sales").
AddBucketsPath("totalSales", "total_sales").
Script(NewScript("params.appleSales / params.totalSales * 100")))
builder = builder.Aggregation("sales_per_month", h)
res, err := builder.Pretty(true).Do(context.TODO())
if err != nil {
t.Fatalf("%v (maybe scripting is disabled?)", err)
}
if res.Hits == nil {
t.Errorf("expected Hits != nil; got: nil")
}
aggs := res.Aggregations
if aggs == nil {
t.Fatal("expected aggregations != nil; got: nil")
}
agg, found := aggs.DateHistogram("sales_per_month")
if !found {
t.Fatal("expected sales_per_month aggregation")
}
if agg == nil {
t.Fatal("expected sales_per_month aggregation")
}
if got, want := len(agg.Buckets), 6; got != want {
t.Fatalf("expected %d buckets; got: %d", want, got)
}
if got, want := agg.Buckets[0].DocCount, int64(1); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[1].DocCount, int64(0); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[2].DocCount, int64(1); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[3].DocCount, int64(3); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[4].DocCount, int64(1); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[5].DocCount, int64(2); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
d, found := agg.Buckets[0].BucketScript("apple_percentage")
if !found {
t.Fatal("expected apple_percentage aggregation")
}
if d == nil {
t.Fatal("expected apple_percentage aggregation")
}
if d.Value == nil {
t.Fatal("expected apple_percentage value != nil")
}
if got, want := *d.Value, float64(100.0); got != want {
t.Fatalf("expected apple_percentage.value=%v; got: %v", want, got)
}
d, found = agg.Buckets[1].BucketScript("apple_percentage")
if !found {
t.Fatal("expected apple_percentage aggregation")
}
if d == nil {
t.Fatal("expected apple_percentage aggregation")
}
if d.Value != nil {
t.Fatal("expected apple_percentage value == nil")
}
d, found = agg.Buckets[2].BucketScript("apple_percentage")
if !found {
t.Fatal("expected apple_percentage aggregation")
}
if d == nil {
t.Fatal("expected apple_percentage aggregation")
}
if d.Value == nil {
t.Fatal("expected apple_percentage value != nil")
}
if got, want := *d.Value, float64(0.0); got != want {
t.Fatalf("expected apple_percentage.value=%v; got: %v", want, got)
}
d, found = agg.Buckets[3].BucketScript("apple_percentage")
if !found {
t.Fatal("expected apple_percentage aggregation")
}
if d == nil {
t.Fatal("expected apple_percentage aggregation")
}
if d.Value == nil {
t.Fatal("expected apple_percentage value != nil")
}
if got, want := *d.Value, float64(34.64052287581699); got != want {
t.Fatalf("expected apple_percentage.value=%v; got: %v", want, got)
}
d, found = agg.Buckets[4].BucketScript("apple_percentage")
if !found {
t.Fatal("expected apple_percentage aggregation")
}
if d == nil {
t.Fatal("expected apple_percentage aggregation")
}
if d.Value == nil {
t.Fatal("expected apple_percentage value != nil")
}
if got, want := *d.Value, float64(0.0); got != want {
t.Fatalf("expected apple_percentage.value=%v; got: %v", want, got)
}
d, found = agg.Buckets[5].BucketScript("apple_percentage")
if !found {
t.Fatal("expected apple_percentage aggregation")
}
if d == nil {
t.Fatal("expected apple_percentage aggregation")
}
if d.Value == nil {
t.Fatal("expected apple_percentage value != nil")
}
if got, want := *d.Value, float64(0.0); got != want {
t.Fatalf("expected apple_percentage.value=%v; got: %v", want, got)
}
}
func TestAggsIntegrationBucketSelector(t *testing.T) {
//client := setupTestClientAndCreateIndexAndAddDocs(t, SetTraceLog(log.New(os.Stdout, "", log.LstdFlags)))
client := setupTestClientAndCreateIndexAndAddDocs(t)
esversion, err := client.ElasticsearchVersion(DefaultURL)
if err != nil {
t.Fatal(err)
}
if esversion < "2.0" {
t.Skipf("Elasticsearch %s does not have pipeline aggregations.", esversion)
return
}
// Match all should return all documents
builder := client.Search().
Index(testIndexName).
Type("order").
Query(NewMatchAllQuery()).
Pretty(true)
h := NewDateHistogramAggregation().Field("time").Interval("month")
h = h.SubAggregation("total_sales", NewSumAggregation().Field("price"))
h = h.SubAggregation("sales_bucket_filter",
NewBucketSelectorAggregation().
AddBucketsPath("totalSales", "total_sales").
Script(NewScript("params.totalSales <= 100")))
builder = builder.Aggregation("sales_per_month", h)
res, err := builder.Do(context.TODO())
if err != nil {
t.Fatalf("%v (maybe scripting is disabled?)", err)
}
if res.Hits == nil {
t.Errorf("expected Hits != nil; got: nil")
}
aggs := res.Aggregations
if aggs == nil {
t.Fatal("expected aggregations != nil; got: nil")
}
agg, found := aggs.DateHistogram("sales_per_month")
if !found {
t.Fatal("expected sales_per_month aggregation")
}
if agg == nil {
t.Fatal("expected sales_per_month aggregation")
}
if got, want := len(agg.Buckets), 2; got != want {
t.Fatalf("expected %d buckets; got: %d", want, got)
}
if got, want := agg.Buckets[0].DocCount, int64(1); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[1].DocCount, int64(2); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
}
func TestAggsIntegrationSerialDiff(t *testing.T) {
//client := setupTestClientAndCreateIndexAndAddDocs(t, SetTraceLog(log.New(os.Stdout, "", log.LstdFlags)))
client := setupTestClientAndCreateIndexAndAddDocs(t)
esversion, err := client.ElasticsearchVersion(DefaultURL)
if err != nil {
t.Fatal(err)
}
if esversion < "2.0" {
t.Skipf("Elasticsearch %s does not have pipeline aggregations.", esversion)
return
}
// Match all should return all documents
builder := client.Search().
Index(testIndexName).
Type("order").
Query(NewMatchAllQuery()).
Pretty(true)
h := NewDateHistogramAggregation().Field("time").Interval("month")
h = h.SubAggregation("sales", NewSumAggregation().Field("price"))
h = h.SubAggregation("the_diff", NewSerialDiffAggregation().BucketsPath("sales").Lag(1))
builder = builder.Aggregation("sales_per_month", h)
res, err := builder.Do(context.TODO())
if err != nil {
t.Fatal(err)
}
if res.Hits == nil {
t.Errorf("expected Hits != nil; got: nil")
}
aggs := res.Aggregations
if aggs == nil {
t.Fatal("expected aggregations != nil; got: nil")
}
agg, found := aggs.DateHistogram("sales_per_month")
if !found {
t.Fatal("expected sales_per_month aggregation")
}
if agg == nil {
t.Fatal("expected sales_per_month aggregation")
}
if got, want := len(agg.Buckets), 6; got != want {
t.Fatalf("expected %d buckets; got: %d", want, got)
}
if got, want := agg.Buckets[0].DocCount, int64(1); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[1].DocCount, int64(0); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[2].DocCount, int64(1); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[3].DocCount, int64(3); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[4].DocCount, int64(1); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
if got, want := agg.Buckets[5].DocCount, int64(2); got != want {
t.Fatalf("expected DocCount=%d; got: %d", want, got)
}
d, found := agg.Buckets[0].SerialDiff("the_diff")
if found {
t.Fatal("expected no the_diff aggregation")
}
if d != nil {
t.Fatal("expected no the_diff aggregation")
}
d, found = agg.Buckets[1].SerialDiff("the_diff")
if found {
t.Fatal("expected no the_diff aggregation")
}
if d != nil {
t.Fatal("expected no the_diff aggregation")
}
d, found = agg.Buckets[2].SerialDiff("the_diff")
if found {
t.Fatal("expected no the_diff aggregation")
}
if d != nil {
t.Fatal("expected no the_diff aggregation")
}
d, found = agg.Buckets[3].SerialDiff("the_diff")
if !found {
t.Fatal("expected the_diff aggregation")
}
if d == nil {
t.Fatal("expected the_diff aggregation")
}
if d.Value == nil {
t.Fatal("expected the_diff value != nil")
}
if got, want := *d.Value, float64(2348.0); got != want {
t.Fatalf("expected the_diff.value=%v; got: %v", want, got)
}
d, found = agg.Buckets[4].SerialDiff("the_diff")
if !found {
t.Fatal("expected the_diff aggregation")
}
if d == nil {
t.Fatal("expected the_diff aggregation")
}
if d.Value == nil {
t.Fatal("expected the_diff value != nil")
}
if got, want := *d.Value, float64(-1658.0); got != want {
t.Fatalf("expected the_diff.value=%v; got: %v", want, got)
}
d, found = agg.Buckets[5].SerialDiff("the_diff")
if !found {
t.Fatal("expected the_diff aggregation")
}
if d == nil {
t.Fatal("expected the_diff aggregation")
}
if d.Value == nil {
t.Fatal("expected the_diff value != nil")
}
if got, want := *d.Value, float64(-722.0); got != want {
t.Fatalf("expected the_diff.value=%v; got: %v", want, got)
}
}