Function TestExplainDenied

ai_api_test.go:121–170

TestExplainDenied tests Explain for a denied request.

(t *testing.T)

// TestExplainDenied tests Explain for a denied request.
func TestExplainDenied(t *testing.T) {
	// Create a mock server
	mockServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		resp := aiChatResponse{
			Choices: []struct {
				Message aiMessage `json:"message"`
			}{
				{
					Message: aiMessage{
						Role:    "assistant",
						Content: "The request was denied because there is no policy rule that allows alice to write to data1.",
					},
				},
			},
		}

		w.Header().Set("Content-Type", "application/json")
		w.WriteHeader(http.StatusOK)
		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:   "test-api-key",
		Model:    "gpt-3.5-turbo",
		Timeout:  5 * time.Second,
	})

	// Test explanation for denied request
	explanation, err := e.Explain("alice", "data1", "write")
	if err != nil {
		t.Fatalf("Failed to get explanation: %v", err)
	}

	if explanation == "" {
		t.Error("Expected non-empty explanation")
	}

	if !strings.Contains(explanation, "denied") {
		t.Errorf("Expected explanation to mention 'denied', got: %s", explanation)
	}
}

// TestExplainAPIError tests handling of API errors.
func TestExplainAPIError(t *testing.T) {