Function TestExplainAPIError

ai_api_test.go:173–212

TestExplainAPIError tests handling of API errors.

(t *testing.T)

// TestExplainAPIError tests handling of API errors.
func TestExplainAPIError(t *testing.T) {
	// Create a mock server that returns an error
	mockServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		resp := aiChatResponse{
			Error: &struct {
				Message string `json:"message"`
			}{
				Message: "Invalid API key",
			},
		}

		w.Header().Set("Content-Type", "application/json")
		w.WriteHeader(http.StatusUnauthorized)
		json.NewEncoder(w).Encode(resp)
	}))
	defer mockServer.Close()

	// Create enforcer
	e, err := NewEnforcer("examples/basic_model.conf", "examples/basic_policy.csv")
	if err != nil {
		t.Fatal(err)
	}

	// Set AI config
	e.SetAIConfig(AIConfig{
		Endpoint: mockServer.URL,
		APIKey:   "invalid-key",
		Model:    "gpt-3.5-turbo",
		Timeout:  5 * time.Second,
	})

	// Test that API error is properly handled
	_, err = e.Explain("alice", "data1", "read")
	if err == nil {
		t.Error("Expected error for API failure")
	}
	if !strings.Contains(err.Error(), "Invalid API key") {
		t.Errorf("Expected API error message, got: %v", err)
	}
}

// TestBuildExplainContext tests the context building function.
func TestBuildExplainContext(t *testing.T) {