feat: enhance audit logging tests with metadata validation and context management

Author: ToeiRei

core/db/db_wrappers_test.go

@@ -4,6 +4,8 @@
 package db
 
 import (
+	"context"
+	"database/sql"
 	"testing"
 	"time"
 
@@ -163,6 +165,34 @@ func TestAccountAndKeyWrappers(t *testing.T) {
 			t.Fatalf("expected audit entry not found: %+v", entries)
 		}
 
+		SetAuditContext("cli", "unit-test")
+		if err := LogAction("TEST_META", "details"); err != nil {
+			t.Fatalf("LogAction with metadata: %v", err)
+		}
+		var meta struct {
+			Hostname   sql.NullString `bun:"hostname"`
+			ClientImpl sql.NullString `bun:"client_impl"`
+			Referrer   sql.NullString `bun:"referrer"`
+		}
+		if err := QueryRawInto(
+			context.Background(),
+			BunDB(),
+			&meta,
+			"SELECT hostname, client_impl, referrer FROM audit_log WHERE action = ? ORDER BY id DESC LIMIT 1",
+			"TEST_META",
+		); err != nil {
+			t.Fatalf("query metadata columns: %v", err)
+		}
+		if !meta.Hostname.Valid {
+			t.Fatalf("expected hostname to be captured")
+		}
+		if !meta.ClientImpl.Valid || meta.ClientImpl.String != "cli" {
+			t.Fatalf("unexpected client_impl: %+v", meta.ClientImpl)
+		}
+		if !meta.Referrer.Valid || meta.Referrer.String != "unit-test" {
+			t.Fatalf("unexpected referrer: %+v", meta.Referrer)
+		}
+
 		// Export/Import backup paths (basic round-trip)
 		b, err := ExportDataForBackup()
 		if err != nil {

core/db/testhelpers_test.go

@@ -20,6 +20,7 @@ func WithTestStore(t *testing.T, fn func(s *BunStore)) {
 	prevDefaultAccountManager := defaultAccountManager
 	prevDefaultKeyManager := defaultKeyManager
 	prevDefaultAuditWriter := defaultAuditWriter
+	prevAuditContext := getAuditContext()
 
 	// Initialize in-memory sqlite DB for this test
 	dsn := "file:" + t.Name() + "?mode=memory&cache=shared"
@@ -40,6 +41,7 @@ func WithTestStore(t *testing.T, fn func(s *BunStore)) {
 		defaultAccountManager = prevDefaultAccountManager
 		defaultKeyManager = prevDefaultKeyManager
 		defaultAuditWriter = prevDefaultAuditWriter
+		SetAuditContext(prevAuditContext.ClientImplementation, prevAuditContext.Referrer)
 	}()
 
 	fn(s)