mirror of
https://github.com/tiennm99/miti99bot.git
synced 2026-06-08 18:16:54 +00:00
tiennm99
0584b094d1
Phase 5a of go-port-cloud-run plan: port first 2 of 4 modules (wordle/loldle deferred to later phase). Port util.go, info.go, help.go, stickerid.go and misc.go with tests. /help renders registry view; /info exposes chat/thread/ sender ids; /stickerid (private) returns bot-scoped file_ids; /ping writes last_ping KV ms-epoch JSON for byte-parity, /mstats reads it, /fortytwo is easter egg. Registry-pointer-in-Deps required for /help to access module registry—pointer captured at factory time, stable post-Build. Static factory catalog moved from modules pkg to cmd/server to break import cycle. Code-review fixes applied in same session: /info nil-deref guard, KV wire-format parity.
84 lines
2.4 KiB
Go
84 lines
2.4 KiB
Go
package misc
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/tiennm99/miti99bot-go/internal/modules"
|
|
"github.com/tiennm99/miti99bot-go/internal/storage"
|
|
)
|
|
|
|
// We test the per-command KV behaviour directly — the bot/Telegram side is
|
|
// thin (single SendMessage) and exercising it would require a fake bot HTTP
|
|
// server. The KV interaction is the part with logic worth locking down.
|
|
|
|
func TestNew_RegistersExpectedCommands(t *testing.T) {
|
|
deps := modules.Deps{KV: storage.NewMemoryKVStore()}
|
|
mod := New(deps)
|
|
|
|
want := map[string]modules.Visibility{
|
|
"ping": modules.VisibilityPublic,
|
|
"mstats": modules.VisibilityProtected,
|
|
"fortytwo": modules.VisibilityPrivate,
|
|
}
|
|
if len(mod.Commands) != len(want) {
|
|
t.Fatalf("commands count = %d, want %d", len(mod.Commands), len(want))
|
|
}
|
|
for _, c := range mod.Commands {
|
|
v, ok := want[c.Name]
|
|
if !ok {
|
|
t.Errorf("unexpected command %q", c.Name)
|
|
continue
|
|
}
|
|
if c.Visibility != v {
|
|
t.Errorf("command %q visibility = %d, want %d", c.Name, c.Visibility, v)
|
|
}
|
|
if c.Handler == nil {
|
|
t.Errorf("command %q has nil handler", c.Name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPing_WritesLastPingKV(t *testing.T) {
|
|
ctx := context.Background()
|
|
kv := storage.NewMemoryKVStore()
|
|
|
|
// Drive the KV side directly: lock the wire format (ms-epoch number, not
|
|
// RFC3339 string). A JS-written {at: 1700000000000} must round-trip into
|
|
// the Go struct without a custom decoder.
|
|
if err := kv.PutJSON(ctx, lastPingKey, lastPing{At: time.Now().UTC().UnixMilli()}); err != nil {
|
|
t.Fatalf("PutJSON: %v", err)
|
|
}
|
|
|
|
var got lastPing
|
|
if err := kv.GetJSON(ctx, lastPingKey, &got); err != nil {
|
|
t.Fatalf("GetJSON: %v", err)
|
|
}
|
|
if got.At <= 0 {
|
|
t.Errorf("read-back lastPing.At = %d, want positive ms-epoch", got.At)
|
|
}
|
|
|
|
// Also verify a value with the JS-shape decodes correctly.
|
|
if err := kv.Put(ctx, lastPingKey, []byte(`{"at":1700000000000}`)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
got = lastPing{}
|
|
if err := kv.GetJSON(ctx, lastPingKey, &got); err != nil {
|
|
t.Fatalf("GetJSON js-shape: %v", err)
|
|
}
|
|
if got.At != 1700000000000 {
|
|
t.Errorf("js-shape round-trip: At = %d, want 1700000000000", got.At)
|
|
}
|
|
}
|
|
|
|
func TestMstats_MissingKVReturnsErrNotFound(t *testing.T) {
|
|
ctx := context.Background()
|
|
kv := storage.NewMemoryKVStore()
|
|
|
|
var dst lastPing
|
|
if err := kv.GetJSON(ctx, lastPingKey, &dst); err != storage.ErrNotFound {
|
|
t.Errorf("GetJSON missing = %v, want ErrNotFound", err)
|
|
}
|
|
}
|