1052 lines
20 KiB
Go
1052 lines
20 KiB
Go
package tgbotapi
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
TestToken = "153667468:AAHlSHlMqSt1f_uFmVRJbm5gntu2HI4WW8I"
|
|
ChatID = 76918703
|
|
Channel = "@tgbotapitest"
|
|
SupergroupChatID = -1001120141283
|
|
ReplyToMessageID = 35
|
|
ExistingPhotoFileID = "AgACAgIAAxkDAAEBFUZhIALQ9pZN4BUe8ZSzUU_2foSo1AACnrMxG0BucEhezsBWOgcikQEAAwIAA20AAyAE"
|
|
ExistingDocumentFileID = "BQADAgADOQADjMcoCcioX1GrDvp3Ag"
|
|
ExistingAudioFileID = "BQADAgADRgADjMcoCdXg3lSIN49lAg"
|
|
ExistingVoiceFileID = "AwADAgADWQADjMcoCeul6r_q52IyAg"
|
|
ExistingVideoFileID = "BAADAgADZgADjMcoCav432kYe0FRAg"
|
|
ExistingVideoNoteFileID = "DQADAgADdQAD70cQSUK41dLsRMqfAg"
|
|
ExistingStickerFileID = "BQADAgADcwADjMcoCbdl-6eB--YPAg"
|
|
)
|
|
|
|
type testLogger struct {
|
|
t *testing.T
|
|
}
|
|
|
|
func (t testLogger) Println(v ...interface{}) {
|
|
t.t.Log(v...)
|
|
}
|
|
|
|
func (t testLogger) Printf(format string, v ...interface{}) {
|
|
t.t.Logf(format, v...)
|
|
}
|
|
|
|
func getBot(t *testing.T) (*BotAPI, error) {
|
|
bot, err := NewBotAPI(TestToken)
|
|
bot.Debug = true
|
|
|
|
logger := testLogger{t}
|
|
SetLogger(logger)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
return bot, err
|
|
}
|
|
|
|
func TestNewBotAPI_notoken(t *testing.T) {
|
|
_, err := NewBotAPI("")
|
|
|
|
if err == nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestGetUpdates(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
u := NewUpdate(0)
|
|
|
|
_, err := bot.GetUpdates(u)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithMessage(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewMessage(ChatID, "A test message from the test library in telegram-bot-api")
|
|
msg.ParseMode = ModeMarkdown
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithMessageReply(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewMessage(ChatID, "A test message from the test library in telegram-bot-api")
|
|
msg.ReplyToMessageID = ReplyToMessageID
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithMessageForward(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewForward(ChatID, ChatID, ReplyToMessageID)
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestCopyMessage(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewMessage(ChatID, "A test message from the test library in telegram-bot-api")
|
|
message, err := bot.Send(msg)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
copyMessageConfig := NewCopyMessage(SupergroupChatID, message.Chat.ID, message.MessageID)
|
|
messageID, err := bot.CopyMessage(copyMessageConfig)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if messageID.MessageID == message.MessageID {
|
|
t.Error("copied message ID was the same as original message")
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewPhoto(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewPhoto(ChatID, FilePath("tests/image.jpg"))
|
|
msg.Caption = "Test"
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewPhotoWithFileBytes(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
data, _ := ioutil.ReadFile("tests/image.jpg")
|
|
b := FileBytes{Name: "image.jpg", Bytes: data}
|
|
|
|
msg := NewPhoto(ChatID, b)
|
|
msg.Caption = "Test"
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewPhotoWithFileReader(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
f, _ := os.Open("tests/image.jpg")
|
|
reader := FileReader{Name: "image.jpg", Reader: f}
|
|
|
|
msg := NewPhoto(ChatID, reader)
|
|
msg.Caption = "Test"
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewPhotoReply(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewPhoto(ChatID, FilePath("tests/image.jpg"))
|
|
msg.ReplyToMessageID = ReplyToMessageID
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendNewPhotoToChannel(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewPhotoToChannel(Channel, FilePath("tests/image.jpg"))
|
|
msg.Caption = "Test"
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendNewPhotoToChannelFileBytes(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
data, _ := ioutil.ReadFile("tests/image.jpg")
|
|
b := FileBytes{Name: "image.jpg", Bytes: data}
|
|
|
|
msg := NewPhotoToChannel(Channel, b)
|
|
msg.Caption = "Test"
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendNewPhotoToChannelFileReader(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
f, _ := os.Open("tests/image.jpg")
|
|
reader := FileReader{Name: "image.jpg", Reader: f}
|
|
|
|
msg := NewPhotoToChannel(Channel, reader)
|
|
msg.Caption = "Test"
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingPhoto(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewPhoto(ChatID, FileID(ExistingPhotoFileID))
|
|
msg.Caption = "Test"
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewDocument(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewDocument(ChatID, FilePath("tests/image.jpg"))
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewDocumentAndThumb(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewDocument(ChatID, FilePath("tests/voice.ogg"))
|
|
msg.Thumb = FilePath("tests/image.jpg")
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingDocument(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewDocument(ChatID, FileID(ExistingDocumentFileID))
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewAudio(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewAudio(ChatID, FilePath("tests/audio.mp3"))
|
|
msg.Title = "TEST"
|
|
msg.Duration = 10
|
|
msg.Performer = "TEST"
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingAudio(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewAudio(ChatID, FileID(ExistingAudioFileID))
|
|
msg.Title = "TEST"
|
|
msg.Duration = 10
|
|
msg.Performer = "TEST"
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewVoice(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewVoice(ChatID, FilePath("tests/voice.ogg"))
|
|
msg.Duration = 10
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingVoice(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewVoice(ChatID, FileID(ExistingVoiceFileID))
|
|
msg.Duration = 10
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithContact(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
contact := NewContact(ChatID, "5551234567", "Test")
|
|
|
|
if _, err := bot.Send(contact); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithLocation(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
_, err := bot.Send(NewLocation(ChatID, 40, 40))
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithVenue(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
venue := NewVenue(ChatID, "A Test Location", "123 Test Street", 40, 40)
|
|
|
|
if _, err := bot.Send(venue); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewVideo(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewVideo(ChatID, FilePath("tests/video.mp4"))
|
|
msg.Duration = 10
|
|
msg.Caption = "TEST"
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingVideo(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewVideo(ChatID, FileID(ExistingVideoFileID))
|
|
msg.Duration = 10
|
|
msg.Caption = "TEST"
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewVideoNote(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewVideoNote(ChatID, 240, FilePath("tests/videonote.mp4"))
|
|
msg.Duration = 10
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingVideoNote(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewVideoNote(ChatID, 240, FileID(ExistingVideoNoteFileID))
|
|
msg.Duration = 10
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewSticker(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewSticker(ChatID, FilePath("tests/image.jpg"))
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingSticker(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewSticker(ChatID, FileID(ExistingStickerFileID))
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewStickerAndKeyboardHide(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewSticker(ChatID, FilePath("tests/image.jpg"))
|
|
msg.ReplyMarkup = ReplyKeyboardRemove{
|
|
RemoveKeyboard: true,
|
|
Selective: false,
|
|
}
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingStickerAndKeyboardHide(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewSticker(ChatID, FileID(ExistingStickerFileID))
|
|
msg.ReplyMarkup = ReplyKeyboardRemove{
|
|
RemoveKeyboard: true,
|
|
Selective: false,
|
|
}
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendWithDice(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewDice(ChatID)
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
t.Fail()
|
|
}
|
|
|
|
}
|
|
|
|
func TestSendWithDiceWithEmoji(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewDiceWithEmoji(ChatID, "🏀")
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
t.Fail()
|
|
}
|
|
|
|
}
|
|
|
|
func TestGetFile(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
file := FileConfig{
|
|
FileID: ExistingPhotoFileID,
|
|
}
|
|
|
|
_, err := bot.GetFile(file)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSendChatConfig(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
_, err := bot.Request(NewChatAction(ChatID, ChatTyping))
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
// TODO: identify why this isn't working
|
|
// func TestSendEditMessage(t *testing.T) {
|
|
// bot, _ := getBot(t)
|
|
|
|
// msg, err := bot.Send(NewMessage(ChatID, "Testing editing."))
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
|
|
// edit := EditMessageTextConfig{
|
|
// BaseEdit: BaseEdit{
|
|
// ChatID: ChatID,
|
|
// MessageID: msg.MessageID,
|
|
// },
|
|
// Text: "Updated text.",
|
|
// }
|
|
|
|
// _, err = bot.Send(edit)
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
// }
|
|
|
|
func TestGetUserProfilePhotos(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
_, err := bot.GetUserProfilePhotos(NewUserProfilePhotos(ChatID))
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSetWebhookWithCert(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
time.Sleep(time.Second * 2)
|
|
|
|
bot.Request(DeleteWebhookConfig{})
|
|
|
|
wh, err := NewWebhookWithCert("https://example.com/tgbotapi-test/"+bot.Token, FilePath("tests/cert.pem"))
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = bot.Request(wh)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
_, err = bot.GetWebhookInfo()
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
bot.Request(DeleteWebhookConfig{})
|
|
}
|
|
|
|
func TestSetWebhookWithoutCert(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
time.Sleep(time.Second * 2)
|
|
|
|
bot.Request(DeleteWebhookConfig{})
|
|
|
|
wh, err := NewWebhook("https://example.com/tgbotapi-test/" + bot.Token)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
_, err = bot.Request(wh)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
info, err := bot.GetWebhookInfo()
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if info.MaxConnections == 0 {
|
|
t.Errorf("Expected maximum connections to be greater than 0")
|
|
}
|
|
if info.LastErrorDate != 0 {
|
|
t.Errorf("failed to set webhook: %s", info.LastErrorMessage)
|
|
}
|
|
|
|
bot.Request(DeleteWebhookConfig{})
|
|
}
|
|
|
|
func TestSendWithMediaGroupPhotoVideo(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
cfg := NewMediaGroup(ChatID, []interface{}{
|
|
NewInputMediaPhoto(FileURL("https://github.com/go-telegram-bot-api/telegram-bot-api/raw/0a3a1c8716c4cd8d26a262af9f12dcbab7f3f28c/tests/image.jpg")),
|
|
NewInputMediaPhoto(FilePath("tests/image.jpg")),
|
|
NewInputMediaVideo(FilePath("tests/video.mp4")),
|
|
})
|
|
|
|
messages, err := bot.SendMediaGroup(cfg)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if messages == nil {
|
|
t.Error("No received messages")
|
|
}
|
|
|
|
if len(messages) != len(cfg.Media) {
|
|
t.Errorf("Different number of messages: %d", len(messages))
|
|
}
|
|
}
|
|
|
|
func TestSendWithMediaGroupDocument(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
cfg := NewMediaGroup(ChatID, []interface{}{
|
|
NewInputMediaDocument(FileURL("https://i.imgur.com/unQLJIb.jpg")),
|
|
NewInputMediaDocument(FilePath("tests/image.jpg")),
|
|
})
|
|
|
|
messages, err := bot.SendMediaGroup(cfg)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if messages == nil {
|
|
t.Error("No received messages")
|
|
}
|
|
|
|
if len(messages) != len(cfg.Media) {
|
|
t.Errorf("Different number of messages: %d", len(messages))
|
|
}
|
|
}
|
|
|
|
func TestSendWithMediaGroupAudio(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
cfg := NewMediaGroup(ChatID, []interface{}{
|
|
NewInputMediaAudio(FilePath("tests/audio.mp3")),
|
|
NewInputMediaAudio(FilePath("tests/audio.mp3")),
|
|
})
|
|
|
|
messages, err := bot.SendMediaGroup(cfg)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if messages == nil {
|
|
t.Error("No received messages")
|
|
}
|
|
|
|
if len(messages) != len(cfg.Media) {
|
|
t.Errorf("Different number of messages: %d", len(messages))
|
|
}
|
|
}
|
|
|
|
func ExampleNewBotAPI() {
|
|
bot, err := NewBotAPI("MyAwesomeBotToken")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
bot.Debug = true
|
|
|
|
log.Printf("Authorized on account %s", bot.Self.UserName)
|
|
|
|
u := NewUpdate(0)
|
|
u.Timeout = 60
|
|
|
|
updates := bot.GetUpdatesChan(u)
|
|
|
|
// Optional: wait for updates and clear them if you don't want to handle
|
|
// a large backlog of old messages
|
|
time.Sleep(time.Millisecond * 500)
|
|
updates.Clear()
|
|
|
|
for update := range updates {
|
|
if update.Message == nil {
|
|
continue
|
|
}
|
|
|
|
log.Printf("[%s] %s", update.Message.From.UserName, update.Message.Text)
|
|
|
|
msg := NewMessage(update.Message.Chat.ID, update.Message.Text)
|
|
msg.ReplyToMessageID = update.Message.MessageID
|
|
|
|
bot.Send(msg)
|
|
}
|
|
}
|
|
|
|
func ExampleNewWebhook() {
|
|
bot, err := NewBotAPI("MyAwesomeBotToken")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
bot.Debug = true
|
|
|
|
log.Printf("Authorized on account %s", bot.Self.UserName)
|
|
|
|
wh, err := NewWebhookWithCert("https://www.google.com:8443/"+bot.Token, FilePath("cert.pem"))
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
_, err = bot.Request(wh)
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
info, err := bot.GetWebhookInfo()
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if info.LastErrorDate != 0 {
|
|
log.Printf("failed to set webhook: %s", info.LastErrorMessage)
|
|
}
|
|
|
|
updates := bot.ListenForWebhook("/" + bot.Token)
|
|
go http.ListenAndServeTLS("0.0.0.0:8443", "cert.pem", "key.pem", nil)
|
|
|
|
for update := range updates {
|
|
log.Printf("%+v\n", update)
|
|
}
|
|
}
|
|
|
|
func ExampleWebhookHandler() {
|
|
bot, err := NewBotAPI("MyAwesomeBotToken")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
bot.Debug = true
|
|
|
|
log.Printf("Authorized on account %s", bot.Self.UserName)
|
|
|
|
wh, err := NewWebhookWithCert("https://www.google.com:8443/"+bot.Token, FilePath("cert.pem"))
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
_, err = bot.Request(wh)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
info, err := bot.GetWebhookInfo()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if info.LastErrorDate != 0 {
|
|
log.Printf("[Telegram callback failed]%s", info.LastErrorMessage)
|
|
}
|
|
|
|
http.HandleFunc("/"+bot.Token, func(w http.ResponseWriter, r *http.Request) {
|
|
update, err := bot.HandleUpdate(r)
|
|
if err != nil {
|
|
log.Printf("%+v\n", err.Error())
|
|
} else {
|
|
log.Printf("%+v\n", *update)
|
|
}
|
|
})
|
|
|
|
go http.ListenAndServeTLS("0.0.0.0:8443", "cert.pem", "key.pem", nil)
|
|
}
|
|
|
|
func ExampleInlineConfig() {
|
|
bot, err := NewBotAPI("MyAwesomeBotToken") // create new bot
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
log.Printf("Authorized on account %s", bot.Self.UserName)
|
|
|
|
u := NewUpdate(0)
|
|
u.Timeout = 60
|
|
|
|
updates := bot.GetUpdatesChan(u)
|
|
|
|
for update := range updates {
|
|
if update.InlineQuery == nil { // if no inline query, ignore it
|
|
continue
|
|
}
|
|
|
|
article := NewInlineQueryResultArticle(update.InlineQuery.ID, "Echo", update.InlineQuery.Query)
|
|
article.Description = update.InlineQuery.Query
|
|
|
|
inlineConf := InlineConfig{
|
|
InlineQueryID: update.InlineQuery.ID,
|
|
IsPersonal: true,
|
|
CacheTime: 0,
|
|
Results: []interface{}{article},
|
|
}
|
|
|
|
if _, err := bot.Request(inlineConf); err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDeleteMessage(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewMessage(ChatID, "A test message from the test library in telegram-bot-api")
|
|
msg.ParseMode = ModeMarkdown
|
|
message, _ := bot.Send(msg)
|
|
|
|
deleteMessageConfig := DeleteMessageConfig{
|
|
ChatID: message.Chat.ID,
|
|
MessageID: message.MessageID,
|
|
}
|
|
_, err := bot.Request(deleteMessageConfig)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestPinChatMessage(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewMessage(SupergroupChatID, "A test message from the test library in telegram-bot-api")
|
|
msg.ParseMode = ModeMarkdown
|
|
message, _ := bot.Send(msg)
|
|
|
|
pinChatMessageConfig := PinChatMessageConfig{
|
|
ChatID: message.Chat.ID,
|
|
MessageID: message.MessageID,
|
|
DisableNotification: false,
|
|
}
|
|
_, err := bot.Request(pinChatMessageConfig)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestUnpinChatMessage(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewMessage(SupergroupChatID, "A test message from the test library in telegram-bot-api")
|
|
msg.ParseMode = ModeMarkdown
|
|
message, _ := bot.Send(msg)
|
|
|
|
// We need pin message to unpin something
|
|
pinChatMessageConfig := PinChatMessageConfig{
|
|
ChatID: message.Chat.ID,
|
|
MessageID: message.MessageID,
|
|
DisableNotification: false,
|
|
}
|
|
|
|
if _, err := bot.Request(pinChatMessageConfig); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
unpinChatMessageConfig := UnpinChatMessageConfig{
|
|
ChatID: message.Chat.ID,
|
|
MessageID: message.MessageID,
|
|
}
|
|
|
|
if _, err := bot.Request(unpinChatMessageConfig); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestUnpinAllChatMessages(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := NewMessage(SupergroupChatID, "A test message from the test library in telegram-bot-api")
|
|
msg.ParseMode = ModeMarkdown
|
|
message, _ := bot.Send(msg)
|
|
|
|
pinChatMessageConfig := PinChatMessageConfig{
|
|
ChatID: message.Chat.ID,
|
|
MessageID: message.MessageID,
|
|
DisableNotification: true,
|
|
}
|
|
|
|
if _, err := bot.Request(pinChatMessageConfig); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
unpinAllChatMessagesConfig := UnpinAllChatMessagesConfig{
|
|
ChatID: message.Chat.ID,
|
|
}
|
|
|
|
if _, err := bot.Request(unpinAllChatMessagesConfig); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestPolls(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
poll := NewPoll(SupergroupChatID, "Are polls working?", "Yes", "No")
|
|
|
|
msg, err := bot.Send(poll)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
result, err := bot.StopPoll(NewStopPoll(SupergroupChatID, msg.MessageID))
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if result.Question != "Are polls working?" {
|
|
t.Error("Poll question did not match")
|
|
}
|
|
|
|
if !result.IsClosed {
|
|
t.Error("Poll did not end")
|
|
}
|
|
|
|
if result.Options[0].Text != "Yes" || result.Options[0].VoterCount != 0 || result.Options[1].Text != "No" || result.Options[1].VoterCount != 0 {
|
|
t.Error("Poll options were incorrect")
|
|
}
|
|
}
|
|
|
|
func TestSendDice(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
dice := NewDice(ChatID)
|
|
|
|
msg, err := bot.Send(dice)
|
|
if err != nil {
|
|
t.Error("Unable to send dice roll")
|
|
}
|
|
|
|
if msg.Dice == nil {
|
|
t.Error("Dice roll was not received")
|
|
}
|
|
}
|
|
|
|
func TestCommands(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
setCommands := NewSetMyCommands(BotCommand{
|
|
Command: "test",
|
|
Description: "a test command",
|
|
})
|
|
|
|
if _, err := bot.Request(setCommands); err != nil {
|
|
t.Error("Unable to set commands")
|
|
}
|
|
|
|
commands, err := bot.GetMyCommands()
|
|
if err != nil {
|
|
t.Error("Unable to get commands")
|
|
}
|
|
|
|
if len(commands) != 1 {
|
|
t.Error("Incorrect number of commands returned")
|
|
}
|
|
|
|
if commands[0].Command != "test" || commands[0].Description != "a test command" {
|
|
t.Error("Commands were incorrectly set")
|
|
}
|
|
|
|
setCommands = NewSetMyCommandsWithScope(NewBotCommandScopeAllPrivateChats(), BotCommand{
|
|
Command: "private",
|
|
Description: "a private command",
|
|
})
|
|
|
|
if _, err := bot.Request(setCommands); err != nil {
|
|
t.Error("Unable to set commands")
|
|
}
|
|
|
|
commands, err = bot.GetMyCommandsWithConfig(NewGetMyCommandsWithScope(NewBotCommandScopeAllPrivateChats()))
|
|
if err != nil {
|
|
t.Error("Unable to get commands")
|
|
}
|
|
|
|
if len(commands) != 1 {
|
|
t.Error("Incorrect number of commands returned")
|
|
}
|
|
|
|
if commands[0].Command != "private" || commands[0].Description != "a private command" {
|
|
t.Error("Commands were incorrectly set")
|
|
}
|
|
}
|
|
|
|
// TODO: figure out why test is failing
|
|
//
|
|
// func TestEditMessageMedia(t *testing.T) {
|
|
// bot, _ := getBot(t)
|
|
|
|
// msg := NewPhoto(ChatID, "tests/image.jpg")
|
|
// msg.Caption = "Test"
|
|
// m, err := bot.Send(msg)
|
|
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
|
|
// edit := EditMessageMediaConfig{
|
|
// BaseEdit: BaseEdit{
|
|
// ChatID: ChatID,
|
|
// MessageID: m.MessageID,
|
|
// },
|
|
// Media: NewInputMediaVideo(FilePath("tests/video.mp4")),
|
|
// }
|
|
|
|
// _, err = bot.Request(edit)
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
// }
|
|
|
|
func TestPrepareInputMediaForParams(t *testing.T) {
|
|
media := []interface{}{
|
|
NewInputMediaPhoto(FilePath("tests/image.jpg")),
|
|
NewInputMediaVideo(FileID("test")),
|
|
}
|
|
|
|
prepared := prepareInputMediaForParams(media)
|
|
|
|
if media[0].(InputMediaPhoto).Media != FilePath("tests/image.jpg") {
|
|
t.Error("Original media was changed")
|
|
}
|
|
|
|
if prepared[0].(InputMediaPhoto).Media != fileAttach("attach://file-0") {
|
|
t.Error("New media was not replaced")
|
|
}
|
|
|
|
if prepared[1].(InputMediaVideo).Media != FileID("test") {
|
|
t.Error("Passthrough value was not the same")
|
|
}
|
|
}
|