github.com/1aal/kubeblocks@v0.0.0-20231107070852-e1c03e598921/pkg/cli/cmd/alert/add_receiver_test.go

/*
Copyright (C) 2022-2023 ApeCloud Co., Ltd

This file is part of KubeBlocks project

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

package alert

import (
	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"

	corev1 "k8s.io/api/core/v1"
	"k8s.io/cli-runtime/pkg/genericiooptions"
	clientfake "k8s.io/client-go/rest/fake"
	cmdtesting "k8s.io/kubectl/pkg/cmd/testing"

	"github.com/1aal/kubeblocks/pkg/cli/testing"
)

const (
	testNamespace = "test"
)

var mockConfigmap = func(name string, key string, value string) *corev1.ConfigMap {
	cm := &corev1.ConfigMap{}
	cm.Name = name
	cm.Namespace = testNamespace
	cm.Data = map[string]string{key: value}
	return cm
}

var mockBaseOptions = func(s genericiooptions.IOStreams) baseOptions {
	o := baseOptions{IOStreams: s}
	alertManagerConfig := `
    global:
      smtp_from: user@kubeblocks.io
      smtp_smarthost: smtp.feishu.cn:587
      smtp_auth_username: admin@kubeblocks.io
      smtp_auth_password: 123456abc
      smtp_auth_identity: admin@kubeblocks.io
    receivers:
    - name: default-receiver
    - name: receiver-7pb52
      webhook_configs:
      - max_alerts: 10
        url: http://kubeblocks-webhook-adaptor-config.default:5001/api/v1/notify/receiver-7pb52
    route:
      group_interval: 30s
      group_wait: 5s
      receiver: default-receiver
      repeat_interval: 10m
      routes:
      - continue: true
        matchers:
        - app_kubernetes_io_instance=~a|b|c
        - severity=~info|warning
        receiver: receiver-7pb52`
	webhookAdaptorConfig := `
    receivers:
    - name: receiver-7pb52
      params:
        url: https://oapi.dingtalk.com/robot/send?access_token=123456
      type: dingtalk-webhook`
	alertCM := mockConfigmap(alertConfigmapName, alertConfigFileName, alertManagerConfig)
	webhookAdaptorCM := mockConfigmap(webhookAdaptorConfigmapName, webhookAdaptorFileName, webhookAdaptorConfig)
	o.alertConfigMap = alertCM
	o.webhookConfigMap = webhookAdaptorCM
	return o
}

var _ = Describe("add receiver", func() {
	var f *cmdtesting.TestFactory
	var s genericiooptions.IOStreams

	BeforeEach(func() {
		f = cmdtesting.NewTestFactory()
		f.Client = &clientfake.RESTClient{}
		s, _, _, _ = genericiooptions.NewTestIOStreams()
	})

	AfterEach(func() {
		f.Cleanup()
	})

	It("create new add receiver cmd", func() {
		cmd := newAddReceiverCmd(f, s)
		Expect(cmd).NotTo(BeNil())
		Expect(cmd.HasSubCommands()).Should(BeFalse())
	})

	It("complete", func() {
		o := baseOptions{IOStreams: s}
		Expect(o.complete(f)).Should(HaveOccurred())
	})

	It("validate", func() {
		By("nothing to be input, should fail")
		o := addReceiverOptions{baseOptions: baseOptions{IOStreams: s}}
		Expect(o.validate([]string{})).Should(HaveOccurred())

		By("set email, do not specify the name")
		o.emails = []string{"user@kubeblocks.io"}
		o.alertConfigMap = mockConfigmap(alertConfigmapName, alertConfigFileName, "")
		Expect(o.validate([]string{})).Should(HaveOccurred())
		Expect(o.name).ShouldNot(BeEmpty())

		By("set email, specify the name")
		Expect(o.validate([]string{"test"})).Should(HaveOccurred())
		Expect(o.name).Should(Equal("test"))

		By("set email, set smtp config in configmap")
		baseOptions := mockBaseOptions(s)
		o.alertConfigMap = baseOptions.alertConfigMap
		Expect(o.validate([]string{})).Should(Succeed())
	})

	It("build receiver", func() {
		o := addReceiverOptions{baseOptions: baseOptions{IOStreams: s}}
		o.emails = []string{"user@kubeblocks.io", "user1@kubeblocks.io,user2@kubeblocks.io"}
		o.webhooks = []string{"url=https://oapi.dingtalk.com/robot/send", "url=https://oapi.dingtalk.com/robot/send,url=https://oapi.dingtalk.com/robot/send?"}
		o.slacks = []string{"api_url=https://foo.com,channel=foo,username=test"}
		o.webhookConfigMap = mockConfigmap(webhookAdaptorConfigmapName, webhookAdaptorFileName, "")
		Expect(o.buildReceiver()).Should(Succeed())
		Expect(o.receiver).ShouldNot(BeNil())
		Expect(o.receiver.EmailConfigs).Should(HaveLen(3))
		Expect(o.receiver.WebhookConfigs).Should(HaveLen(2))
		Expect(o.receiver.SlackConfigs).Should(HaveLen(1))
	})

	It("build routes", func() {
		o := addReceiverOptions{baseOptions: baseOptions{IOStreams: s}}
		o.name = "receiver-test"
		o.clusters = []string{"cluster1", "cluster2"}
		o.severities = []string{"critical", "warning"}
		o.buildRoute()
		Expect(o.route).ShouldNot(BeNil())
		Expect(o.route.Receiver).Should(Equal(o.name))
		Expect(o.route.Matchers).Should(HaveLen(2))
		Expect(o.route.Matchers[0]).Should(ContainSubstring(routeMatcherClusterKey))
		Expect(o.route.Matchers[1]).Should(ContainSubstring(routeMatcherSeverityKey))
	})

	It("run", func() {
		o := addReceiverOptions{baseOptions: baseOptions{IOStreams: s}}
		alertCM := mockConfigmap(alertConfigmapName, alertConfigFileName, "")
		webhookAdaptorCM := mockConfigmap(webhookAdaptorConfigmapName, webhookAdaptorFileName, "")
		o.baseOptions.alertConfigMap = alertCM
		o.baseOptions.webhookConfigMap = webhookAdaptorCM
		o.client = testing.FakeClientSet(alertCM, webhookAdaptorCM)
		o.name = "receiver-test"
		Expect(o.addReceiver()).Should(Succeed())
		Expect(o.addWebhookReceivers()).Should(Succeed())
	})
})