Cloudpods/backend/pkg/monitor/tsdb/driver/influxdb/query_part_test.go

// Copyright 2019 Yunion
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package influxdb

import (
	"testing"

	"yunion.io/x/onecloud/pkg/monitor/tsdb"
)

func TestInfluxdbQueryPart(t *testing.T) {
	tcs := []struct {
		mode     string
		input    string
		params   []string
		expected string
	}{
		{mode: "field", params: []string{"value"}, input: "value", expected: `"value"`},
		{mode: "derivative", params: []string{"10s"}, input: "mean(value)", expected: `derivative(mean(value), 10s)`},
		{mode: "bottom", params: []string{"3"}, input: "value", expected: `bottom(value, 3)`},
		{mode: "time", params: []string{"$interval"}, input: "", expected: `time($interval)`},
		{mode: "time", params: []string{"auto"}, input: "", expected: `time($__interval)`},
		{mode: "spread", params: []string{}, input: "value", expected: `spread(value)`},
		{mode: "math", params: []string{"/ 100"}, input: "mean(value)", expected: `mean(value) / 100`},
		{mode: "alias", params: []string{"test"}, input: "mean(value)", expected: `mean(value) AS "test"`},
		{mode: "count", params: []string{}, input: "distinct(value)", expected: `count(distinct(value))`},
		{mode: "mode", params: []string{}, input: "value", expected: `mode(value)`},
		{mode: "cumulative_sum", params: []string{}, input: "mean(value)", expected: `cumulative_sum(mean(value))`},
		{mode: "non_negative_difference", params: []string{}, input: "max(value)", expected: `non_negative_difference(max(value))`},
	}

	queryCtx := &tsdb.TsdbQuery{TimeRange: tsdb.NewTimeRange("5m", "now")}
	query := &Query{}

	for _, tc := range tcs {
		part, err := NewQueryPart(tc.mode, tc.params)
		if err != nil {
			t.Errorf("Expected NewQueryPart to not return an error. error: %v", err)
		}

		res := part.Render(query, queryCtx, tc.input)
		if res != tc.expected {
			t.Errorf("expected %v to render into %s", tc, tc.expected)
		}
	}
}