Function TestEnforcerWithDefaultLogger

logger_test.go:68–137

(t *testing.T)

}

func TestEnforcerWithDefaultLogger(t *testing.T) {
	// Create enforcer with RBAC model and policy
	e, err := NewEnforcer("examples/rbac_model.conf", "examples/rbac_policy.csv")
	if err != nil {
		t.Fatalf("Failed to create enforcer: %v", err)
	}

	// Create a buffer to capture log output
	var buf bytes.Buffer
	logger := log.NewDefaultLogger()
	logger.SetOutput(&buf)

	// Set up a callback to track log entries
	var callbackEntries []*log.LogEntry
	err = logger.SetLogCallback(func(entry *log.LogEntry) error {
		// Create a copy of the entry to store
		entryCopy := *entry
		callbackEntries = append(callbackEntries, &entryCopy)
		return nil
	})
	if err != nil {
		t.Fatalf("Failed to set log callback: %v", err)
	}

	// Set the logger on the enforcer
	e.SetLogger(logger)

	// Test Enforce events
	if result, err := e.Enforce("alice", "data1", "read"); err != nil {
		t.Fatalf("Enforce failed: %v", err)
	} else if !result {
		t.Errorf("Expected alice to have read access to data1")
	}

	if result, err := e.Enforce("bob", "data2", "write"); err != nil {
		t.Fatalf("Enforce failed: %v", err)
	} else if !result {
		t.Errorf("Expected bob to have write access to data2")
	}

	// Test AddPolicy event
	if added, err := e.AddPolicy("charlie", "data3", "read"); err != nil {
		t.Fatalf("AddPolicy failed: %v", err)
	} else if !added {
		t.Errorf("Expected policy to be added")
	}

	// Test RemovePolicy event
	if removed, err := e.RemovePolicy("charlie", "data3", "read"); err != nil {
		t.Fatalf("RemovePolicy failed: %v", err)
	} else if !removed {
		t.Errorf("Expected policy to be removed")
	}

	// Test SavePolicy and LoadPolicy events
	if err := e.SavePolicy(); err != nil {
		t.Fatalf("SavePolicy failed: %v", err)
	}