chore: establish local platform baseline

2026-08-26 06:41:54 +08:00
commit 29bdf4900f
575 changed files with 95644 additions and 0 deletions
@@ -0,0 +1,75 @@
+package orchestration
+
+import (
+	"context"
+	"os"
+	"testing"
+	"time"
+
+	"data-terminal-cloud/internal/database"
+)
+
+func TestProviderHealthPersistsCircuitAndRecovers(t *testing.T) {
+	databaseURL := os.Getenv("TEST_DATABASE_URL")
+	if databaseURL == "" {
+		t.Skip("TEST_DATABASE_URL is not set")
+	}
+	ctx := context.Background()
+	pool, err := database.Open(ctx, databaseURL, "provider-health-test")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer pool.Close()
+	tenantID := "10000000-0000-0000-0000-000000000001"
+	provider := "provider-health-test"
+	defer pool.Exec(context.Background(), `DELETE FROM provider_dependency_state WHERE tenant_id=$1::uuid AND provider=$2`, tenantID, provider)
+
+	health, err := NewProviderHealth(pool, 3, time.Minute)
+	if err != nil {
+		t.Fatal(err)
+	}
+	now := time.Date(2026, 8, 23, 12, 0, 0, 0, time.UTC)
+	health.now = func() time.Time { return now }
+	for attempt := 1; attempt <= 2; attempt++ {
+		if err := health.RecordFailure(ctx, tenantID, provider, "PROVIDER_START_TEMPORARY"); err != nil {
+			t.Fatal(err)
+		}
+		allowed, err := health.AllowStart(ctx, tenantID, provider)
+		if err != nil || !allowed {
+			t.Fatalf("attempt %d allowed=%v err=%v", attempt, allowed, err)
+		}
+	}
+	if err := health.RecordFailure(ctx, tenantID, provider, "PROVIDER_START_TEMPORARY"); err != nil {
+		t.Fatal(err)
+	}
+	allowed, err := health.AllowStart(ctx, tenantID, provider)
+	if err != nil || allowed {
+		t.Fatalf("open circuit allowed=%v err=%v", allowed, err)
+	}
+	var status, errorCode string
+	var failures int
+	if err := pool.QueryRow(ctx, `SELECT status,consecutive_failures,last_error_code FROM provider_dependency_state WHERE tenant_id=$1::uuid AND provider=$2`, tenantID, provider).Scan(&status, &failures, &errorCode); err != nil {
+		t.Fatal(err)
+	}
+	if status != "UNAVAILABLE" || failures != 3 || errorCode != "PROVIDER_START_TEMPORARY" {
+		t.Fatalf("state=%s failures=%d error=%s", status, failures, errorCode)
+	}
+	now = now.Add(61 * time.Second)
+	allowed, err = health.AllowStart(ctx, tenantID, provider)
+	if err != nil || !allowed {
+		t.Fatalf("cooldown probe allowed=%v err=%v", allowed, err)
+	}
+	if err := health.RecordSuccess(ctx, tenantID, provider); err != nil {
+		t.Fatal(err)
+	}
+	allowed, err = health.AllowStart(ctx, tenantID, provider)
+	if err != nil || !allowed {
+		t.Fatalf("recovered allowed=%v err=%v", allowed, err)
+	}
+	if err := pool.QueryRow(ctx, `SELECT status,consecutive_failures FROM provider_dependency_state WHERE tenant_id=$1::uuid AND provider=$2`, tenantID, provider).Scan(&status, &failures); err != nil {
+		t.Fatal(err)
+	}
+	if status != "AVAILABLE" || failures != 0 {
+		t.Fatalf("recovered state=%s failures=%d", status, failures)
+	}
+}