go-trustlog/api/persistence/standalone_test.go

// Package persistence_test provides standalone tests that don't depend on internal packages
package persistence_test

import (
	"context"
	"database/sql"
	"strings"
	"testing"
	"time"

	_ "github.com/mattn/go-sqlite3"

	"go.yandata.net/iod/iod/go-trustlog/api/persistence"
)

// Standalone tests - 独立测试，不依赖复杂模块
// Run with: go test -v -run Standalone ./api/persistence/

func TestStandaloneConfig(t *testing.T) {
	cfg := persistence.DefaultDBConfig("postgres", "test-dsn")
	if cfg.DriverName != "postgres" {
		t.Errorf("expected DriverName=postgres, got %s", cfg.DriverName)
	}
	if cfg.MaxOpenConns != 25 {
		t.Error("MaxOpenConns should be 25")
	}
}

func TestStandaloneStrategy(t *testing.T) {
	tests := []struct {
		strategy persistence.PersistenceStrategy
		want     string
	}{
		{persistence.StrategyDBOnly, "DB_ONLY"},
		{persistence.StrategyDBAndTrustlog, "DB_AND_TRUSTLOG"},
		{persistence.StrategyTrustlogOnly, "TRUSTLOG_ONLY"},
	}

	for _, tt := range tests {
		got := tt.strategy.String()
		if got != tt.want {
			t.Errorf("strategy.String() = %s, want %s", got, tt.want)
		}
	}
}

func TestStandaloneEnums(t *testing.T) {
	if persistence.StatusNotTrustlogged != "NOT_TRUSTLOGGED" {
		t.Error("StatusNotTrustlogged value incorrect")
	}
	if persistence.StatusTrustlogged != "TRUSTLOGGED" {
		t.Error("StatusTrustlogged value incorrect")
	}
	if persistence.RetryStatusPending != "PENDING" {
		t.Error("RetryStatusPending value incorrect")
	}
	if persistence.RetryStatusRetrying != "RETRYING" {
		t.Error("RetryStatusRetrying value incorrect")
	}
	if persistence.RetryStatusDeadLetter != "DEAD_LETTER" {
		t.Error("RetryStatusDeadLetter value incorrect")
	}
}

func TestStandaloneDDL(t *testing.T) {
	drivers := []string{"postgres", "mysql", "sqlite3", "unknown"}
	
	for _, driver := range drivers {
		opDDL, cursorDDL, retryDDL, err := persistence.GetDialectDDL(driver)
		if err != nil {
			t.Fatalf("GetDialectDDL(%s) failed: %v", driver, err)
		}
		
		if len(opDDL) == 0 {
			t.Errorf("%s: operation DDL is empty", driver)
		}
		if len(cursorDDL) == 0 {
			t.Errorf("%s: cursor DDL is empty", driver)
		}
		if len(retryDDL) == 0 {
			t.Errorf("%s: retry DDL is empty", driver)
		}
		
		// 验证必需字段
		if !strings.Contains(opDDL, "op_id") {
			t.Errorf("%s: missing op_id field", driver)
		}
		if !strings.Contains(opDDL, "client_ip") {
			t.Errorf("%s: missing client_ip field", driver)
		}
		if !strings.Contains(opDDL, "server_ip") {
			t.Errorf("%s: missing server_ip field", driver)
		}
		if !strings.Contains(opDDL, "trustlog_status") {
			t.Errorf("%s: missing trustlog_status field", driver)
		}
	}
}

func TestStandaloneDatabase(t *testing.T) {
	db, err := sql.Open("sqlite3", ":memory:")
	if err != nil {
		t.Fatalf("failed to open database: %v", err)
	}
	defer db.Close()

	// 获取 DDL
	opDDL, cursorDDL, retryDDL, err := persistence.GetDialectDDL("sqlite3")
	if err != nil {
		t.Fatalf("GetDialectDDL failed: %v", err)
	}

	// 创建表
	if _, err := db.Exec(opDDL); err != nil {
		t.Fatalf("failed to create operation table: %v", err)
	}
	if _, err := db.Exec(cursorDDL); err != nil {
		t.Fatalf("failed to create cursor table: %v", err)
	}
	if _, err := db.Exec(retryDDL); err != nil {
		t.Fatalf("failed to create retry table: %v", err)
	}

	// 验证表存在
	tables := []string{"operation", "trustlog_cursor", "trustlog_retry"}
	for _, table := range tables {
		var name string
		err = db.QueryRow(
			"SELECT name FROM sqlite_master WHERE type='table' AND name=?",
			table,
		).Scan(&name)
		if err != nil {
			t.Errorf("table %s not found: %v", table, err)
		}
	}
}

