alertmanager-irc-relay/reconciler_test.go

101 lines
2.4 KiB
Go
Raw Normal View History

// Copyright 2021 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package main
import (
"bufio"
"context"
"reflect"
"sort"
"sync"
"testing"
irc "github.com/fluffle/goirc/client"
)
func makeReconcilerTestIRCConfig(IRCPort int) *Config {
config := makeTestIRCConfig(IRCPort)
config.IRCChannels = []IRCChannel{
IRCChannel{Name: "#foo"},
IRCChannel{Name: "#bar"},
IRCChannel{Name: "#baz"},
}
return config
}
func makeTestReconciler(config *Config) (*ChannelReconciler, chan bool, chan bool) {
sessionUp := make(chan bool)
sessionDown := make(chan bool)
client := irc.Client(makeGOIRCConfig(config))
client.Config().Flood = true
client.HandleFunc(irc.CONNECTED,
func(*irc.Conn, *irc.Line) {
sessionUp <- true
})
client.HandleFunc(irc.DISCONNECTED,
func(*irc.Conn, *irc.Line) {
sessionDown <- false
})
fakeDelayerMaker := &FakeDelayerMaker{}
reconciler := NewChannelReconciler(config, client, fakeDelayerMaker)
return reconciler, sessionUp, sessionDown
}
func TestPreJoinChannels(t *testing.T) {
server, port := makeTestServer(t)
config := makeReconcilerTestIRCConfig(port)
reconciler, sessionUp, sessionDown := makeTestReconciler(config)
var testStep sync.WaitGroup
joinedChannels := []string{}
joinHandler := func(conn *bufio.ReadWriter, line *irc.Line) error {
joinedChannels = append(joinedChannels, line.Args[0])
if len(joinedChannels) == 3 {
testStep.Done()
}
return hJOIN(conn, line)
}
server.SetHandler("JOIN", joinHandler)
testStep.Add(1)
reconciler.client.Connect()
<-sessionUp
reconciler.Start(context.Background())
testStep.Wait()
reconciler.client.Quit("see ya")
<-sessionDown
reconciler.Stop()
server.Stop()
expectedJoinedChannels := []string{"#bar", "#baz", "#foo"}
sort.Strings(joinedChannels)
if !reflect.DeepEqual(expectedJoinedChannels, joinedChannels) {
t.Error("Did not pre-join channels")
}
}