influxdb/notification/check/deadman_test.go

package check_test

import (
	"testing"

	"github.com/influxdata/influxdb/v2"
	"github.com/influxdata/influxdb/v2/notification"
	"github.com/influxdata/influxdb/v2/notification/check"
)

func TestDeadman_GenerateFlux(t *testing.T) {
	type args struct {
		deadman check.Deadman
	}
	type wants struct {
		script string
	}

	tests := []struct {
		name  string
		args  args
		wants wants
	}{
		{
			name: "with aggregateWindow",
			args: args{
				deadman: check.Deadman{
					Base: check.Base{
						ID:   10,
						Name: "moo",
						Tags: []influxdb.Tag{
							{Key: "aaa", Value: "vaaa"},
							{Key: "bbb", Value: "vbbb"},
						},
						Every:                 mustDuration("1h"),
						StatusMessageTemplate: "whoa! {r[\"dead\"]}",
						Query: influxdb.DashboardQuery{
							Text: `from(bucket: "foo") |> range(start: -1d, stop: now()) |> aggregateWindow(fn: mean, every: 1m) |> yield()`,
							BuilderConfig: influxdb.BuilderConfig{
								Tags: []struct {
									Key                   string   `json:"key"`
									Values                []string `json:"values"`
									AggregateFunctionType string   `json:"aggregateFunctionType"`
								}{
									{
										Key:                   "_field",
										Values:                []string{"usage_user"},
										AggregateFunctionType: "filter",
									},
								},
							},
						},
					},
					TimeSince: mustDuration("60s"),
					StaleTime: mustDuration("10m"),
					Level:     notification.Info,
				},
			},
			wants: wants{
				script: `package main
import "influxdata/influxdb/monitor"
import "experimental"
import "influxdata/influxdb/v1"

data = from(bucket: "foo")
	|> range(start: -10m)

option task = {name: "moo", every: 1h}

check = {
	_check_id: "000000000000000a",
	_check_name: "moo",
	_type: "deadman",
	tags: {aaa: "vaaa", bbb: "vbbb"},
}
info = (r) =>
	(r["dead"])
messageFn = (r) =>
	("whoa! {r[\"dead\"]}")

data
	|> v1["fieldsAsCols"]()
	|> monitor["deadman"](t: experimental["subDuration"](from: now(), d: 60s))
	|> monitor["check"](data: check, messageFn: messageFn, info: info)`,
			},
		},
		{
			name: "basic",
			args: args{
				deadman: check.Deadman{
					Base: check.Base{
						ID:   10,
						Name: "moo",
						Tags: []influxdb.Tag{
							{Key: "aaa", Value: "vaaa"},
							{Key: "bbb", Value: "vbbb"},
						},
						Every:                 mustDuration("1h"),
						StatusMessageTemplate: "whoa! {r[\"dead\"]}",
						Query: influxdb.DashboardQuery{
							Text: `from(bucket: "foo") |> range(start: -1d, stop: now()) |> yield()`,
							BuilderConfig: influxdb.BuilderConfig{
								Tags: []struct {
									Key                   string   `json:"key"`
									Values                []string `json:"values"`
									AggregateFunctionType string   `json:"aggregateFunctionType"`
								}{
									{
										Key:                   "_field",
										Values:                []string{"usage_user"},
										AggregateFunctionType: "filter",
									},
								},
							},
						},
					},
					TimeSince: mustDuration("60s"),
					StaleTime: mustDuration("10m"),
					Level:     notification.Info,
				},
			},
			wants: wants{
				script: `package main
import "influxdata/influxdb/monitor"
import "experimental"
import "influxdata/influxdb/v1"

data = from(bucket: "foo")
	|> range(start: -10m)

option task = {name: "moo", every: 1h}

check = {
	_check_id: "000000000000000a",
	_check_name: "moo",
	_type: "deadman",
	tags: {aaa: "vaaa", bbb: "vbbb"},
}
info = (r) =>
	(r["dead"])
messageFn = (r) =>
	("whoa! {r[\"dead\"]}")

data
	|> v1["fieldsAsCols"]()
	|> monitor["deadman"](t: experimental["subDuration"](from: now(), d: 60s))
	|> monitor["check"](data: check, messageFn: messageFn, info: info)`,
			},
		},
		{
			name: "basic with space in field name",
			args: args{
				deadman: check.Deadman{
					Base: check.Base{
						ID:   10,
						Name: "moo",
						Tags: []influxdb.Tag{
							{Key: "aaa", Value: "vaaa"},
							{Key: "bbb", Value: "vbbb"},
						},
						Every:                 mustDuration("1h"),
						StatusMessageTemplate: "whoa! {r[\"dead\"]}",
						Query: influxdb.DashboardQuery{
							Text: `from(bucket: "foo") |> range(start: -1d, stop: now()) |> filter(fn: (r) => r._field == "usage user") |> yield()`,
							BuilderConfig: influxdb.BuilderConfig{
								Tags: []struct {
									Key                   string   `json:"key"`
									Values                []string `json:"values"`
									AggregateFunctionType string   `json:"aggregateFunctionType"`
								}{
									{
										Key:                   "_field",
										Values:                []string{"usage_user"},
										AggregateFunctionType: "filter",
									},
								},
							},
						},
					},
					TimeSince: mustDuration("60s"),
					StaleTime: mustDuration("10m"),
					Level:     notification.Info,
				},
			},
			wants: wants{
				script: `package main
import "influxdata/influxdb/monitor"
import "experimental"
import "influxdata/influxdb/v1"

data = from(bucket: "foo")
	|> range(start: -10m)
	|> filter(fn: (r) =>
		(r._field == "usage user"))

option task = {name: "moo", every: 1h}

check = {
	_check_id: "000000000000000a",
	_check_name: "moo",
	_type: "deadman",
	tags: {aaa: "vaaa", bbb: "vbbb"},
}
info = (r) =>
	(r["dead"])
messageFn = (r) =>
	("whoa! {r[\"dead\"]}")

data
	|> v1["fieldsAsCols"]()
	|> monitor["deadman"](t: experimental["subDuration"](from: now(), d: 60s))
	|> monitor["check"](data: check, messageFn: messageFn, info: info)`,
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			s, err := tt.args.deadman.GenerateFlux()
			if err != nil {
				t.Fatalf("unexpected error: %v", err)
			}

			if exp, got := tt.wants.script, s; exp != got {
				t.Errorf("expected:\n%v\n\ngot:\n%v\n", exp, got)
			}
		})
	}

}