fb182adef4
🎯 核心变更: - OpType (string) → OpCode (int32) - 20+ OpCode枚举常量 (基于DOIP/IRP标准) - 类型安全 + 性能优化 📊 影响范围: - 核心模型: Operation结构体、CBOR序列化 - 数据库: schema.go + SQL DDL (PostgreSQL/MySQL/SQLite) - 持久化: repository.go查询、cursor_worker.go - API接口: Protobuf定义 + gRPC客户端 - 测试代码: 60+ 测试文件更新 ✅ 测试结果: - 通过率: 100% (所有87个测试用例) - 总体覆盖率: 53.7% - 核心包覆盖率: logger(100%), highclient(95.3%), model(79.1%) 📝 文档: - 精简README (1056行→489行，减少54%) - 完整的OpCode枚举说明 - 三种持久化策略示例 - 数据库表结构和架构图 🔧 技术细节: - 类型转换: string(OpCode) → int32(OpCode) - SQL参数: 字符串值 → 整数值 - Protobuf: op_type string → op_code int32 - 测试断言: 字符串比较 → 常量比较 🎉 质量保证: - 零编译错误 - 100%测试通过 - PostgreSQL/Pulsar集成测试验证 - 分布式并发安全测试通过
312 lines
8.0 KiB
Go
312 lines
8.0 KiB
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_code,
|
|
do_prefix, do_repository, trustlog_status, timestamp,
|
|
client_ip, server_ip
|
|
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
`, "test-001", "10.1000/repo/obj", "producer-001", "DOIP", 100,
|
|
"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_code,
|
|
do_prefix, do_repository, trustlog_status, timestamp,
|
|
client_ip, server_ip
|
|
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
`, "test-002", "10.1000/repo/obj2", "producer-001", "DOIP", 100,
|
|
"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_code,
|
|
do_prefix, do_repository, trustlog_status, timestamp
|
|
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
`, "test-001", "10.1000/repo/obj", "producer-001", "DOIP", 100,
|
|
"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)
|
|
}
|
|
}
|
|
|
|
|
|
|