Add error case to TestKickTroll

commonlispbr · Jul 28, 2020 · 1398ef0 · 1398ef0
1 parent 641d00c
commit 1398ef0
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/troll_shield_test.go b/troll_shield_test.go
@@ -26,7 +26,13 @@ func (bot *BotMockup) GetChatMember(c telegram.ChatConfigWithUser) (telegram.Cha
 }
 
 func (bot *BotMockup) KickChatMember(c telegram.KickChatMemberConfig) (telegram.APIResponse, error) {
-	return telegram.APIResponse{Ok: true}, nil
+	switch c.ChatMemberConfig.UserID {
+	case 0:
+		return telegram.APIResponse{Ok: true}, nil
+	default:
+		return telegram.APIResponse{Ok: false}, errors.New("error")
+	}
+
 }
 
 func (bot *BotMockup) Send(c telegram.Chattable) (telegram.Message, error) {
@@ -137,6 +143,8 @@ func TestKickTroll(t *testing.T) {
 	update.Message = &message
 	user := telegram.User{}
 	kickTroll(&botnilson, &update, user, "@trollhouse")
+	user.ID = 1
+	kickTroll(&botnilson, &update, user, "@trollhouse")
 }
 
 func TestWelcomeMessage(t *testing.T) {