influxdb/notification/rule/slack_test.go

package rule_test

import (
	"testing"

	"github.com/influxdata/flux/parser"
	"github.com/influxdata/influxdb"
	"github.com/influxdata/influxdb/notification"
	"github.com/influxdata/influxdb/notification/endpoint"
	"github.com/influxdata/influxdb/notification/rule"
)

func mustDuration(d string) *notification.Duration {
	dur, err := parser.ParseDuration(d)
	if err != nil {
		panic(err)
	}

	return (*notification.Duration)(dur)
}

func statusRulePtr(r notification.CheckLevel) *notification.CheckLevel {
	return &r
}

func TestSlack_GenerateFlux(t *testing.T) {
	want := `package main
// foo
import "influxdata/influxdb/monitor"
import "slack"
import "influxdata/influxdb/secrets"
import "experimental"
import "influxdata/influxdb/v1"

option task = {name: "foo", every: 2h}

slack_secret = secrets.get(key: "slack_token")
slack_endpoint = slack.endpoint(token: slack_secret, url: "http://localhost:7777")
notification = {
	_notification_rule_id: "0000000000000001",
	_notification_rule_name: "foo",
	_notification_endpoint_id: "0000000000000002",
	_notification_endpoint_name: "foo",
}
statuses = monitor.from(start: -2h, fn: (r) =>
	(r.foo == "bar" and r.baz == "bang"))
	|> v1.fieldsAsCols()
any_to_crit = statuses
	|> monitor.stateChanges(fromLevel: "any", toLevel: "crit")
info_to_warn = statuses
	|> monitor.stateChanges(fromLevel: "info", toLevel: "warn")
all_statuses = union(tables: [any_to_crit, info_to_warn])
	|> sort(columns: ["_time"])
	|> filter(fn: (r) =>
		(r._time > experimental.subDuration(from: now(), d: 1h)))

all_statuses
	|> monitor.notify(data: notification, endpoint: slack_endpoint(mapFn: (r) =>
		({channel: "bar", text: "blah", color: if r._level == "crit" then "danger" else if r._level == "warn" then "warning" else "good"})))`

	s := &rule.Slack{
		Channel:         "bar",
		MessageTemplate: "blah",
		Base: rule.Base{
			ID:         1,
			EndpointID: 2,
			Name:       "foo",
			Every:      mustDuration("1h"),
			TagRules: []notification.TagRule{
				{
					Tag: influxdb.Tag{
						Key:   "foo",
						Value: "bar",
					},
					Operator: notification.Equal,
				},
				{
					Tag: influxdb.Tag{
						Key:   "baz",
						Value: "bang",
					},
					Operator: notification.Equal,
				},
			},
			StatusRules: []notification.StatusRule{
				{
					CurrentLevel: notification.Critical,
				},
				{
					CurrentLevel:  notification.Warn,
					PreviousLevel: statusRulePtr(notification.Info),
				},
			},
		},
	}
	e := &endpoint.Slack{
		Base: endpoint.Base{
			ID:   2,
			Name: "foo",
		},
		URL: "http://localhost:7777",
		Token: influxdb.SecretField{
			Key: "slack_token",
		},
	}

	f, err := s.GenerateFlux(e)
	if err != nil {
		panic(err)
	}

	if f != want {
		t.Errorf("scripts did not match. want:\n%v\n\ngot:\n%v", want, f)
	}
}