mirror of
https://github.com/tiennm99/miti99bot.git
synced 2026-06-08 08:16:03 +00:00
tiennm99
daeaf0c605
The CF→AWS data migration (closed 2026-05-16) is long done and the tooling isn't wired into any production path. Remove the one-shot binary, its support package, and the migration runbook. In live code, replace 'JS-parity' / 'same shape as JS' / 'cross-runtime KV migration' comments with the real, stable reason for each behavior (wire-format invariant, null-vs-zero distinction, CloudWatch alarm field name, etc.). 24 files touched across lolschedule, loldle, wordle, twentyq, trading, misc, util, server, metrics, ai, keylock. - delete cmd/migrate_cf_data/ - delete internal/migration/ - delete docs/cf-to-aws-migration-runbook.md
84 lines
2.4 KiB
Go
84 lines
2.4 KiB
Go
package misc
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/tiennm99/miti99bot/internal/modules"
|
|
"github.com/tiennm99/miti99bot/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,
|
|
"trongtruonghop": modules.VisibilityPublic,
|
|
}
|
|
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 as a ms-epoch number
|
|
// (not an RFC3339 string), matching every other timestamp in the bot's KV.
|
|
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)
|
|
}
|
|
|
|
// Verify a hand-written {"at": <ms-epoch>} document 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 ms-epoch shape: %v", err)
|
|
}
|
|
if got.At != 1700000000000 {
|
|
t.Errorf("ms-epoch 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)
|
|
}
|
|
}
|