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

Test refactored

bot-api-6.1
Gleb Sinyavsky 2015-11-21 15:34:30 +03:00
parent 0332f792db
commit 653939995c
2 changed files with 103 additions and 219 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.travis.yml
View File

@ -1,6 +1,5 @@
language: go language: go
go: go:
- 1.3
- 1.4 - 1.4
- tip - tip

321
bot_test.go
View File

@ -11,14 +11,26 @@ import (
"testing" "testing"
) )
func TestMain(m *testing.M) { const (
botToken := os.Getenv("TELEGRAM_API_TOKEN") TestToken = "153667468:AAHlSHlMqSt1f_uFmVRJbm5gntu2HI4WW8I"
ChatID = 76918703
ReplyToMessageID = 35
ExistingPhotoFileID = "AgADAgADw6cxG4zHKAkr42N7RwEN3IFShCoABHQwXEtVks4EH2wBAAEC"
ExistingDocumentFileID = "BQADAgADOQADjMcoCcioX1GrDvp3Ag"
ExistingAudioFileID = "BQADAgADRgADjMcoCdXg3lSIN49lAg"
ExistingVoiceFileID = "AwADAgADWQADjMcoCeul6r_q52IyAg"
ExistingVideoFileID = "BAADAgADZgADjMcoCav432kYe0FRAg"
ExistingStickerFileID = "BQADAgADcwADjMcoCbdl-6eB--YPAg"
)
if botToken == "" { func getBot(t *testing.T) (*tgbotapi.BotAPI, error) {
log.Panic("You must provide a TELEGRAM_API_TOKEN env variable to test!") bot, err := tgbotapi.NewBotAPI(TestToken)
if err != nil {
t.Fail()
} }
os.Exit(m.Run()) return bot, err
} }
func TestNewBotAPI_notoken(t *testing.T) { func TestNewBotAPI_notoken(t *testing.T) {
@ -29,24 +41,12 @@ func TestNewBotAPI_notoken(t *testing.T) {
} }
} }
func TestNewBotAPI_token(t *testing.T) {
_, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN"))
if err != nil {
t.Fail()
}
}
func TestGetUpdates(t *testing.T) { func TestGetUpdates(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil {
t.Fail()
}
u := tgbotapi.NewUpdate(0) u := tgbotapi.NewUpdate(0)
_, err = bot.GetUpdates(u) _, err := bot.GetUpdates(u)
if err != nil { if err != nil {
t.Log(err.Error()) t.Log(err.Error())
@ -55,15 +55,11 @@ func TestGetUpdates(t *testing.T) {
} }
func TestSendWithMessage(t *testing.T) { func TestSendWithMessage(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewMessage(ChatID, "A test message from the test library in telegram-bot-api")
t.Fail()
}
msg := tgbotapi.NewMessage(76918703, "A test message from the test library in telegram-bot-api")
msg.ParseMode = "markdown" msg.ParseMode = "markdown"
_, err = bot.Send(msg) _, err := bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -71,15 +67,11 @@ func TestSendWithMessage(t *testing.T) {
} }
func TestSendWithMessageReply(t *testing.T) { func TestSendWithMessageReply(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewMessage(ChatID, "A test message from the test library in telegram-bot-api")
t.Fail() msg.ReplyToMessageID = ReplyToMessageID
} _, err := bot.Send(msg)
msg := tgbotapi.NewMessage(76918703, "A test message from the test library in telegram-bot-api")
msg.ReplyToMessageID = 480
_, err = bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -87,14 +79,10 @@ func TestSendWithMessageReply(t *testing.T) {
} }
func TestSendWithMessageForward(t *testing.T) { func TestSendWithMessageForward(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewForward(ChatID, ChatID, ReplyToMessageID)
t.Fail() _, err := bot.Send(msg)
}
msg := tgbotapi.NewForward(76918703, 76918703, 480)
_, err = bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -102,15 +90,11 @@ func TestSendWithMessageForward(t *testing.T) {
} }
func TestSendWithNewPhoto(t *testing.T) { func TestSendWithNewPhoto(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewPhotoUpload(ChatID, "tests/image.jpg")
t.Fail()
}
msg := tgbotapi.NewPhotoUpload(76918703, "tests/image.jpg")
msg.Caption = "Test" msg.Caption = "Test"
_, err = bot.Send(msg) _, err := bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -118,18 +102,14 @@ func TestSendWithNewPhoto(t *testing.T) {
} }
func TestSendWithNewPhotoWithFileBytes(t *testing.T) { func TestSendWithNewPhotoWithFileBytes(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil {
t.Fail()
}
data, _ := ioutil.ReadFile("tests/image.jpg") data, _ := ioutil.ReadFile("tests/image.jpg")
b := tgbotapi.FileBytes{Name: "image.jpg", Bytes: data} b := tgbotapi.FileBytes{Name: "image.jpg", Bytes: data}
msg := tgbotapi.NewPhotoUpload(76918703, b) msg := tgbotapi.NewPhotoUpload(ChatID, b)
msg.Caption = "Test" msg.Caption = "Test"
_, err = bot.Send(msg) _, err := bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -137,18 +117,14 @@ func TestSendWithNewPhotoWithFileBytes(t *testing.T) {
} }
func TestSendWithNewPhotoWithFileReader(t *testing.T) { func TestSendWithNewPhotoWithFileReader(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil {
t.Fail()
}
f, _ := os.Open("tests/image.jpg") f, _ := os.Open("tests/image.jpg")
reader := tgbotapi.FileReader{Name: "image.jpg", Reader: f, Size: -1} reader := tgbotapi.FileReader{Name: "image.jpg", Reader: f, Size: -1}
msg := tgbotapi.NewPhotoUpload(76918703, reader) msg := tgbotapi.NewPhotoUpload(ChatID, reader)
msg.Caption = "Test" msg.Caption = "Test"
_, err = bot.Send(msg) _, err := bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -156,16 +132,12 @@ func TestSendWithNewPhotoWithFileReader(t *testing.T) {
} }
func TestSendWithNewPhotoReply(t *testing.T) { func TestSendWithNewPhotoReply(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewPhotoUpload(ChatID, "tests/image.jpg")
t.Fail() msg.ReplyToMessageID = ReplyToMessageID
}
msg := tgbotapi.NewPhotoUpload(76918703, "tests/image.jpg") _, err := bot.Send(msg)
msg.ReplyToMessageID = 480
_, err = bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -173,15 +145,11 @@ func TestSendWithNewPhotoReply(t *testing.T) {
} }
func TestSendWithExistingPhoto(t *testing.T) { func TestSendWithExistingPhoto(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewPhotoShare(ChatID, ExistingPhotoFileID)
t.Fail()
}
msg := tgbotapi.NewPhotoShare(76918703, "AgADAgADxKcxG4cBswqt13DnHOgbmBxDhCoABC0h01_AL4SKe20BAAEC")
msg.Caption = "Test" msg.Caption = "Test"
_, err = bot.Send(msg) _, err := bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -189,14 +157,10 @@ func TestSendWithExistingPhoto(t *testing.T) {
} }
func TestSendWithNewDocument(t *testing.T) { func TestSendWithNewDocument(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewDocumentUpload(ChatID, "tests/image.jpg")
t.Fail() _, err := bot.Send(msg)
}
msg := tgbotapi.NewDocumentUpload(76918703, "tests/image.jpg")
_, err = bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -204,14 +168,10 @@ func TestSendWithNewDocument(t *testing.T) {
} }
func TestSendWithExistingDocument(t *testing.T) { func TestSendWithExistingDocument(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewDocumentShare(ChatID, ExistingDocumentFileID)
t.Fail() _, err := bot.Send(msg)
}
msg := tgbotapi.NewDocumentShare(76918703, "BQADAgADBwADhwGzCjWgiUU4T8VNAg")
_, err = bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -219,17 +179,13 @@ func TestSendWithExistingDocument(t *testing.T) {
} }
func TestSendWithNewAudio(t *testing.T) { func TestSendWithNewAudio(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewAudioUpload(ChatID, "tests/audio.mp3")
t.Fail()
}
msg := tgbotapi.NewAudioUpload(76918703, "tests/audio.mp3")
msg.Title = "TEST" msg.Title = "TEST"
msg.Duration = 10 msg.Duration = 10
msg.Performer = "TEST" msg.Performer = "TEST"
_, err = bot.Send(msg) _, err := bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -237,18 +193,14 @@ func TestSendWithNewAudio(t *testing.T) {
} }
func TestSendWithExistingAudio(t *testing.T) { func TestSendWithExistingAudio(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewAudioShare(ChatID, ExistingAudioFileID)
t.Fail()
}
msg := tgbotapi.NewAudioShare(76918703, "BQADAgADMwADhwGzCkYFlCTpxiP6Ag")
msg.Title = "TEST" msg.Title = "TEST"
msg.Duration = 10 msg.Duration = 10
msg.Performer = "TEST" msg.Performer = "TEST"
_, err = bot.Send(msg) _, err := bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -256,15 +208,11 @@ func TestSendWithExistingAudio(t *testing.T) {
} }
func TestSendWithNewVoice(t *testing.T) { func TestSendWithNewVoice(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewVoiceUpload(ChatID, "tests/voice.ogg")
t.Fail()
}
msg := tgbotapi.NewVoiceUpload(76918703, "tests/voice.ogg")
msg.Duration = 10 msg.Duration = 10
_, err = bot.Send(msg) _, err := bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -272,15 +220,11 @@ func TestSendWithNewVoice(t *testing.T) {
} }
func TestSendWithExistingVoice(t *testing.T) { func TestSendWithExistingVoice(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewVoiceShare(ChatID, ExistingVoiceFileID)
t.Fail()
}
msg := tgbotapi.NewVoiceShare(76918703, "AwADAgADIgADhwGzCigyMW_GUtWIAg")
msg.Duration = 10 msg.Duration = 10
_, err = bot.Send(msg) _, err := bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -288,13 +232,9 @@ func TestSendWithExistingVoice(t *testing.T) {
} }
func TestSendWithLocation(t *testing.T) { func TestSendWithLocation(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { _, err := bot.Send(tgbotapi.NewLocation(ChatID, 40, 40))
t.Fail()
}
_, err = bot.Send(tgbotapi.NewLocation(76918703, 40, 40))
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -302,105 +242,78 @@ func TestSendWithLocation(t *testing.T) {
} }
func TestSendWithNewVideo(t *testing.T) { func TestSendWithNewVideo(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewVideoUpload(ChatID, "tests/video.mp4")
t.Fail()
}
msg := tgbotapi.NewVideoUpload(76918703, "tests/video.mp4")
msg.Duration = 10 msg.Duration = 10
msg.Caption = "TEST" msg.Caption = "TEST"
_, err = bot.Send(msg) _, err := bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
} }
} }
func TestSendWithExistingVideo(t *testing.T) { func TestSendWithExistingVideo(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewVideoShare(ChatID, ExistingVideoFileID)
t.Fail()
}
msg := tgbotapi.NewVideoShare(76918703, "BAADAgADRgADhwGzCopBeKJ7rv9SAg")
msg.Duration = 10 msg.Duration = 10
msg.Caption = "TEST" msg.Caption = "TEST"
_, err = bot.Send(msg) _, err := bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
} }
} }
func TestSendWithNewSticker(t *testing.T) { func TestSendWithNewSticker(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
msg := tgbotapi.NewStickerUpload(ChatID, "tests/image.jpg")
resp, err := bot.Send(msg)
t.Log(resp)
if err != nil { if err != nil {
t.Fail() t.Fail()
} }
msg := tgbotapi.NewStickerUpload(76918703, "tests/image.jpg")
_, err = bot.Send(msg)
if err != nil {
t.Fail()
}
} }
func TestSendWithExistingSticker(t *testing.T) { func TestSendWithExistingSticker(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
msg := tgbotapi.NewStickerShare(ChatID, ExistingStickerFileID)
_, err := bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
} }
msg := tgbotapi.NewStickerShare(76918703, "BQADAgADUwADhwGzCmwtOypNFlrRAg")
_, err = bot.Send(msg)
if err != nil {
t.Fail()
}
} }
func TestSendWithNewStickerAndKeyboardHide(t *testing.T) { func TestSendWithNewStickerAndKeyboardHide(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewStickerUpload(ChatID, "tests/image.jpg")
t.Fail()
}
msg := tgbotapi.NewStickerUpload(76918703, "tests/image.jpg")
msg.ReplyMarkup = tgbotapi.ReplyKeyboardHide{true, false} msg.ReplyMarkup = tgbotapi.ReplyKeyboardHide{true, false}
_, err = bot.Send(msg) _, err := bot.Send(msg)
if err != nil { if err != nil {
t.Fail() t.Fail()
} }
} }
func TestSendWithExistingStickerAndKeyboardHide(t *testing.T) { func TestSendWithExistingStickerAndKeyboardHide(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { msg := tgbotapi.NewStickerShare(ChatID, ExistingStickerFileID)
t.Fail()
}
msg := tgbotapi.NewStickerShare(76918703, "BQADAgADUwADhwGzCmwtOypNFlrRAg")
msg.ReplyMarkup = tgbotapi.ReplyKeyboardHide{true, false} msg.ReplyMarkup = tgbotapi.ReplyKeyboardHide{true, false}
_, err = bot.Send(msg) _, err := bot.Send(msg)
if err != nil { if err != nil {
@ -409,15 +322,11 @@ func TestSendWithExistingStickerAndKeyboardHide(t *testing.T) {
} }
func TestGetFile(t *testing.T) { func TestGetFile(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { file := tgbotapi.FileConfig{ExistingPhotoFileID}
t.Fail()
}
file := tgbotapi.FileConfig{"BQADAgADBwADhwGzCjWgiUU4T8VNAg"} _, err := bot.GetFile(file)
_, err = bot.GetFile(file)
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -425,13 +334,9 @@ func TestGetFile(t *testing.T) {
} }
func TestSendChatConfig(t *testing.T) { func TestSendChatConfig(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { _, err := bot.Send(tgbotapi.NewChatAction(ChatID, tgbotapi.ChatTyping))
t.Fail()
}
_, err = bot.Send(tgbotapi.NewChatAction(76918703, tgbotapi.ChatTyping))
if err != nil { if err != nil {
t.Fail() t.Fail()
@ -439,24 +344,16 @@ func TestSendChatConfig(t *testing.T) {
} }
func TestGetUserProfilePhotos(t *testing.T) { func TestGetUserProfilePhotos(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil { _, err := bot.GetUserProfilePhotos(tgbotapi.NewUserProfilePhotos(ChatID))
t.Fail()
}
_, err = bot.GetUserProfilePhotos(tgbotapi.NewUserProfilePhotos(76918703))
if err != nil { if err != nil {
t.Fail() t.Fail()
} }
} }
func TestListenForWebhook(t *testing.T) { func TestListenForWebhook(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil {
t.Fail()
}
handler := bot.ListenForWebhook("/") handler := bot.ListenForWebhook("/")
@ -470,16 +367,12 @@ func TestListenForWebhook(t *testing.T) {
} }
func TestSetWebhookWithCert(t *testing.T) { func TestSetWebhookWithCert(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil {
t.Fail()
}
bot.RemoveWebhook() bot.RemoveWebhook()
wh := tgbotapi.NewWebhookWithCert("https://example.com/tgbotapi-test/"+bot.Token, "tests/cert.pem") wh := tgbotapi.NewWebhookWithCert("https://example.com/tgbotapi-test/"+bot.Token, "tests/cert.pem")
_, err = bot.SetWebhook(wh) _, err := bot.SetWebhook(wh)
if err != nil { if err != nil {
t.Fail() t.Fail()
} }
@ -488,16 +381,12 @@ func TestSetWebhookWithCert(t *testing.T) {
} }
func TestSetWebhookWithoutCert(t *testing.T) { func TestSetWebhookWithoutCert(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil {
t.Fail()
}
bot.RemoveWebhook() bot.RemoveWebhook()
wh := tgbotapi.NewWebhook("https://example.com/tgbotapi-test/" + bot.Token) wh := tgbotapi.NewWebhook("https://example.com/tgbotapi-test/" + bot.Token)
_, err = bot.SetWebhook(wh) _, err := bot.SetWebhook(wh)
if err != nil { if err != nil {
t.Fail() t.Fail()
} }
@ -506,15 +395,11 @@ func TestSetWebhookWithoutCert(t *testing.T) {
} }
func TestUpdatesChan(t *testing.T) { func TestUpdatesChan(t *testing.T) {
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_API_TOKEN")) bot, _ := getBot(t)
if err != nil {
t.Fail()
}
var ucfg tgbotapi.UpdateConfig = tgbotapi.NewUpdate(0) var ucfg tgbotapi.UpdateConfig = tgbotapi.NewUpdate(0)
ucfg.Timeout = 60 ucfg.Timeout = 60
err = bot.UpdatesChan(ucfg) err := bot.UpdatesChan(ucfg)
if err != nil { if err != nil {
t.Fail() t.Fail()