func TestStandaloneIPFields(t *testing.T) {
	db, err := sql.Open("sqlite3", ":memory:")
	if err != nil {
		t.Fatalf("failed to open database: %v", err)
	}
	defer db.Close()

	// 创建表
	opDDL, _, _, _ := persistence.GetDialectDDL("sqlite3")
	if _, err := db.Exec(opDDL); err != nil {
		t.Fatalf("failed to create table: %v", err)
	}

	ctx := context.Background()

	// 测试 NULL IP
	_, err = db.ExecContext(ctx, `
		INSERT INTO operation (
			op_id, doid, producer_id, op_source, op_type,
			do_prefix, do_repository, trustlog_status, timestamp,
			client_ip, server_ip
		) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
	`, "test-001", "10.1000/repo/obj", "producer-001", "DOIP", "Create",
		"10.1000", "repo", "NOT_TRUSTLOGGED", time.Now(), nil, nil)

	if err != nil {
		t.Fatalf("failed to insert with NULL IPs: %v", err)
	}

	var clientIP, serverIP sql.NullString
	err = db.QueryRowContext(ctx,
		"SELECT client_ip, server_ip FROM operation WHERE op_id = ?",
		"test-001",
	).Scan(&clientIP, &serverIP)

	if err != nil {
		t.Fatalf("failed to query: %v", err)
	}

	if clientIP.Valid {
		t.Error("client_ip should be NULL")
	}
	if serverIP.Valid {
		t.Error("server_ip should be NULL")
	}

	// 测试非 NULL IP
	_, err = db.ExecContext(ctx, `
		INSERT INTO operation (
			op_id, doid, producer_id, op_source, op_type,
			do_prefix, do_repository, trustlog_status, timestamp,
			client_ip, server_ip
		) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
	`, "test-002", "10.1000/repo/obj2", "producer-001", "DOIP", "Create",
		"10.1000", "repo", "NOT_TRUSTLOGGED", time.Now(),
		"192.168.1.100", "10.0.0.50")

	if err != nil {
		t.Fatalf("failed to insert with IP values: %v", err)
	}

	err = db.QueryRowContext(ctx,
		"SELECT client_ip, server_ip FROM operation WHERE op_id = ?",
		"test-002",
	).Scan(&clientIP, &serverIP)

	if err != nil {
		t.Fatalf("failed to query: %v", err)
	}

	if !clientIP.Valid || clientIP.String != "192.168.1.100" {
		t.Errorf("client_ip should be '192.168.1.100', got %v", clientIP)
	}
	if !serverIP.Valid || serverIP.String != "10.0.0.50" {
		t.Errorf("server_ip should be '10.0.0.50', got %v", serverIP)
	}
}

func TestStandaloneStatusFlow(t *testing.T) {
	db, err := sql.Open("sqlite3", ":memory:")
	if err != nil {
		t.Fatalf("failed to open database: %v", err)
	}
	defer db.Close()

	opDDL, _, _, _ := persistence.GetDialectDDL("sqlite3")
	if _, err := db.Exec(opDDL); err != nil {
		t.Fatalf("failed to create table: %v", err)
	}

	ctx := context.Background()

	// 插入未存证记录
	_, err = db.ExecContext(ctx, `
		INSERT INTO operation (
			op_id, doid, producer_id, op_source, op_type,
			do_prefix, do_repository, trustlog_status, timestamp
		) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
	`, "test-001", "10.1000/repo/obj", "producer-001", "DOIP", "Create",
		"10.1000", "repo", "NOT_TRUSTLOGGED", time.Now())

	if err != nil {
		t.Fatalf("failed to insert: %v", err)
	}

	// 更新为已存证
	_, err = db.ExecContext(ctx,
		"UPDATE operation SET trustlog_status = ? WHERE op_id = ?",
		"TRUSTLOGGED", "test-001",
	)

	if err != nil {
		t.Fatalf("failed to update: %v", err)
	}

	// 验证状态
	var status string
	err = db.QueryRowContext(ctx,
		"SELECT trustlog_status FROM operation WHERE op_id = ?",
		"test-001",
	).Scan(&status)

	if err != nil {
		t.Fatalf("failed to query: %v", err)
	}

	if status != "TRUSTLOGGED" {
		t.Errorf("expected status=TRUSTLOGGED, got %s", status)
	}
}

func TestStandaloneCursorInit(t *testing.T) {
	db, err := sql.Open("sqlite3", ":memory:")
	if err != nil {
		t.Fatalf("failed to open database: %v", err)
	}
	defer db.Close()

	_, cursorDDL, _, _ := persistence.GetDialectDDL("sqlite3")
	if _, err := db.Exec(cursorDDL); err != nil {
		t.Fatalf("failed to create cursor table: %v", err)
	}

	// 验证表结构（新的 cursor 表不再自动插入初始记录）
	var count int
	err = db.QueryRow("SELECT COUNT(*) FROM trustlog_cursor").Scan(&count)
	if err != nil {
		t.Fatalf("failed to count: %v", err)
	}

	if count != 0 {
		t.Errorf("expected 0 initial cursor records (empty table), got %d", count)
	}
	
	// 测试插入新记录
	_, err = db.Exec(`INSERT INTO trustlog_cursor (cursor_key, cursor_value) VALUES (?, ?)`,
		"test-cursor", time.Now().Format(time.RFC3339Nano))
	if err != nil {
		t.Fatalf("failed to insert cursor: %v", err)
	}
	
	// 验证插入
	err = db.QueryRow("SELECT COUNT(*) FROM trustlog_cursor WHERE cursor_key = ?", "test-cursor").Scan(&count)
	if err != nil {
		t.Fatalf("failed to count after insert: %v", err)
	}
	
	if count != 1 {
		t.Errorf("expected 1 cursor record after insert, got %d", count)
	}
}