package gateway import ( "fmt" "github.com/42wim/matterbridge/bridge/config" "github.com/BurntSushi/toml" "github.com/stretchr/testify/assert" "strconv" "testing" ) var testconfig = ` [irc.freenode] [mattermost.test] [gitter.42wim] [discord.test] [slack.test] [[gateway]] name = "bridge1" enable=true [[gateway.inout]] account = "irc.freenode" channel = "#wimtesting" [[gateway.inout]] account="gitter.42wim" channel="42wim/testroom" #channel="matterbridge/Lobby" [[gateway.inout]] account = "discord.test" channel = "general" [[gateway.inout]] account="slack.test" channel="testing" ` var testconfig2 = ` [irc.freenode] [mattermost.test] [gitter.42wim] [discord.test] [slack.test] [[gateway]] name = "bridge1" enable=true [[gateway.in]] account = "irc.freenode" channel = "#wimtesting" [[gateway.in]] account="gitter.42wim" channel="42wim/testroom" [[gateway.inout]] account = "discord.test" channel = "general" [[gateway.out]] account="slack.test" channel="testing" [[gateway]] name = "bridge2" enable=true [[gateway.in]] account = "irc.freenode" channel = "#wimtesting2" [[gateway.out]] account="gitter.42wim" channel="42wim/testroom" [[gateway.out]] account = "discord.test" channel = "general2" ` var testconfig3 = ` [irc.zzz] [telegram.zzz] [slack.zzz] [[gateway]] name="bridge" enable=true [[gateway.inout]] account="irc.zzz" channel="#main" [[gateway.inout]] account="telegram.zzz" channel="-1111111111111" [[gateway.inout]] account="slack.zzz" channel="irc" [[gateway]] name="announcements" enable=true [[gateway.in]] account="telegram.zzz" channel="-2222222222222" [[gateway.out]] account="irc.zzz" channel="#main" [[gateway.out]] account="irc.zzz" channel="#main-help" [[gateway.out]] account="telegram.zzz" channel="--333333333333" [[gateway.out]] account="slack.zzz" channel="general" [[gateway]] name="bridge2" enable=true [[gateway.inout]] account="irc.zzz" channel="#main-help" [[gateway.inout]] account="telegram.zzz" channel="--444444444444" [[gateway]] name="bridge3" enable=true [[gateway.inout]] account="irc.zzz" channel="#main-telegram" [[gateway.inout]] account="telegram.zzz" channel="--333333333333" ` func maketestRouter(input string) *Router { var cfg *config.Config if _, err := toml.Decode(input, &cfg); err != nil { fmt.Println(err) } r, err := NewRouter(cfg) if err != nil { fmt.Println(err) } return r } func TestNewRouter(t *testing.T) { var cfg *config.Config if _, err := toml.Decode(testconfig, &cfg); err != nil { fmt.Println(err) } r, err := NewRouter(cfg) if err != nil { fmt.Println(err) } assert.Equal(t, 1, len(r.Gateways)) assert.Equal(t, 4, len(r.Gateways["bridge1"].Bridges)) assert.Equal(t, 4, len(r.Gateways["bridge1"].Channels)) r = maketestRouter(testconfig2) assert.Equal(t, 2, len(r.Gateways)) assert.Equal(t, 4, len(r.Gateways["bridge1"].Bridges)) assert.Equal(t, 3, len(r.Gateways["bridge2"].Bridges)) assert.Equal(t, 4, len(r.Gateways["bridge1"].Channels)) assert.Equal(t, 3, len(r.Gateways["bridge2"].Channels)) assert.Equal(t, &config.ChannelInfo{Name: "42wim/testroom", Direction: "out", ID: "42wim/testroomgitter.42wim", Account: "gitter.42wim", SameChannel: map[string]bool{"bridge2": false}}, r.Gateways["bridge2"].Channels["42wim/testroomgitter.42wim"]) assert.Equal(t, &config.ChannelInfo{Name: "42wim/testroom", Direction: "in", ID: "42wim/testroomgitter.42wim", Account: "gitter.42wim", SameChannel: map[string]bool{"bridge1": false}}, r.Gateways["bridge1"].Channels["42wim/testroomgitter.42wim"]) assert.Equal(t, &config.ChannelInfo{Name: "general", Direction: "inout", ID: "generaldiscord.test", Account: "discord.test", SameChannel: map[string]bool{"bridge1": false}}, r.Gateways["bridge1"].Channels["generaldiscord.test"]) } func TestGetDestChannel(t *testing.T) { r := maketestRouter(testconfig2) msg := &config.Message{Text: "test", Channel: "general", Account: "discord.test", Gateway: "bridge1", Protocol: "discord", Username: "test"} for _, br := range r.Gateways["bridge1"].Bridges { switch br.Account { case "discord.test": assert.Equal(t, []config.ChannelInfo{config.ChannelInfo{Name: "general", Account: "discord.test", Direction: "inout", ID: "generaldiscord.test", SameChannel: map[string]bool{"bridge1": false}, Options: config.ChannelOptions{Key: ""}}}, r.Gateways["bridge1"].getDestChannel(msg, *br)) case "slack.test": assert.Equal(t, []config.ChannelInfo{config.ChannelInfo{Name: "testing", Account: "slack.test", Direction: "out", ID: "testingslack.test", SameChannel: map[string]bool{"bridge1": false}, Options: config.ChannelOptions{Key: ""}}}, r.Gateways["bridge1"].getDestChannel(msg, *br)) case "gitter.42wim": assert.Equal(t, []config.ChannelInfo(nil), r.Gateways["bridge1"].getDestChannel(msg, *br)) case "irc.freenode": assert.Equal(t, []config.ChannelInfo(nil), r.Gateways["bridge1"].getDestChannel(msg, *br)) } } } func TestGetDestChannelAdvanced(t *testing.T) { r := maketestRouter(testconfig3) var msgs []*config.Message i := 0 for _, gw := range r.Gateways { for _, channel := range gw.Channels { msgs = append(msgs, &config.Message{Text: "text" + strconv.Itoa(i), Channel: channel.Name, Account: channel.Account, Gateway: gw.Name, Username: "user" + strconv.Itoa(i)}) i++ } } hits := make(map[string]int) for _, gw := range r.Gateways { for _, br := range gw.Bridges { for _, msg := range msgs { channels := gw.getDestChannel(msg, *br) if gw.Name != msg.Gateway { assert.Equal(t, []config.ChannelInfo(nil), channels) continue } switch gw.Name { case "bridge": if (msg.Channel == "#main" || msg.Channel == "-1111111111111" || msg.Channel == "irc") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz" || msg.Account == "slack.zzz") { hits[gw.Name]++ switch br.Account { case "irc.zzz": assert.Equal(t, []config.ChannelInfo{config.ChannelInfo{Name: "#main", Account: "irc.zzz", Direction: "inout", ID: "#mainirc.zzz", SameChannel: map[string]bool{"bridge": false}, Options: config.ChannelOptions{Key: ""}}}, channels) case "telegram.zzz": assert.Equal(t, []config.ChannelInfo{config.ChannelInfo{Name: "-1111111111111", Account: "telegram.zzz", Direction: "inout", ID: "-1111111111111telegram.zzz", SameChannel: map[string]bool{"bridge": false}, Options: config.ChannelOptions{Key: ""}}}, channels) case "slack.zzz": assert.Equal(t, []config.ChannelInfo{config.ChannelInfo{Name: "irc", Account: "slack.zzz", Direction: "inout", ID: "ircslack.zzz", SameChannel: map[string]bool{"bridge": false}, Options: config.ChannelOptions{Key: ""}}}, channels) } } case "bridge2": if (msg.Channel == "#main-help" || msg.Channel == "--444444444444") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz") { hits[gw.Name]++ switch br.Account { case "irc.zzz": assert.Equal(t, []config.ChannelInfo{config.ChannelInfo{Name: "#main-help", Account: "irc.zzz", Direction: "inout", ID: "#main-helpirc.zzz", SameChannel: map[string]bool{"bridge2": false}, Options: config.ChannelOptions{Key: ""}}}, channels) case "telegram.zzz": assert.Equal(t, []config.ChannelInfo{config.ChannelInfo{Name: "--444444444444", Account: "telegram.zzz", Direction: "inout", ID: "--444444444444telegram.zzz", SameChannel: map[string]bool{"bridge2": false}, Options: config.ChannelOptions{Key: ""}}}, channels) } } case "bridge3": if (msg.Channel == "#main-telegram" || msg.Channel == "--333333333333") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz") { hits[gw.Name]++ switch br.Account { case "irc.zzz": assert.Equal(t, []config.ChannelInfo{config.ChannelInfo{Name: "#main-telegram", Account: "irc.zzz", Direction: "inout", ID: "#main-telegramirc.zzz", SameChannel: map[string]bool{"bridge3": false}, Options: config.ChannelOptions{Key: ""}}}, channels) case "telegram.zzz": assert.Equal(t, []config.ChannelInfo{config.ChannelInfo{Name: "--333333333333", Account: "telegram.zzz", Direction: "inout", ID: "--333333333333telegram.zzz", SameChannel: map[string]bool{"bridge3": false}, Options: config.ChannelOptions{Key: ""}}}, channels) } } case "announcements": if msg.Channel != "-2222222222222" && msg.Account != "telegram" { assert.Equal(t, []config.ChannelInfo(nil), channels) continue } hits[gw.Name]++ switch br.Account { case "irc.zzz": assert.Equal(t, []config.ChannelInfo{config.ChannelInfo{Name: "#main", Account: "irc.zzz", Direction: "out", ID: "#mainirc.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}, config.ChannelInfo{Name: "#main-help", Account: "irc.zzz", Direction: "out", ID: "#main-helpirc.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}}, channels) case "slack.zzz": assert.Equal(t, []config.ChannelInfo{config.ChannelInfo{Name: "general", Account: "slack.zzz", Direction: "out", ID: "generalslack.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}}, channels) case "telegram.zzz": assert.Equal(t, []config.ChannelInfo{config.ChannelInfo{Name: "--333333333333", Account: "telegram.zzz", Direction: "out", ID: "--333333333333telegram.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}}, channels) } } } } } assert.Equal(t, map[string]int{"bridge3": 4, "bridge": 9, "announcements": 3, "bridge2": 4}, hits) }