604 lines
12 KiB
Go
604 lines
12 KiB
Go
package tgbotapi_test
|
|
|
|
import (
|
|
"github.com/go-telegram-bot-api/telegram-bot-api"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
const (
|
|
TestToken = "153667468:AAHlSHlMqSt1f_uFmVRJbm5gntu2HI4WW8I"
|
|
ChatID = 76918703
|
|
ReplyToMessageID = 35
|
|
ExistingPhotoFileID = "AgADAgADw6cxG4zHKAkr42N7RwEN3IFShCoABHQwXEtVks4EH2wBAAEC"
|
|
ExistingDocumentFileID = "BQADAgADOQADjMcoCcioX1GrDvp3Ag"
|
|
ExistingAudioFileID = "BQADAgADRgADjMcoCdXg3lSIN49lAg"
|
|
ExistingVoiceFileID = "AwADAgADWQADjMcoCeul6r_q52IyAg"
|
|
ExistingVideoFileID = "BAADAgADZgADjMcoCav432kYe0FRAg"
|
|
ExistingStickerFileID = "BQADAgADcwADjMcoCbdl-6eB--YPAg"
|
|
)
|
|
|
|
func getBot(t *testing.T) (*tgbotapi.BotAPI, error) {
|
|
bot, err := tgbotapi.NewBotAPI(TestToken)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
|
|
return bot, err
|
|
}
|
|
|
|
func TestNewBotAPI_notoken(t *testing.T) {
|
|
_, err := tgbotapi.NewBotAPI("")
|
|
|
|
if err == nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestGetUpdates(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
u := tgbotapi.NewUpdate(0)
|
|
|
|
_, err := bot.GetUpdates(u)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithMessage(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewMessage(ChatID, "A test message from the test library in telegram-bot-api")
|
|
msg.ParseMode = "markdown"
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithMessageReply(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewMessage(ChatID, "A test message from the test library in telegram-bot-api")
|
|
msg.ReplyToMessageID = ReplyToMessageID
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithMessageForward(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewForward(ChatID, ChatID, ReplyToMessageID)
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewPhoto(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewPhotoUpload(ChatID, "tests/image.jpg")
|
|
msg.Caption = "Test"
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewPhotoWithFileBytes(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
data, _ := ioutil.ReadFile("tests/image.jpg")
|
|
b := tgbotapi.FileBytes{Name: "image.jpg", Bytes: data}
|
|
|
|
msg := tgbotapi.NewPhotoUpload(ChatID, b)
|
|
msg.Caption = "Test"
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewPhotoWithFileReader(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
f, _ := os.Open("tests/image.jpg")
|
|
reader := tgbotapi.FileReader{Name: "image.jpg", Reader: f, Size: -1}
|
|
|
|
msg := tgbotapi.NewPhotoUpload(ChatID, reader)
|
|
msg.Caption = "Test"
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewPhotoReply(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewPhotoUpload(ChatID, "tests/image.jpg")
|
|
msg.ReplyToMessageID = ReplyToMessageID
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingPhoto(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewPhotoShare(ChatID, ExistingPhotoFileID)
|
|
msg.Caption = "Test"
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewDocument(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewDocumentUpload(ChatID, "tests/image.jpg")
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingDocument(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewDocumentShare(ChatID, ExistingDocumentFileID)
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewAudio(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewAudioUpload(ChatID, "tests/audio.mp3")
|
|
msg.Title = "TEST"
|
|
msg.Duration = 10
|
|
msg.Performer = "TEST"
|
|
msg.MimeType = "audio/mpeg"
|
|
msg.FileSize = 688
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingAudio(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewAudioShare(ChatID, ExistingAudioFileID)
|
|
msg.Title = "TEST"
|
|
msg.Duration = 10
|
|
msg.Performer = "TEST"
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewVoice(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewVoiceUpload(ChatID, "tests/voice.ogg")
|
|
msg.Duration = 10
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingVoice(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewVoiceShare(ChatID, ExistingVoiceFileID)
|
|
msg.Duration = 10
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithContact(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
contact := tgbotapi.NewContact(ChatID, "5551234567", "Test")
|
|
|
|
if _, err := bot.Send(contact); err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithLocation(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
_, err := bot.Send(tgbotapi.NewLocation(ChatID, 40, 40))
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithVenue(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
venue := tgbotapi.NewVenue(ChatID, "A Test Location", "123 Test Street", 40, 40)
|
|
|
|
if _, err := bot.Send(venue); err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewVideo(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewVideoUpload(ChatID, "tests/video.mp4")
|
|
msg.Duration = 10
|
|
msg.Caption = "TEST"
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingVideo(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewVideoShare(ChatID, ExistingVideoFileID)
|
|
msg.Duration = 10
|
|
msg.Caption = "TEST"
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewSticker(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewStickerUpload(ChatID, "tests/image.jpg")
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingSticker(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewStickerShare(ChatID, ExistingStickerFileID)
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithNewStickerAndKeyboardHide(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewStickerUpload(ChatID, "tests/image.jpg")
|
|
msg.ReplyMarkup = tgbotapi.ReplyKeyboardHide{true, false}
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendWithExistingStickerAndKeyboardHide(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg := tgbotapi.NewStickerShare(ChatID, ExistingStickerFileID)
|
|
msg.ReplyMarkup = tgbotapi.ReplyKeyboardHide{true, false}
|
|
|
|
_, err := bot.Send(msg)
|
|
|
|
if err != nil {
|
|
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestGetFile(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
file := tgbotapi.FileConfig{ExistingPhotoFileID}
|
|
|
|
_, err := bot.GetFile(file)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendChatConfig(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
_, err := bot.Send(tgbotapi.NewChatAction(ChatID, tgbotapi.ChatTyping))
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSendEditMessage(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
msg, err := bot.Send(tgbotapi.NewMessage(ChatID, "Testing editing."))
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
|
|
edit := tgbotapi.EditMessageTextConfig{
|
|
BaseEdit: tgbotapi.BaseEdit{
|
|
ChatID: ChatID,
|
|
MessageID: msg.MessageID,
|
|
},
|
|
Text: "Updated text.",
|
|
}
|
|
|
|
_, err = bot.Send(edit)
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestGetUserProfilePhotos(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
_, err := bot.GetUserProfilePhotos(tgbotapi.NewUserProfilePhotos(ChatID))
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSetWebhookWithCert(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
bot.RemoveWebhook()
|
|
|
|
wh := tgbotapi.NewWebhookWithCert("https://example.com/tgbotapi-test/"+bot.Token, "tests/cert.pem")
|
|
_, err := bot.SetWebhook(wh)
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
|
|
bot.RemoveWebhook()
|
|
}
|
|
|
|
func TestSetWebhookWithoutCert(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
bot.RemoveWebhook()
|
|
|
|
wh := tgbotapi.NewWebhook("https://example.com/tgbotapi-test/" + bot.Token)
|
|
_, err := bot.SetWebhook(wh)
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
|
|
bot.RemoveWebhook()
|
|
}
|
|
|
|
func TestNewInlineQueryResultArticle(t *testing.T) {
|
|
result := tgbotapi.NewInlineQueryResultArticle("id", "title", "message")
|
|
|
|
if result.Type != "article" ||
|
|
result.ID != "id" ||
|
|
result.Title != "title" ||
|
|
result.InputMessageContent.(tgbotapi.InputTextMessageContent).Text != "message" {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestNewInlineQueryResultGIF(t *testing.T) {
|
|
result := tgbotapi.NewInlineQueryResultGIF("id", "google.com")
|
|
|
|
if result.Type != "gif" ||
|
|
result.ID != "id" ||
|
|
result.URL != "google.com" {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestNewInlineQueryResultMPEG4GIF(t *testing.T) {
|
|
result := tgbotapi.NewInlineQueryResultMPEG4GIF("id", "google.com")
|
|
|
|
if result.Type != "mpeg4_gif" ||
|
|
result.ID != "id" ||
|
|
result.URL != "google.com" {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestNewInlineQueryResultPhoto(t *testing.T) {
|
|
result := tgbotapi.NewInlineQueryResultPhoto("id", "google.com")
|
|
|
|
if result.Type != "photo" ||
|
|
result.ID != "id" ||
|
|
result.URL != "google.com" {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestNewInlineQueryResultVideo(t *testing.T) {
|
|
result := tgbotapi.NewInlineQueryResultVideo("id", "google.com")
|
|
|
|
if result.Type != "video" ||
|
|
result.ID != "id" ||
|
|
result.URL != "google.com" {
|
|
}
|
|
}
|
|
|
|
func TestNewEditMessageText(t *testing.T) {
|
|
edit := tgbotapi.NewEditMessageText(ChatID, ReplyToMessageID, "new text")
|
|
|
|
if edit.Text != "new text" ||
|
|
edit.BaseEdit.ChatID != ChatID ||
|
|
edit.BaseEdit.MessageID != ReplyToMessageID {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestNewEditMessageCaption(t *testing.T) {
|
|
edit := tgbotapi.NewEditMessageCaption(ChatID, ReplyToMessageID, "new caption")
|
|
|
|
if edit.Caption != "new caption" ||
|
|
edit.BaseEdit.ChatID != ChatID ||
|
|
edit.BaseEdit.MessageID != ReplyToMessageID {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestNewEditMessageReplyMarkup(t *testing.T) {
|
|
markup := tgbotapi.InlineKeyboardMarkup{
|
|
InlineKeyboard: [][]tgbotapi.InlineKeyboardButton{
|
|
[]tgbotapi.InlineKeyboardButton{
|
|
tgbotapi.InlineKeyboardButton{Text: "test"},
|
|
},
|
|
},
|
|
}
|
|
|
|
edit := tgbotapi.NewEditMessageReplyMarkup(ChatID, ReplyToMessageID, markup)
|
|
|
|
if edit.ReplyMarkup.InlineKeyboard[0][0].Text != "test" ||
|
|
edit.BaseEdit.ChatID != ChatID ||
|
|
edit.BaseEdit.MessageID != ReplyToMessageID {
|
|
t.Fail()
|
|
}
|
|
|
|
}
|
|
|
|
func TestUpdatesChan(t *testing.T) {
|
|
bot, _ := getBot(t)
|
|
|
|
var ucfg tgbotapi.UpdateConfig = tgbotapi.NewUpdate(0)
|
|
ucfg.Timeout = 60
|
|
_, err := bot.GetUpdatesChan(ucfg)
|
|
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func ExampleNewBotAPI() {
|
|
bot, err := tgbotapi.NewBotAPI("MyAwesomeBotToken")
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
|
|
bot.Debug = true
|
|
|
|
log.Printf("Authorized on account %s", bot.Self.UserName)
|
|
|
|
u := tgbotapi.NewUpdate(0)
|
|
u.Timeout = 60
|
|
|
|
updates, err := bot.GetUpdatesChan(u)
|
|
|
|
for update := range updates {
|
|
log.Printf("[%s] %s", update.Message.From.UserName, update.Message.Text)
|
|
|
|
msg := tgbotapi.NewMessage(update.Message.Chat.ID, update.Message.Text)
|
|
msg.ReplyToMessageID = update.Message.MessageID
|
|
|
|
bot.Send(msg)
|
|
}
|
|
}
|
|
|
|
func ExampleNewWebhook() {
|
|
bot, err := tgbotapi.NewBotAPI("MyAwesomeBotToken")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
bot.Debug = true
|
|
|
|
log.Printf("Authorized on account %s", bot.Self.UserName)
|
|
|
|
_, err = bot.SetWebhook(tgbotapi.NewWebhookWithCert("https://www.google.com:8443/"+bot.Token, "cert.pem"))
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
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 ExampleAnswerInlineQuery() {
|
|
bot, err := tgbotapi.NewBotAPI("MyAwesomeBotToken") // create new bot
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
|
|
log.Printf("Authorized on account %s", bot.Self.UserName)
|
|
|
|
u := tgbotapi.NewUpdate(0)
|
|
u.Timeout = 60
|
|
|
|
updates, err := bot.GetUpdatesChan(u)
|
|
|
|
for update := range updates {
|
|
if update.InlineQuery.Query == "" { // if no inline query, ignore it
|
|
continue
|
|
}
|
|
|
|
article := tgbotapi.NewInlineQueryResultArticle(update.InlineQuery.ID, "Echo", update.InlineQuery.Query)
|
|
article.Description = update.InlineQuery.Query
|
|
|
|
inlineConf := tgbotapi.InlineConfig{
|
|
InlineQueryID: update.InlineQuery.ID,
|
|
IsPersonal: true,
|
|
CacheTime: 0,
|
|
Results: []interface{}{article},
|
|
}
|
|
|
|
if _, err := bot.AnswerInlineQuery(inlineConf); err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
}
|