77 lines
3.2 KiB
Go
77 lines
3.2 KiB
Go
package domain
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func cb(text string, data []byte) MarkupButton {
|
|
return MarkupButton{Type: MarkupButtonCallback, Text: text, Data: data}
|
|
}
|
|
|
|
func TestValidateReplyMarkup(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
m *MessageReplyMarkup
|
|
want error
|
|
}{
|
|
{"nil ok", nil, nil},
|
|
{"empty ok", &MessageReplyMarkup{}, nil},
|
|
{"callback ok", &MessageReplyMarkup{Inline: [][]MarkupButton{{cb("ok", []byte("d"))}}}, nil},
|
|
{"callback 64B ok", &MessageReplyMarkup{Inline: [][]MarkupButton{{cb("ok", make([]byte, 64))}}}, nil},
|
|
{"callback 65B bad", &MessageReplyMarkup{Inline: [][]MarkupButton{{cb("ok", make([]byte, 65))}}}, ErrButtonDataInvalid},
|
|
{"empty text bad", &MessageReplyMarkup{Inline: [][]MarkupButton{{cb("", []byte("d"))}}}, ErrButtonInvalid},
|
|
{"url https ok", &MessageReplyMarkup{Inline: [][]MarkupButton{{{Type: MarkupButtonURL, Text: "go", URL: "https://example.com/x"}}}}, nil},
|
|
{"url http bad", &MessageReplyMarkup{Inline: [][]MarkupButton{{{Type: MarkupButtonURL, Text: "go", URL: "http://example.com"}}}}, ErrButtonURLInvalid},
|
|
{"url javascript bad", &MessageReplyMarkup{Inline: [][]MarkupButton{{{Type: MarkupButtonURL, Text: "go", URL: "javascript:alert(1)"}}}}, ErrButtonURLInvalid},
|
|
{"url empty bad", &MessageReplyMarkup{Inline: [][]MarkupButton{{{Type: MarkupButtonURL, Text: "go", URL: ""}}}}, ErrButtonURLInvalid},
|
|
{"unknown type bad", &MessageReplyMarkup{Inline: [][]MarkupButton{{{Type: "webview", Text: "x"}}}}, ErrButtonTypeInvalid},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if err := ValidateReplyMarkup(tt.m); !errors.Is(err, tt.want) {
|
|
t.Fatalf("ValidateReplyMarkup = %v, want %v", err, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestValidateReplyMarkupLimits(t *testing.T) {
|
|
// 行数上限。
|
|
tooManyRows := &MessageReplyMarkup{Inline: make([][]MarkupButton, MaxMarkupRows+1)}
|
|
for i := range tooManyRows.Inline {
|
|
tooManyRows.Inline[i] = []MarkupButton{cb("x", []byte("d"))}
|
|
}
|
|
if err := ValidateReplyMarkup(tooManyRows); !errors.Is(err, ErrButtonInvalid) {
|
|
t.Fatalf("rows over limit = %v, want ErrButtonInvalid", err)
|
|
}
|
|
// 单行按钮数上限。
|
|
wideRow := make([]MarkupButton, MaxMarkupButtonsPerRow+1)
|
|
for i := range wideRow {
|
|
wideRow[i] = cb("x", []byte("d"))
|
|
}
|
|
if err := ValidateReplyMarkup(&MessageReplyMarkup{Inline: [][]MarkupButton{wideRow}}); !errors.Is(err, ErrButtonInvalid) {
|
|
t.Fatalf("row width over limit = %v, want ErrButtonInvalid", err)
|
|
}
|
|
// 文本长度上限。
|
|
longText := strings.Repeat("a", MaxMarkupButtonTextLen+1)
|
|
if err := ValidateReplyMarkup(&MessageReplyMarkup{Inline: [][]MarkupButton{{cb(longText, []byte("d"))}}}); !errors.Is(err, ErrButtonInvalid) {
|
|
t.Fatalf("text over limit = %v, want ErrButtonInvalid", err)
|
|
}
|
|
}
|
|
|
|
func TestMessageReplyMarkupIsZero(t *testing.T) {
|
|
if !(*MessageReplyMarkup)(nil).IsZero() {
|
|
t.Fatal("nil markup must be zero")
|
|
}
|
|
if !(&MessageReplyMarkup{}).IsZero() {
|
|
t.Fatal("empty markup must be zero")
|
|
}
|
|
if !(&MessageReplyMarkup{Inline: [][]MarkupButton{{}}}).IsZero() {
|
|
t.Fatal("markup with only empty rows must be zero")
|
|
}
|
|
if (&MessageReplyMarkup{Inline: [][]MarkupButton{{cb("x", nil)}}}).IsZero() {
|
|
t.Fatal("markup with a button must not be zero")
|
|
}
|
|
}
|