1228 lines
28 KiB
Go
1228 lines
28 KiB
Go
package tgbotapi
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/mock/gomock"
|
|
)
|
|
|
|
const (
|
|
TestToken = "153667468:AAHlSHlMqSt1f_uFmVRJbm5gntu2HI4WW8I"
|
|
ChatID = 111
|
|
SupergroupChatID = -1111
|
|
ReplyToMessageID = 1
|
|
)
|
|
|
|
func prepareHttpClient(t *testing.T) *MockHTTPClient {
|
|
ctrl := gomock.NewController(t)
|
|
httpMock := NewMockHTTPClient(ctrl)
|
|
return httpMock
|
|
}
|
|
|
|
func expectGetMe(t *testing.T, c *MockHTTPClient) {
|
|
meResp := `{"ok": true, "result": {"id": 123456789, "is_bot": true, "first_name": "MyBot", "username": "my_bot"}}`
|
|
c.EXPECT().
|
|
Do(gomock.Any()).
|
|
DoAndReturn(func(req *http.Request) (*http.Response, error) {
|
|
valid := isRequestValid(req, TestToken, "getMe")
|
|
require.True(t, valid)
|
|
return newOKResponse(meResp), nil
|
|
})
|
|
}
|
|
|
|
func TestNewBotAPI_empty_token(t *testing.T) {
|
|
_, err := NewBotAPI("")
|
|
|
|
if err == nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func isRequestValid(r *http.Request, token, path string) bool {
|
|
return r.Method == http.MethodPost &&
|
|
r.URL.Path == fmt.Sprintf("/bot%s/%s", token, path) &&
|
|
r.URL.Scheme == "https"
|
|
}
|
|
|
|
func newOKResponse(body string) *http.Response {
|
|
return &http.Response{
|
|
StatusCode: http.StatusOK,
|
|
Status: http.StatusText(http.StatusOK),
|
|
Body: io.NopCloser(bytes.NewBufferString(body)),
|
|
Header: make(http.Header),
|
|
}
|
|
}
|
|
|
|
func TestGetUpdates(t *testing.T) {
|
|
updateResp := `{
|
|
"ok": true,
|
|
"result": [
|
|
{
|
|
"update_id": 123456789,
|
|
"message": {
|
|
"message_id": 111,
|
|
"from": {
|
|
"id": 222,
|
|
"is_bot": false,
|
|
"first_name": "John",
|
|
"username": "john_doe"
|
|
},
|
|
"chat": {
|
|
"id": 333,
|
|
"first_name": "John",
|
|
"username": "john_doe",
|
|
"type": "private"
|
|
},
|
|
"date": 1640001112,
|
|
"text": "Hello, bot!"
|
|
}
|
|
}
|
|
]
|
|
}`
|
|
|
|
client := prepareHttpClient(t)
|
|
defer client.ctrl.Finish()
|
|
expectGetMe(t, client)
|
|
|
|
client.EXPECT().
|
|
Do(gomock.Any()).
|
|
DoAndReturn(func(req *http.Request) (*http.Response, error) {
|
|
valid := isRequestValid(req, TestToken, "getUpdates")
|
|
require.True(t, valid)
|
|
return newOKResponse(updateResp), nil
|
|
})
|
|
|
|
bot, err := NewBotAPIWithClient(TestToken, APIEndpoint, client)
|
|
require.NoError(t, err)
|
|
|
|
u := NewUpdate(0)
|
|
_, err = bot.GetUpdates(u)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestSendWithMessage(t *testing.T) {
|
|
client := prepareHttpClient(t)
|
|
defer client.ctrl.Finish()
|
|
|
|
responseBody := `{
|
|
"ok": true,
|
|
"result": {
|
|
"message_id": 123,
|
|
"from": {
|
|
"id": 123456789,
|
|
"is_bot": true,
|
|
"first_name": "MyBot",
|
|
"username": "my_bot"
|
|
},
|
|
"chat": {
|
|
"id": 987654321,
|
|
"first_name": "John",
|
|
"username": "john_doe",
|
|
"type": "private"
|
|
},
|
|
"date": 1640001112,
|
|
"text": "Hello, John!"
|
|
}
|
|
}`
|
|
|
|
expectGetMe(t, client)
|
|
|
|
client.EXPECT().
|
|
Do(gomock.Any()).
|
|
DoAndReturn(func(req *http.Request) (*http.Response, error) {
|
|
valid := isRequestValid(req, TestToken, "sendMessage")
|
|
require.True(t, valid)
|
|
return newOKResponse(responseBody), nil
|
|
})
|
|
|
|
bot, err := NewBotAPIWithClient(TestToken, APIEndpoint, client)
|
|
require.NoError(t, err)
|
|
|
|
msg := NewMessage(ChatID, "A test message from the test library in telegram-bot-api")
|
|
msg.ParseMode = ModeMarkdown
|
|
_, err = bot.Send(msg)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestSendWithMessageReply(t *testing.T) {
|
|
client := prepareHttpClient(t)
|
|
defer client.ctrl.Finish()
|
|
|
|
responseBody := `{
|
|
"ok": true,
|
|
"result": {
|
|
"message_id": 123,
|
|
"from": {
|
|
"id": 123456789,
|
|
"is_bot": true,
|
|
"first_name": "MyBot",
|
|
"username": "my_bot"
|
|
},
|
|
"chat": {
|
|
"id": 987654321,
|
|
"first_name": "John",
|
|
"username": "john_doe",
|
|
"type": "private"
|
|
},
|
|
"date": 1640001112,
|
|
"text": "Hello, John!"
|
|
}
|
|
}`
|
|
|
|
expectGetMe(t, client)
|
|
|
|
client.EXPECT().
|
|
Do(gomock.Any()).
|
|
DoAndReturn(func(req *http.Request) (*http.Response, error) {
|
|
valid := isRequestValid(req, TestToken, "sendMessage")
|
|
require.True(t, valid)
|
|
return newOKResponse(responseBody), nil
|
|
})
|
|
|
|
bot, err := NewBotAPIWithClient(TestToken, APIEndpoint, client)
|
|
require.NoError(t, err)
|
|
|
|
msg := NewMessage(ChatID, "A test message from the test library in telegram-bot-api")
|
|
msg.ReplyParameters.MessageID = ReplyToMessageID
|
|
_, err = bot.Send(msg)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
//func TestSendWithMessageForward(t *testing.T) {
|
|
// bot, _ := configureBot(t)
|
|
//
|
|
// msg := NewForward(ChatID, ChatID, ReplyToMessageID)
|
|
// _, err := bot.Send(msg)
|
|
//
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
|
|
func TestCopyMessage(t *testing.T) {
|
|
client := prepareHttpClient(t)
|
|
defer client.ctrl.Finish()
|
|
expectGetMe(t, client)
|
|
|
|
responseBody := `{
|
|
"ok": true,
|
|
"result": {
|
|
"message_id": 123,
|
|
"from": {
|
|
"id": 123456789,
|
|
"is_bot": true,
|
|
"first_name": "MyBot",
|
|
"username": "my_bot"
|
|
},
|
|
"chat": {
|
|
"id": 987654321,
|
|
"first_name": "John",
|
|
"username": "john_doe",
|
|
"type": "private"
|
|
},
|
|
"date": 1640001112,
|
|
"text": "Hello, John!"
|
|
}
|
|
}`
|
|
|
|
client.EXPECT().
|
|
Do(gomock.Any()).
|
|
DoAndReturn(func(req *http.Request) (*http.Response, error) {
|
|
valid := isRequestValid(req, TestToken, "sendMessage")
|
|
require.True(t, valid)
|
|
return newOKResponse(responseBody), nil
|
|
})
|
|
|
|
client.EXPECT().
|
|
Do(gomock.Any()).
|
|
DoAndReturn(func(req *http.Request) (*http.Response, error) {
|
|
valid := isRequestValid(req, TestToken, "copyMessage")
|
|
require.True(t, valid)
|
|
return newOKResponse(responseBody), nil
|
|
})
|
|
|
|
bot, err := NewBotAPIWithClient(TestToken, APIEndpoint, client)
|
|
require.NoError(t, err)
|
|
|
|
msg := NewMessage(ChatID, "A test message from the test library in telegram-bot-api")
|
|
message, err := bot.Send(msg)
|
|
require.NoError(t, err)
|
|
|
|
copyMessageConfig := NewCopyMessage(SupergroupChatID, message.Chat.ID, message.MessageID)
|
|
messageID, err := bot.CopyMessage(copyMessageConfig)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, messageID.MessageID, message.MessageID)
|
|
}
|
|
|
|
//func TestSendWithNewPhoto(t *testing.T) {
|
|
// bot, _ := configureBot(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, _ := configureBot(t)
|
|
//
|
|
// data, _ := os.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, _ := configureBot(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, _ := configureBot(t)
|
|
//
|
|
// msg := NewPhoto(ChatID, FilePath("tests/image.jpg"))
|
|
// msg.ReplyParameters.MessageID = ReplyToMessageID
|
|
//
|
|
// _, err := bot.Send(msg)
|
|
//
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
//
|
|
//func TestSendNewPhotoToChannel(t *testing.T) {
|
|
// bot, _ := configureBot(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, _ := configureBot(t)
|
|
//
|
|
// data, _ := os.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, _ := configureBot(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, _ := configureBot(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, _ := configureBot(t)
|
|
//
|
|
// msg := NewDocument(ChatID, FilePath("tests/image.jpg"))
|
|
// _, err := bot.Send(msg)
|
|
//
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
//
|
|
//func TestSendWithNewDocumentAndThumb(t *testing.T) {
|
|
// bot, _ := configureBot(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, _ := configureBot(t)
|
|
//
|
|
// msg := NewDocument(ChatID, FileID(ExistingDocumentFileID))
|
|
// _, err := bot.Send(msg)
|
|
//
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
//
|
|
//func TestSendWithNewAudio(t *testing.T) {
|
|
// bot, _ := configureBot(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, _ := configureBot(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, _ := configureBot(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, _ := configureBot(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, _ := configureBot(t)
|
|
//
|
|
// contact := NewContact(ChatID, "5551234567", "Test")
|
|
//
|
|
// if _, err := bot.Send(contact); err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
//
|
|
//func TestSendWithLocation(t *testing.T) {
|
|
// bot, _ := configureBot(t)
|
|
//
|
|
// _, err := bot.Send(NewLocation(ChatID, 40, 40))
|
|
//
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
//
|
|
//func TestSendWithVenue(t *testing.T) {
|
|
// bot, _ := configureBot(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, _ := configureBot(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, _ := configureBot(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, _ := configureBot(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, _ := configureBot(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, _ := configureBot(t)
|
|
//
|
|
// msg := NewSticker(ChatID, FilePath("tests/image.jpg"))
|
|
//
|
|
// _, err := bot.Send(msg)
|
|
//
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
//
|
|
//func TestSendWithExistingSticker(t *testing.T) {
|
|
// bot, _ := configureBot(t)
|
|
//
|
|
// msg := NewSticker(ChatID, FileID(ExistingStickerFileID))
|
|
//
|
|
// _, err := bot.Send(msg)
|
|
//
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
//
|
|
//func TestSendWithNewStickerAndKeyboardHide(t *testing.T) {
|
|
// bot, _ := configureBot(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, _ := configureBot(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, _ := configureBot(t)
|
|
//
|
|
// msg := NewDice(ChatID)
|
|
// _, err := bot.Send(msg)
|
|
//
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// t.Fail()
|
|
// }
|
|
//
|
|
//}
|
|
//
|
|
//func TestSendWithDiceWithEmoji(t *testing.T) {
|
|
// bot, _ := configureBot(t)
|
|
//
|
|
// msg := NewDiceWithEmoji(ChatID, "🏀")
|
|
// _, err := bot.Send(msg)
|
|
//
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// t.Fail()
|
|
// }
|
|
//
|
|
//}
|
|
//
|
|
//func TestGetFile(t *testing.T) {
|
|
// bot, _ := configureBot(t)
|
|
//
|
|
// file := FileConfig{
|
|
// FileID: ExistingPhotoFileID,
|
|
// }
|
|
//
|
|
// _, err := bot.GetFile(file)
|
|
//
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
//
|
|
//func TestSendChatConfig(t *testing.T) {
|
|
// bot, _ := configureBot(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, _ := configureBot(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, _ := configureBot(t)
|
|
//
|
|
// _, err := bot.GetUserProfilePhotos(NewUserProfilePhotos(ChatID))
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
//
|
|
//func TestSetWebhookWithCert(t *testing.T) {
|
|
// bot, _ := configureBot(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, _ := configureBot(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, _ := configureBot(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, _ := configureBot(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, _ := configureBot(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.ReplyParameters.MessageID = 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, _ := configureBot(t)
|
|
//
|
|
// msg := NewMessage(ChatID, "A test message from the test library in telegram-bot-api")
|
|
// msg.ParseMode = ModeMarkdown
|
|
// message, _ := bot.Send(msg)
|
|
//
|
|
// deleteMessageConfig := DeleteMessageConfig{
|
|
// BaseChatMessage: BaseChatMessage{
|
|
// ChatConfig: ChatConfig{
|
|
// ChatID: message.Chat.ID,
|
|
// },
|
|
// MessageID: message.MessageID,
|
|
// },
|
|
// }
|
|
// _, err := bot.Request(deleteMessageConfig)
|
|
//
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
//
|
|
//func TestPinChatMessage(t *testing.T) {
|
|
// bot, _ := configureBot(t)
|
|
//
|
|
// msg := NewMessage(SupergroupChatID, "A test message from the test library in telegram-bot-api")
|
|
// msg.ParseMode = ModeMarkdown
|
|
// message, _ := bot.Send(msg)
|
|
//
|
|
// pinChatMessageConfig := PinChatMessageConfig{
|
|
// BaseChatMessage: BaseChatMessage{
|
|
// ChatConfig: ChatConfig{
|
|
// ChatID: ChatID,
|
|
// },
|
|
// MessageID: message.MessageID,
|
|
// },
|
|
// DisableNotification: false,
|
|
// }
|
|
// _, err := bot.Request(pinChatMessageConfig)
|
|
//
|
|
// if err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
//
|
|
//func TestUnpinChatMessage(t *testing.T) {
|
|
// bot, _ := configureBot(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{
|
|
// BaseChatMessage: BaseChatMessage{
|
|
// ChatConfig: ChatConfig{
|
|
// ChatID: message.Chat.ID,
|
|
// },
|
|
// MessageID: message.MessageID,
|
|
// },
|
|
// DisableNotification: false,
|
|
// }
|
|
//
|
|
// if _, err := bot.Request(pinChatMessageConfig); err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//
|
|
// unpinChatMessageConfig := UnpinChatMessageConfig{
|
|
// BaseChatMessage: BaseChatMessage{
|
|
// ChatConfig: ChatConfig{
|
|
// ChatID: message.Chat.ID,
|
|
// },
|
|
// MessageID: message.MessageID,
|
|
// },
|
|
// }
|
|
//
|
|
// if _, err := bot.Request(unpinChatMessageConfig); err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
//
|
|
//func TestUnpinAllChatMessages(t *testing.T) {
|
|
// bot, _ := configureBot(t)
|
|
//
|
|
// msg := NewMessage(SupergroupChatID, "A test message from the test library in telegram-bot-api")
|
|
// msg.ParseMode = ModeMarkdown
|
|
// message, _ := bot.Send(msg)
|
|
//
|
|
// pinChatMessageConfig := PinChatMessageConfig{
|
|
// BaseChatMessage: BaseChatMessage{
|
|
// ChatConfig: ChatConfig{
|
|
// ChatID: message.Chat.ID,
|
|
// },
|
|
// MessageID: message.MessageID,
|
|
// },
|
|
// DisableNotification: true,
|
|
// }
|
|
//
|
|
// if _, err := bot.Request(pinChatMessageConfig); err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//
|
|
// unpinAllChatMessagesConfig := UnpinAllChatMessagesConfig{
|
|
// ChatConfig: ChatConfig{ChatID: message.Chat.ID},
|
|
// }
|
|
//
|
|
// if _, err := bot.Request(unpinAllChatMessagesConfig); err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
//}
|
|
//
|
|
//func TestPolls(t *testing.T) {
|
|
// bot, _ := configureBot(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, _ := configureBot(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, _ := configureBot(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, _ := configureBot(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")
|
|
// }
|
|
//}
|
|
|
|
func TestPrepareInputMediaForParams(t *testing.T) {
|
|
media := []any{
|
|
NewInputMediaPhoto(FilePath("./image.jpg")),
|
|
NewInputMediaVideo(FileID("test")),
|
|
}
|
|
|
|
prepared := prepareInputMediaForParams(media)
|
|
|
|
if media[0].(InputMediaPhoto).Media != FilePath("./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")
|
|
}
|
|
}
|