github.com/go-graphite/carbonapi@v0.17.0/expr/functions/legendValue/function_test.go (about)

     1  package legendValue
     2  
     3  import (
     4  	"testing"
     5  	"time"
     6  
     7  	"github.com/go-graphite/carbonapi/expr/interfaces"
     8  	"github.com/go-graphite/carbonapi/expr/metadata"
     9  	"github.com/go-graphite/carbonapi/expr/types"
    10  	"github.com/go-graphite/carbonapi/pkg/parser"
    11  	th "github.com/go-graphite/carbonapi/tests"
    12  )
    13  
    14  var (
    15  	md []interfaces.FunctionMetadata = New("")
    16  )
    17  
    18  func init() {
    19  	for _, m := range md {
    20  		metadata.RegisterFunction(m.Name, m.F)
    21  	}
    22  }
    23  
    24  func TestFunction(t *testing.T) {
    25  	now32 := int64(time.Now().Unix())
    26  
    27  	tests := []th.EvalTestItem{
    28  		{
    29  			"legendValue(metric1,\"avg\")",
    30  			map[parser.MetricRequest][]*types.MetricData{
    31  				{Metric: "metric1", From: 0, Until: 1}: {types.MakeMetricData("metric1", []float64{1, 2, 3, 4, 5}, 1, now32)},
    32  			},
    33  			[]*types.MetricData{types.MakeMetricData("metric1 (avg: 3)",
    34  				[]float64{1, 2, 3, 4, 5}, 1, now32).SetNameTag("metric1")},
    35  		},
    36  		{
    37  			"legendValue(metric1,\"sum\")",
    38  			map[parser.MetricRequest][]*types.MetricData{
    39  				{Metric: "metric1", From: 0, Until: 1}: {types.MakeMetricData("metric1", []float64{1, 2, 3, 4, 5}, 1, now32)},
    40  			},
    41  			[]*types.MetricData{types.MakeMetricData("metric1 (sum: 15)",
    42  				[]float64{1, 2, 3, 4, 5}, 1, now32).SetNameTag("metric1")},
    43  		},
    44  		{
    45  			"legendValue(metric1,\"total\")",
    46  			map[parser.MetricRequest][]*types.MetricData{
    47  				{Metric: "metric1", From: 0, Until: 1}: {types.MakeMetricData("metric1", []float64{1, 2, 3, 4, 5}, 1, now32)},
    48  			},
    49  			[]*types.MetricData{types.MakeMetricData("metric1 (total: 15)",
    50  				[]float64{1, 2, 3, 4, 5}, 1, now32).SetNameTag("metric1")},
    51  		},
    52  		{
    53  			"legendValue(metric1,\"sum\",\"avg\")",
    54  			map[parser.MetricRequest][]*types.MetricData{
    55  				{Metric: "metric1", From: 0, Until: 1}: {types.MakeMetricData("metric1", []float64{1, 2, 3, 4, 5}, 1, now32)},
    56  			},
    57  			[]*types.MetricData{types.MakeMetricData("metric1 (sum: 15) (avg: 3)",
    58  				[]float64{1, 2, 3, 4, 5}, 1, now32).SetNameTag("metric1")},
    59  		},
    60  		{
    61  			"legendValue(metric1,\"sum\",\"si\")",
    62  			map[parser.MetricRequest][]*types.MetricData{
    63  				{Metric: "metric1", From: 0, Until: 1}: {types.MakeMetricData("metric1", []float64{0, 10000, 20000, -30000, -40000}, 1, now32)},
    64  			},
    65  			[]*types.MetricData{types.MakeMetricData("metric1 (sum: -40.00K )",
    66  				[]float64{0, 10000, 20000, -30000, -40000}, 1, now32).SetNameTag("metric1")},
    67  		},
    68  		{
    69  			"legendValue(metric1,\"avg\",\"total\",\"si\")",
    70  			map[parser.MetricRequest][]*types.MetricData{
    71  				{Metric: "metric1", From: 0, Until: 1}: {types.MakeMetricData("metric1", []float64{0, 10000, 20000, -30000, -40000}, 1, now32)},
    72  			},
    73  			[]*types.MetricData{types.MakeMetricData("metric1 (avg: -8.00K ) (total: -40.00K )",
    74  				[]float64{0, 10000, 20000, -30000, -40000}, 1, now32).SetNameTag("metric1")},
    75  		},
    76  		{
    77  			"legendValue(metric1,\"sum\",\"binary\")",
    78  			map[parser.MetricRequest][]*types.MetricData{
    79  				{Metric: "metric1", From: 0, Until: 1}: {types.MakeMetricData("metric1", []float64{0, 10000, 20000, -30000, -40000}, 1, now32)},
    80  			},
    81  			[]*types.MetricData{types.MakeMetricData("metric1 (sum: -39.06Ki )",
    82  				[]float64{0, 10000, 20000, -30000, -40000}, 1, now32).SetNameTag("metric1")},
    83  		},
    84  		{
    85  			"legendValue(metric1,\"avg\",\"total\",\"binary\")",
    86  			map[parser.MetricRequest][]*types.MetricData{
    87  				{Metric: "metric1", From: 0, Until: 1}: {types.MakeMetricData("metric1", []float64{0, 10000, 20000, -30000, -40000}, 1, now32)},
    88  			},
    89  			[]*types.MetricData{types.MakeMetricData("metric1 (avg: -7.81Ki ) (total: -39.06Ki )",
    90  				[]float64{0, 10000, 20000, -30000, -40000}, 1, now32).SetNameTag("metric1")},
    91  		},
    92  	}
    93  
    94  	for _, tt := range tests {
    95  		testName := tt.Target
    96  		t.Run(testName, func(t *testing.T) {
    97  			eval := th.EvaluatorFromFunc(md[0].F)
    98  			th.TestEvalExpr(t, eval, &tt)
    99  		})
   100  	}
   101  
   102  }