mirror of https://github.com/tucnak/telebot
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.5 KiB
Go
49 lines
1.5 KiB
Go
package telebot
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestBtn(t *testing.T) {
|
|
r := &ReplyMarkup{}
|
|
|
|
assert.Equal(t, &ReplyButton{Text: "T"}, r.Text("T").Reply())
|
|
assert.Equal(t, &ReplyButton{Text: "T", Contact: true}, r.Contact("T").Reply())
|
|
assert.Equal(t, &ReplyButton{Text: "T", Location: true}, r.Location("T").Reply())
|
|
assert.Equal(t, &ReplyButton{Text: "T", Poll: PollAny}, r.Poll("T", PollAny).Reply())
|
|
|
|
assert.Nil(t, r.Data("T", "u").Reply())
|
|
assert.Equal(t, &InlineButton{Unique: "u", Text: "T"}, r.Data("T", "u").Inline())
|
|
assert.Equal(t, &InlineButton{Unique: "u", Text: "T", Data: "1|2"}, r.Data("T", "u", "1", "2").Inline())
|
|
assert.Equal(t, &InlineButton{Text: "T", URL: "url"}, r.URL("T", "url").Inline())
|
|
assert.Equal(t, &InlineButton{Text: "T", InlineQuery: "q"}, r.Query("T", "q").Inline())
|
|
assert.Equal(t, &InlineButton{Text: "T", InlineQueryChat: "q"}, r.QueryChat("T", "q").Inline())
|
|
assert.Equal(t, &InlineButton{Text: "T", Login: &Login{Text: "T"}}, r.Login("T", &Login{Text: "T"}).Inline())
|
|
}
|
|
|
|
func TestReplyInline(t *testing.T) {
|
|
r := &ReplyMarkup{}
|
|
r.Reply(
|
|
r.Row(r.Text("Menu")),
|
|
r.Row(r.Text("Settings")),
|
|
)
|
|
|
|
assert.Equal(t, [][]ReplyButton{
|
|
{{Text: "Menu"}},
|
|
{{Text: "Settings"}},
|
|
}, r.ReplyKeyboard)
|
|
|
|
i := &ReplyMarkup{}
|
|
i.Inline(r.Row(
|
|
r.Data("Previous", "prev"),
|
|
r.Data("Next", "next"),
|
|
))
|
|
|
|
assert.Equal(t, [][]InlineButton{{
|
|
{Unique: "prev", Text: "Previous"},
|
|
{Unique: "next", Text: "Next"},
|
|
}}, i.InlineKeyboard)
|
|
}
|