diff --git a/troll_shield_test.go b/troll_shield_test.go index 726072f..6737d41 100644 --- 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) {