package workspace

import (
	"chat-agent-system/internal/db"
	"context"
	"fmt"
	"testing"
	"time"

	"github.com/jackc/pgx/v5"
	"github.com/jackc/pgx/v5/pgxpool"
)

func TestPGRegistryRoundTrip(t *testing.T) {
	dsn := db.SkipIfNoTestDatabaseURL(t)
	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()
	schema := fmt.Sprintf("agentx_reg_%d", time.Now().UnixNano())
	boot, err := pgxpool.New(ctx, dsn)
	if err != nil {
		t.Fatal(err)
	}
	defer boot.Close()
	if _, err := boot.Exec(ctx, "CREATE SCHEMA "+schema); err != nil {
		t.Fatal(err)
	}
	t.Cleanup(func() {
		cctx, ccancel := context.WithTimeout(context.Background(), 10*time.Second)
		defer ccancel()
		_, _ = boot.Exec(cctx, "DROP SCHEMA "+schema+" CASCADE")
	})
	cfg, err := pgxpool.ParseConfig(dsn)
	if err != nil {
		t.Fatal(err)
	}
	cfg.AfterConnect = func(ctx context.Context, conn *pgx.Conn) error {
		_, err := conn.Exec(ctx, "SET search_path TO "+schema)
		return err
	}
	pool, err := pgxpool.NewWithConfig(ctx, cfg)
	if err != nil {
		t.Fatal(err)
	}
	defer pool.Close()
	reg := NewPGRegistry(pool)
	if err := reg.Ensure(ctx); err != nil {
		t.Fatal(err)
	}
	frozen, receipts := sampleFrozen("ws-pg")
	rec := Record{
		ID: "ws-pg", RunID: "run-pg", Bytes: 12,
		CreatedAt: time.Now().UTC().Add(-time.Hour),
		Frozen:    &frozen, Receipts: receipts,
	}
	if err := reg.Upsert(ctx, rec); err != nil {
		t.Fatal(err)
	}
	got, ok, err := reg.Get(ctx, "ws-pg")
	if err != nil || !ok {
		t.Fatalf("get %#v %v %v", got, ok, err)
	}
	if got.RunID != "run-pg" || got.Frozen == nil || got.Frozen.SHA256 != "abc" || len(got.Receipts) != 1 || !got.Receipts[0].Verified {
		t.Fatalf("round trip %#v", got)
	}
}