VictoriaMetrics/app/vmctl/influx/influx_test.go

package influx

import "testing"

func TestFetchQuery(t *testing.T) {
	testCases := []struct {
		s          Series
		timeFilter string
		expected   string
	}{
		{
			s: Series{
				Measurement: "cpu",
				Field:       "value",
				LabelPairs: []LabelPair{
					{
						Name:  "foo",
						Value: "bar",
					},
				},
			},
			expected: `select "value" from "cpu" where "foo"::tag='bar'`,
		},
		{
			s: Series{
				Measurement: "cpu",
				Field:       "value",
				LabelPairs: []LabelPair{
					{
						Name:  "foo",
						Value: "bar",
					},
					{
						Name:  "baz",
						Value: "qux",
					},
				},
			},
			expected: `select "value" from "cpu" where "foo"::tag='bar' and "baz"::tag='qux'`,
		},
		{
			s: Series{
				Measurement: "cpu",
				Field:       "value",
				LabelPairs: []LabelPair{
					{
						Name:  "foo",
						Value: "b'ar",
					},
				},
			},
			timeFilter: "time >= now()",
			expected:   `select "value" from "cpu" where "foo"::tag='b\'ar' and time >= now()`,
		},
		{
			s: Series{
				Measurement: "cpu",
				Field:       "value",
				LabelPairs: []LabelPair{
					{
						Name:  "name",
						Value: `dev-mapper-centos\x2dswap.swap`,
					},
					{
						Name:  "state",
						Value: "dev-mapp'er-c'en'tos",
					},
				},
			},
			timeFilter: "time >= now()",
			expected:   `select "value" from "cpu" where "name"::tag='dev-mapper-centos\\x2dswap.swap' and "state"::tag='dev-mapp\'er-c\'en\'tos' and time >= now()`,
		},
		{
			s: Series{
				Measurement: "cpu",
				Field:       "value",
			},
			timeFilter: "time >= now()",
			expected:   `select "value" from "cpu" where time >= now()`,
		},
		{
			s: Series{
				Measurement: "cpu",
				Field:       "value",
			},
			expected: `select "value" from "cpu"`,
		},
	}

	for _, tc := range testCases {
		query := tc.s.fetchQuery(tc.timeFilter)
		if query != tc.expected {
			t.Fatalf("got: \n%s;\nexpected: \n%s", query, tc.expected)
		}
	}
}

func TestTimeFilter(t *testing.T) {
	testCases := []struct {
		start    string
		end      string
		expected string
	}{
		{
			start:    "2020-01-01T20:07:00Z",
			end:      "2020-01-01T21:07:00Z",
			expected: "time >= '2020-01-01T20:07:00Z' and time <= '2020-01-01T21:07:00Z'",
		},
		{
			expected: "",
		},
		{
			start:    "2020-01-01T20:07:00Z",
			expected: "time >= '2020-01-01T20:07:00Z'",
		},
		{
			end:      "2020-01-01T21:07:00Z",
			expected: "time <= '2020-01-01T21:07:00Z'",
		},
	}
	for _, tc := range testCases {
		f := timeFilter(tc.start, tc.end)
		if f != tc.expected {
			t.Fatalf("got: \n%q;\nexpected: \n%q", f, tc.expected)
		}
	}
}