Fix build fail

Author: ginaxu1

audit-service/handlers/data_exchange_event.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"net/http"
 	"strconv"
+	"strings"
 	"time"
 
 	"github.com/gov-dx-sandbox/audit-service/models"
@@ -41,6 +42,13 @@ func (h *DataExchangeEventHandler) CreateDataExchangeEvent(w http.ResponseWriter
 	// Create event
 	event, err := h.DataExchangeEventService.CreateDataExchangeEvent(r.Context(), &req)
 	if err != nil {
+		// Check for validation errors
+		errMsg := err.Error()
+		if strings.Contains(errMsg, "required") || strings.Contains(errMsg, "invalid") {
+			http.Error(w, fmt.Sprintf("Invalid request: %v", err), http.StatusBadRequest)
+			return
+		}
+		
 		http.Error(w, fmt.Sprintf("Failed to create event log: %v", err), http.StatusInternalServerError)
 		return
 	}

audit-service/main.go

@@ -13,6 +13,7 @@ import (
 
 	"github.com/gov-dx-sandbox/audit-service/handlers"
 	"github.com/gov-dx-sandbox/audit-service/middleware"
+	"github.com/gov-dx-sandbox/audit-service/models"
 	"github.com/gov-dx-sandbox/audit-service/services"
 )
 
@@ -50,6 +51,20 @@ func main() {
 		os.Exit(1)
 	}
 
+	// Enable pgcrypto extension for UUID generation
+	if err := gormDB.Exec("CREATE EXTENSION IF NOT EXISTS pgcrypto").Error; err != nil {
+		slog.Error("Failed to enable pgcrypto extension", "error", err)
+		os.Exit(1)
+	}
+
+	// Auto-migrate database schema
+	slog.Info("Running database migrations...")
+	if err := gormDB.AutoMigrate(&models.DataExchangeEvent{}, &models.ManagementEvent{}); err != nil {
+		slog.Error("Failed to migrate database", "error", err)
+		os.Exit(1)
+	}
+	slog.Info("Database migrations completed successfully")
+
 	// Initialize services
 	dataExchangeEventService := services.NewDataExchangeEventService(gormDB)
 	managementEventService := services.NewManagementEventService(gormDB)

audit-service/services/data_exchange_event.go

@@ -35,6 +35,17 @@ func (s *DataExchangeEventService) CreateDataExchangeEvent(ctx context.Context,
 		return nil, fmt.Errorf("invalid status: %s. Must be success or failure", req.Status)
 	}
 
+	// Validate required fields
+	if req.ApplicationID == "" {
+		return nil, fmt.Errorf("applicationId is required")
+	}
+	if req.SchemaID == "" {
+		return nil, fmt.Errorf("schemaId is required")
+	}
+	if len(req.RequestedData) == 0 {
+		return nil, fmt.Errorf("requestedData is required")
+	}
+
 	// Create the event record
 	event := &models.DataExchangeEvent{
 		Timestamp:         timestamp,

tests/integration/audit/audit_test.go

@@ -231,7 +231,7 @@ func TestAudit_FilterByStatus(t *testing.T) {
 
 	// Filter by success status
 	params := url.Values{}
-	params.Add("transaction_status", "SUCCESS")
+	params.Add("status", "success")
 	resp, err = http.Get(auditBaseURL + "/api/data-exchange-events?" + params.Encode())
 	require.NoError(t, err)
 	defer resp.Body.Close()
@@ -461,13 +461,19 @@ func TestAudit_DatabaseVerification(t *testing.T) {
 func TestAudit_CreateManagementEvent(t *testing.T) {
 	timestamp := time.Now().UTC().Format(time.RFC3339)
 	createReq := map[string]interface{}{
-		"timestamp":    timestamp,
-		"eventType":    "schema_created",
-		"userId":       "test-user-123",
-		"resourceType": "schema",
-		"resourceId":   "test-schema-123",
-		"action":       "create",
-		"details": map[string]interface{}{
+		"timestamp": timestamp,
+		"eventType": "CREATE",
+		"status":    "success",
+		"actor": map[string]interface{}{
+			"type": "USER",
+			"id":   "test-user-123",
+			"role": "ADMIN",
+		},
+		"target": map[string]interface{}{
+			"resource":   "SCHEMAS",
+			"resourceId": "test-schema-123",
+		},
+		"metadata": map[string]interface{}{
 			"schemaName": "Test Schema",
 		},
 	}