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

     1  package limit
     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 TestLimit(t *testing.T) {
    25  	now32 := int64(time.Now().Unix())
    26  
    27  	tests := []th.MultiReturnEvalTestItem{
    28  		{
    29  			"limit(metric1,2)",
    30  			map[parser.MetricRequest][]*types.MetricData{
    31  				{Metric: "metric1", From: 0, Until: 1}: {
    32  					types.MakeMetricData("metricA", []float64{0, 1, 0, 0, 0, 0}, 1, now32),
    33  					types.MakeMetricData("metricB", []float64{0, 0, 1, 0, 0, 0}, 1, now32),
    34  					types.MakeMetricData("metricC", []float64{0, 0, 0, 1, 0, 0}, 1, now32),
    35  					types.MakeMetricData("metricD", []float64{0, 0, 0, 0, 1, 0}, 1, now32),
    36  					types.MakeMetricData("metricE", []float64{0, 0, 0, 0, 0, 1}, 1, now32),
    37  				},
    38  			},
    39  			"limit",
    40  			map[string][]*types.MetricData{
    41  				"metricA": {types.MakeMetricData("metricA", []float64{0, 1, 0, 0, 0, 0}, 1, now32)},
    42  				"metricB": {types.MakeMetricData("metricB", []float64{0, 0, 1, 0, 0, 0}, 1, now32)},
    43  			},
    44  		},
    45  		{
    46  			"limit(metric1,20)",
    47  			map[parser.MetricRequest][]*types.MetricData{
    48  				{Metric: "metric1", From: 0, Until: 1}: {
    49  					types.MakeMetricData("metricA", []float64{0, 1, 0, 0, 0, 0}, 1, now32),
    50  					types.MakeMetricData("metricB", []float64{0, 0, 1, 0, 0, 0}, 1, now32),
    51  					types.MakeMetricData("metricC", []float64{0, 0, 0, 1, 0, 0}, 1, now32),
    52  					types.MakeMetricData("metricD", []float64{0, 0, 0, 0, 1, 0}, 1, now32),
    53  					types.MakeMetricData("metricE", []float64{0, 0, 0, 0, 0, 1}, 1, now32),
    54  				},
    55  			},
    56  			"limit",
    57  			map[string][]*types.MetricData{
    58  				"metricA": {types.MakeMetricData("metricA", []float64{0, 1, 0, 0, 0, 0}, 1, now32)},
    59  				"metricB": {types.MakeMetricData("metricB", []float64{0, 0, 1, 0, 0, 0}, 1, now32)},
    60  				"metricC": {types.MakeMetricData("metricC", []float64{0, 0, 0, 1, 0, 0}, 1, now32)},
    61  				"metricD": {types.MakeMetricData("metricD", []float64{0, 0, 0, 0, 1, 0}, 1, now32)},
    62  				"metricE": {types.MakeMetricData("metricE", []float64{0, 0, 0, 0, 0, 1}, 1, now32)},
    63  			},
    64  		},
    65  	}
    66  
    67  	for _, tt := range tests {
    68  		testName := tt.Target
    69  		t.Run(testName, func(t *testing.T) {
    70  			eval := th.EvaluatorFromFunc(md[0].F)
    71  			th.TestMultiReturnEvalExpr(t, eval, &tt)
    72  		})
    73  	}
    74  }