FILE / ScuroNeko/Laniakea
keyboard_test.go
Исходный файл и его история в репозитории.
ScuroNeko
945b8240e6
- add centralized message validation errors - switch command handlers to return error - add explicit long plain-text reply helpers - support strict payload type policy with debug logging - document versioning and changelog workflow
98 lines
2.9 KiB
Go
98 lines
2.9 KiB
Go
package laniakea
|
|
|
|
import (
|
|
"errors"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestInlineKeyboardWrapsRowsAndEncodesJSONPayloads(t *testing.T) {
|
|
kb := NewInlineKeyboardJson(2).
|
|
AddCallbackButton("A", "cmd", 1).
|
|
AddCallbackButton("B", "cmd", 2).
|
|
AddCallbackButton("C", "cmd", 3)
|
|
|
|
markup := kb.Get()
|
|
if got := len(markup.InlineKeyboard); got != 2 {
|
|
t.Fatalf("unexpected row count: %d", got)
|
|
}
|
|
if got := len(markup.InlineKeyboard[0]); got != 2 {
|
|
t.Fatalf("unexpected first row size: %d", got)
|
|
}
|
|
if got := len(markup.InlineKeyboard[1]); got != 1 {
|
|
t.Fatalf("unexpected second row size: %d", got)
|
|
}
|
|
if !strings.Contains(markup.InlineKeyboard[0][0].CallbackData, `"cmd":"cmd"`) {
|
|
t.Fatalf("expected JSON callback payload, got %q", markup.InlineKeyboard[0][0].CallbackData)
|
|
}
|
|
}
|
|
|
|
func TestInlineKeyboardBuilderPreservesConfiguredButtonFields(t *testing.T) {
|
|
kb := NewInlineKeyboardBase64(3).
|
|
AddButton(
|
|
NewInlineKbButton("Docs").
|
|
SetStyle(ButtonStylePrimary).
|
|
SetUrl("https://example.test"),
|
|
)
|
|
|
|
button := kb.Get().InlineKeyboard[0][0]
|
|
if button.Style != ButtonStylePrimary {
|
|
t.Fatalf("unexpected style: %q", button.Style)
|
|
}
|
|
if button.URL != "https://example.test" {
|
|
t.Fatalf("unexpected url: %q", button.URL)
|
|
}
|
|
}
|
|
|
|
func TestInlineKeyboardGetPayloadTypeReturnsLocalOverride(t *testing.T) {
|
|
kb := NewInlineKeyboardJson(2)
|
|
if got := kb.GetPayloadType(); got != BotPayloadJson {
|
|
t.Fatalf("unexpected initial payload type: %q", got)
|
|
}
|
|
kb.SetPayloadType(BotPayloadBase64)
|
|
if got := kb.GetPayloadType(); got != BotPayloadBase64 {
|
|
t.Fatalf("unexpected updated payload type: %q", got)
|
|
}
|
|
}
|
|
|
|
func TestDecodePayloadAcceptsBase64KeyboardPayloadWhenBotPrefersJSON(t *testing.T) {
|
|
kb := NewInlineKeyboardBase64(1).
|
|
AddCallbackButton("A", "cmd", 1, "two")
|
|
|
|
got, _, err := decodePayload(BotPayloadJson, kb.Get().InlineKeyboard[0][0].CallbackData, false)
|
|
if err != nil {
|
|
t.Fatalf("decodePayload returned error: %v", err)
|
|
}
|
|
|
|
want := CallbackData{Command: "cmd", Args: []string{"1", "two"}}
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Fatalf("unexpected payload: got %#v want %#v", got, want)
|
|
}
|
|
}
|
|
|
|
func TestDecodePayloadAcceptsJSONKeyboardPayloadWhenBotPrefersBase64(t *testing.T) {
|
|
kb := NewInlineKeyboardJson(1).
|
|
AddCallbackButton("A", "cmd", 1, "two")
|
|
|
|
got, _, err := decodePayload(BotPayloadBase64, kb.Get().InlineKeyboard[0][0].CallbackData, false)
|
|
if err != nil {
|
|
t.Fatalf("decodePayload returned error: %v", err)
|
|
}
|
|
|
|
want := CallbackData{Command: "cmd", Args: []string{"1", "two"}}
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Fatalf("unexpected payload: got %#v want %#v", got, want)
|
|
}
|
|
}
|
|
|
|
func TestDecodePayloadStrictRejectsMismatchedType(t *testing.T) {
|
|
kb := NewInlineKeyboardBase64(1).
|
|
AddCallbackButton("A", "cmd", 1)
|
|
|
|
_, _, err := decodePayload(BotPayloadJson, kb.Get().InlineKeyboard[0][0].CallbackData, true)
|
|
if !errors.Is(err, ErrPayloadTypeMismatch) {
|
|
t.Fatalf("expected ErrPayloadTypeMismatch, got %v", err)
|
|
}
|
|
}
|