diff --git a/troll_shield_test.go b/troll_shield_test.go index e92518d..ba9c9ba 100644 --- a/troll_shield_test.go +++ b/troll_shield_test.go @@ -93,3 +93,24 @@ func TestEvents(t *testing.T) { } } + +func TestFindTrollHouses(t *testing.T) { + botnilson := BotMockup{} + trollGroups = []string{"@rolisvaldo"} + if got := findTrollHouses(&botnilson, 1); got != "@rolisvaldo" { + t.Errorf("findTrollHouses expects @rolisvaldo, got: %v", got) + } + if got := findTrollHouses(&botnilson, 2); got != "@rolisvaldo" { + t.Errorf("findTrollHouses expects @rolisvaldo, got: %v", got) + } + if got := findTrollHouses(&botnilson, 3); got != "@rolisvaldo" { + t.Errorf("findTrollHouses expects @rolisvaldo, got: %v", got) + } + if got := findTrollHouses(&botnilson, 4); got != "" { + t.Errorf("findTrollHouses expects empty string, got: %v", got) + } + if got := findTrollHouses(&botnilson, -1); got != "" { + t.Errorf("findTrollHouses expects empty string, got: %v", got) + } +} +