126 lines
2.9 KiB
Go
126 lines
2.9 KiB
Go
package layout
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
tele "git.belvedersky.ru/common/telebot"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestLayout(t *testing.T) {
|
|
os.Setenv("TOKEN", "TEST")
|
|
|
|
lt, err := New("example.yml")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
pref := lt.Settings()
|
|
assert.Equal(t, "TEST", pref.Token)
|
|
assert.Equal(t, "html", pref.ParseMode)
|
|
assert.Equal(t, &tele.LongPoller{}, pref.Poller)
|
|
|
|
assert.ElementsMatch(t, []tele.Command{{
|
|
Text: "start",
|
|
Description: "Start the bot",
|
|
}, {
|
|
Text: "help",
|
|
Description: "How to use the bot",
|
|
}}, lt.Commands())
|
|
|
|
assert.Equal(t, "string", lt.String("str"))
|
|
assert.Equal(t, 123, lt.Int("num"))
|
|
assert.Equal(t, int64(123), lt.Int64("num"))
|
|
assert.Equal(t, float64(123), lt.Float("num"))
|
|
assert.Equal(t, tele.ChatID(123), lt.ChatID("num"))
|
|
|
|
assert.Equal(t, []string{"abc", "def"}, lt.Strings("strs"))
|
|
assert.Equal(t, []int{123, 456}, lt.Ints("nums"))
|
|
assert.Equal(t, []int64{123, 456}, lt.Int64s("nums"))
|
|
assert.Equal(t, []float64{123, 456}, lt.Floats("nums"))
|
|
|
|
obj := lt.Get("obj")
|
|
assert.NotNil(t, obj)
|
|
|
|
const dur = 10 * time.Minute
|
|
assert.Equal(t, dur, obj.Duration("dur"))
|
|
assert.True(t, lt.Duration("obj.dur") == obj.Duration("dur"))
|
|
|
|
arr := lt.Slice("arr")
|
|
assert.Len(t, arr, 2)
|
|
|
|
for _, v := range arr {
|
|
assert.Equal(t, dur, v.Duration("dur"))
|
|
}
|
|
|
|
assert.Equal(t, &tele.Btn{
|
|
Unique: "pay",
|
|
Text: "Pay",
|
|
Data: "1|100.00|USD",
|
|
}, lt.ButtonLocale("en", "pay", struct {
|
|
UserID int
|
|
Amount string
|
|
Currency string
|
|
}{
|
|
UserID: 1,
|
|
Amount: "100.00",
|
|
Currency: "USD",
|
|
}))
|
|
|
|
assert.Equal(t, &tele.ReplyMarkup{
|
|
ReplyKeyboard: [][]tele.ReplyButton{
|
|
{{Text: "Help"}},
|
|
{{Text: "Settings"}},
|
|
},
|
|
ResizeKeyboard: true,
|
|
}, lt.MarkupLocale("en", "reply_shortened"))
|
|
|
|
assert.Equal(t, &tele.ReplyMarkup{
|
|
ReplyKeyboard: [][]tele.ReplyButton{{{Text: "Send a contact", Contact: true}}},
|
|
ResizeKeyboard: true,
|
|
OneTimeKeyboard: true,
|
|
}, lt.MarkupLocale("en", "reply_extended"))
|
|
|
|
assert.Equal(t, &tele.ReplyMarkup{
|
|
InlineKeyboard: [][]tele.InlineButton{{
|
|
{
|
|
Unique: "stop",
|
|
Text: "Stop",
|
|
Data: "1",
|
|
},
|
|
}},
|
|
}, lt.MarkupLocale("en", "inline", 1))
|
|
|
|
assert.Equal(t, &tele.ReplyMarkup{
|
|
InlineKeyboard: [][]tele.InlineButton{{
|
|
{
|
|
Text: "This is a web app",
|
|
WebApp: &tele.WebApp{URL: "https://google.com"},
|
|
},
|
|
}},
|
|
}, lt.MarkupLocale("en", "web_app"))
|
|
|
|
assert.Equal(t, &tele.ArticleResult{
|
|
ResultBase: tele.ResultBase{
|
|
ID: "1853",
|
|
Type: "article",
|
|
},
|
|
Title: "Some title",
|
|
Description: "Some description",
|
|
ThumbURL: "https://preview.picture",
|
|
Text: "This is an article.",
|
|
}, lt.ResultLocale("en", "article", struct {
|
|
ID int
|
|
Title string
|
|
Description string
|
|
PreviewURL string
|
|
}{
|
|
ID: 1853,
|
|
Title: "Some title",
|
|
Description: "Some description",
|
|
PreviewURL: "https://preview.picture",
|
|
}))
|
|
}
|