mirror of
https://github.com/memohai/Memoh.git
synced 2026-04-27 07:16:19 +09:00
BBQ
bc374fe8cd
* refactor(attachment): multimodal attachment refactor with snapshot schema and storage layer - Add snapshot schema migration (0008) and update init/versions/snapshots - Add internal/attachment and internal/channel normalize for unified attachment handling - Move containerfs provider from internal/media to internal/storage - Update agent types, channel adapters (Telegram/Feishu), inbound and handlers - Add containerd snapshot lineage and local_channel tests - Regenerate sqlc, swagger and SDK * refactor(media): content-addressed asset system with unified naming - Replace asset_id foreign key with content_hash as sole identifier for bot_history_message_assets (pure soft-link model) - Remove mime, size_bytes, storage_key from DB; derive at read time via media.Resolve from actual storage - Merge migrations 0008/0009 into single 0008; keep 0001 as canonical schema - Add Docker initdb script for deterministic migration execution order - Fix cross-channel real-time image display (Telegram → WebUI SSE) - Fix message disappearing on refresh (null assets fallback) - Fix file icon instead of image preview (mime derivation from storage) - Unify AssetID → ContentHash naming across Go, Agent, and Frontend - Change storage key prefix from 4-char to 2-char for directory sharding - Add server-entrypoint.sh for Docker deployment migration handling * refactor(infra): embedded migrations, Docker simplification, and config consolidation - Embed SQL migrations into Go binary, removing shell-based migration scripts - Consolidate config files into conf/ directory (app.example.toml, app.docker.toml, app.dev.toml) - Simplify Docker setup: remove initdb.d scripts, streamline nginx config and entrypoint - Remove legacy CLI, feishu-echo commands, and obsolete incremental migration files - Update install script and docs to require sudo for one-click install - Add mise tasks for dev environment orchestration * chore: recover migrations --------- Co-authored-by: Acbox <acbox0328@gmail.com>
88 lines
2.2 KiB
Go
88 lines
2.2 KiB
Go
package handlers
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"encoding/json"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
type testFlusher struct{}
|
|
|
|
func (f *testFlusher) Flush() {}
|
|
|
|
func TestParseSinceParam(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
now := time.Now().UTC().Truncate(time.Second)
|
|
parsed, ok, err := parseSinceParam(now.Format(time.RFC3339))
|
|
if err != nil {
|
|
t.Fatalf("parse RFC3339 failed: %v", err)
|
|
}
|
|
if !ok {
|
|
t.Fatalf("expected parseSinceParam ok=true")
|
|
}
|
|
if !parsed.Equal(now) {
|
|
t.Fatalf("expected parsed time %s, got %s", now, parsed)
|
|
}
|
|
|
|
parsedEpoch, ok, err := parseSinceParam("1735689600000")
|
|
if err != nil {
|
|
t.Fatalf("parse epoch millis failed: %v", err)
|
|
}
|
|
if !ok {
|
|
t.Fatalf("expected epoch parse ok=true")
|
|
}
|
|
if parsedEpoch.UnixMilli() != 1735689600000 {
|
|
t.Fatalf("expected parsed epoch millis 1735689600000, got %d", parsedEpoch.UnixMilli())
|
|
}
|
|
|
|
if _, _, err := parseSinceParam("invalid-time"); err == nil {
|
|
t.Fatalf("expected invalid since parameter error")
|
|
}
|
|
}
|
|
|
|
func TestParseBeforeParam(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
if _, ok := parseBeforeParam(""); ok {
|
|
t.Fatalf("expected empty before value to be ignored")
|
|
}
|
|
parsed, ok := parseBeforeParam("1735689600000")
|
|
if !ok {
|
|
t.Fatalf("expected epoch millis before value to parse")
|
|
}
|
|
if parsed.UnixMilli() != 1735689600000 {
|
|
t.Fatalf("expected parsed epoch millis 1735689600000, got %d", parsed.UnixMilli())
|
|
}
|
|
}
|
|
|
|
func TestWriteSSEJSON(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
var output bytes.Buffer
|
|
writer := bufio.NewWriter(&output)
|
|
flusher := &testFlusher{}
|
|
|
|
if err := writeSSEJSON(writer, flusher, map[string]any{"type": "ping"}); err != nil {
|
|
t.Fatalf("writeSSEJSON failed: %v", err)
|
|
}
|
|
raw := output.String()
|
|
if !strings.HasPrefix(raw, "data: ") {
|
|
t.Fatalf("expected SSE data prefix, got %q", raw)
|
|
}
|
|
if !strings.HasSuffix(raw, "\n\n") {
|
|
t.Fatalf("expected SSE payload suffix, got %q", raw)
|
|
}
|
|
payloadText := strings.TrimSuffix(strings.TrimPrefix(raw, "data: "), "\n\n")
|
|
var payload map[string]any
|
|
if err := json.Unmarshal([]byte(payloadText), &payload); err != nil {
|
|
t.Fatalf("decode SSE payload failed: %v", err)
|
|
}
|
|
if payload["type"] != "ping" {
|
|
t.Fatalf("expected payload type ping, got %#v", payload["type"])
|
|
}
|
|
}
|