package v071

import (
	"context"
	"os"
	"testing"

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

func TestPostgresExporterV071Integration(t *testing.T) {
	databaseURL := os.Getenv("TEST_WEKNORA_V071_DATABASE_URL")
	if databaseURL == "" {
		t.Skip("TEST_WEKNORA_V071_DATABASE_URL is not set")
	}
	ctx := context.Background()
	pool, err := pgxpool.New(ctx, databaseURL)
	if err != nil {
		t.Fatal(err)
	}
	defer pool.Close()
	exporter, err := NewPostgresExporter(pool)
	if err != nil {
		t.Fatal(err)
	}
	snapshot, err := exporter.Export(ctx, PostgresIdentity{TenantID: 10000, KnowledgeBaseID: "kb-release-safe", KnowledgeID: "knowledge-release-safe"})
	if err != nil {
		t.Fatal(err)
	}
	for table, count := range map[string]int{"knowledges": 1, "chunks": 1, "knowledge_tags": 1, "knowledge_tag_relations": 1, "resources": 2, "resource_bindings": 1} {
		actual := snapshot.Table(table)
		if actual.RowCount != count || len(actual.SHA256) != 64 {
			t.Fatalf("%s=%#v", table, actual)
		}
	}
}