adapter_test.go

package pgxadapter

import (
	"os"
	"sort"
	"strings"
	"testing"

	"github.com/casbin/casbin/v3"
	"github.com/casbin/casbin/v3/util"
)

func TestAdapter(t *testing.T) {
	// Use env var or default
	connString := os.Getenv("PG_CONN_STR")
	if connString == "" {
		connString = "postgres://postgres:password@localhost:5432/test"
	}

	// Create adapter
	a, err := NewAdapter(connString)
	if err != nil {
		t.Skipf("Skipping test, failed to connect to database: %v", err)
	}
	defer a.Close()

	// Initialize basic enforcer
	e, err := casbin.NewEnforcer("rbac_model.conf", a)
	if err != nil {
		t.Fatalf("Failed to create enforcer: %v", err)
	}

	// Clear existing policies
	if err := a.SavePolicy(e.GetModel()); err != nil {
		t.Fatalf("Failed to clear policy: %v", err)
	}

	// Test SavePolicy
	e.AddPolicy("alice", "data1", "read")
	e.AddPolicy("bob", "data2", "write")
	e.AddGroupingPolicy("alice", "data2_admin")

	if err := e.SavePolicy(); err != nil {
		t.Fatalf("Failed to save policy: %v", err)
	}

	e.ClearPolicy()
	if err := e.LoadPolicy(); err != nil {
		t.Fatalf("Failed to load policy: %v", err)
	}

	testGetPolicy(t, e, [][]string{
		{"alice", "data1", "read"},
		{"bob", "data2", "write"},
		{"alice", "data2_admin"},
	})

	// Test AddPolicy
	if _, err := e.AddPolicy("charlie", "data3", "read"); err != nil {
		t.Fatalf("Failed to add policy: %v", err)
	}

	e.ClearPolicy()
	if err := e.LoadPolicy(); err != nil {
		t.Fatalf("Failed to load policy: %v", err)
	}

	testGetPolicy(t, e, [][]string{
		{"alice", "data1", "read"},
		{"bob", "data2", "write"},
		{"charlie", "data3", "read"},
		{"alice", "data2_admin"},
	})

	// Test RemovePolicy
	if _, err := e.RemovePolicy("charlie", "data3", "read"); err != nil {
		t.Fatalf("Failed to remove policy: %v", err)
	}

	e.ClearPolicy()
	if err := e.LoadPolicy(); err != nil {
		t.Fatalf("Failed to load policy: %v", err)
	}

	testGetPolicy(t, e, [][]string{
		{"alice", "data1", "read"},
		{"bob", "data2", "write"},
		{"alice", "data2_admin"},
	})

	// Test RemoveFilteredPolicy
	if _, err := e.RemoveFilteredPolicy(0, "alice"); err != nil {
		t.Fatalf("Failed to remove filtered policy: %v", err)
	}

	e.ClearPolicy()
	if err := e.LoadPolicy(); err != nil {
		t.Fatalf("Failed to load policy: %v", err)
	}

	testGetPolicy(t, e, [][]string{
		{"bob", "data2", "write"},
		{"alice", "data2_admin"},
	})
}

func testGetPolicy(t *testing.T, e *casbin.Enforcer, res [][]string) {
	t.Helper()
	myRes, err := e.GetPolicy()
	if err != nil {
		t.Errorf("Failed to get policy: %v", err)
	}
	// append grouping policies
	groupingPolicies, err := e.GetGroupingPolicy()
	if err != nil {
		t.Errorf("Failed to get grouping policy: %v", err)
	}
	myRes = append(myRes, groupingPolicies...)

	if !util.Array2DEquals(res, myRes) {
		// try sorting
		// simple sort based on string representation
		sortPolicies(res)
		sortPolicies(myRes)
		if !util.Array2DEquals(res, myRes) {
			t.Errorf("Policy check failed: expected %v, got %v", res, myRes)
		}
	}
}

func sortPolicies(pol [][]string) {
	sort.Slice(pol, func(i, j int) bool {
		return strings.Join(pol[i], ",") < strings.Join(pol[j], ",")
	})
}