astra/telegram-bot-api
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add set message reaction helpers.

Browse source
This commit is contained in:
Ilja Lapkovskis 2024-01-20 18:17:17 +02:00
parent e41be9f718
commit a304e9002d
No known key found for this signature in database
GPG key ID: 53D2AA4F0D1079C4
3 changed files with 76 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
bot.go
View file

@ -746,3 +746,15 @@ func EscapeText(parseMode string, text string) string {
return replacer.Replace(text)
}
func (bot *BotAPI) SendReaction(config SetMessageReactionConfig) (Message, error) {
resp, err := bot.Request(config)
if err != nil {
return Message{}, err
}
var message Message
err = json.Unmarshal(resp.Result, &message)
return message, err
}

28
helper_methods.go
View file

@ -1141,3 +1141,31 @@ func ValidateWebAppData(token, telegramInitData string) (bool, error) {
return true, nil
}
func NewSetMessageReactionType(chatID int64, messageID int, reaction ReactionType, isBig bool) SetMessageReactionConfig {
return SetMessageReactionConfig{
BaseChatMessage: BaseChatMessage{
ChatConfig: ChatConfig{
ChatID: chatID,
},
MessageID: messageID,
},
Reaction: []ReactionType{reaction},
IsBig: isBig,
}
}
func NewSetMessageReactionEmoji(chatID int64, messageID int, reaction string, isBig bool) SetMessageReactionConfig {
return SetMessageReactionConfig{
BaseChatMessage: BaseChatMessage{
ChatConfig: ChatConfig{
ChatID: chatID,
},
MessageID: messageID,
},
Reaction: []ReactionType{
{Type: "emoji", Emoji: reaction},
},
IsBig: isBig,
}
}

37
tests/bot_integration_test.go
View file

@ -845,7 +845,6 @@ func TestCommands(t *testing.T) {
require.Equal(t, "a private command", commands[0].Description)
}
// TODO: figure out why test is failing
func TestEditMessageMedia(t *testing.T) {
bot, err := getBot(t)
require.NoError(t, err)
@ -872,3 +871,39 @@ func TestEditMessageMedia(t *testing.T) {
require.NoError(t, err)
require.NotNil(t, res)
}
func TestSetReaction(t *testing.T) {
bot, err := getBot(t)
require.NoError(t, err)
t.Run("set reaction using reaction type", func(t *testing.T) {
msg := tgbotapi.NewMessage(ChatID, "An initial message to test reaction type")
msg.ParseMode = tgbotapi.ModeMarkdown
m, err := bot.Send(msg)
require.NoError(t, err)
require.NotNil(t, m)
reaction := tgbotapi.NewSetMessageReactionType(ChatID, m.MessageID, tgbotapi.ReactionType{
Type: "emoji",
Emoji: "👍",
}, true)
res, err := bot.Request(reaction)
require.NoError(t, err)
require.NotNil(t, res)
})
t.Run("set reaction using reaction emoji", func(t *testing.T) {
msg := tgbotapi.NewMessage(ChatID, "An initial message to test reaction emoji")
msg.ParseMode = tgbotapi.ModeMarkdown
m, err := bot.Send(msg)
require.NoError(t, err)
require.NotNil(t, m)
reaction := tgbotapi.NewSetMessageReactionEmoji(ChatID, m.MessageID, "👀", true)
res, err := bot.Request(reaction)
require.NoError(t, err)
require.NotNil(t, res)
})
}