blob: a3e66fd1824d4caa1c913ddc3b5643bcc7734965 [file] [log] [blame]
package notifier
import (
type sentMessage struct {
subject string
msg *Message
type testNotifier struct {
sent []*sentMessage
func (n *testNotifier) Send(_ context.Context, subject string, msg *Message) error {
n.sent = append(n.sent, &sentMessage{
subject: subject,
msg: msg,
return nil
func TestRouter(t *testing.T) {
m := NewRouter(nil, nil, nil)
ctx := context.Background()
n1 := &testNotifier{}
m.Add(n1, FILTER_DEBUG, nil, "")
n2 := &testNotifier{}
m.Add(n2, FILTER_WARNING, nil, "")
n3 := &testNotifier{}
m.Add(n3, Filter(0), []string{"whitelisted type"}, "")
require.NoError(t, m.Send(ctx, &Message{
Subject: "Hi!",
Body: "Message body",
Type: "my-msg-type",
require.Equal(t, 1, len(n1.sent))
require.Equal(t, "Hi!", n1.sent[0].subject)
require.Equal(t, "Message body", n1.sent[0].msg.Body)
require.Equal(t, 0, len(n2.sent))
require.Equal(t, 0, len(n3.sent))
n4 := &testNotifier{}
m.Add(n4, FILTER_INFO, nil, "One subject to rule them all")
require.NoError(t, m.Send(ctx, &Message{
Subject: "My subject",
Body: "Second Message",
Type: "whitelisted type",
require.Equal(t, 1, len(n4.sent))
require.Equal(t, "One subject to rule them all", n4.sent[0].subject)
require.Equal(t, "Second Message", n4.sent[0].msg.Body)
require.Equal(t, 1, len(n3.sent))
require.Equal(t, "My subject", n3.sent[0].subject)
require.Equal(t, "Second Message", n3.sent[0].msg.Body)