astra/owpengram-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: refresh gramsrv public release

Browse source
This commit is contained in:
A 2026-06-30 14:37:43 +08:00
parent 75cebe8dbf
commit 70b6820474
1274 changed files with 378751 additions and 59919 deletions
Download patch file Download diff file Expand all files Collapse all files

77
internal/domain/message_markup_test.go Normal file
View file

@ -0,0 +1,77 @@
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")
}
}