upstream-corteza/server/pkg/rbac/resource_test.go

package rbac

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestResourceType(t *testing.T) {
	var (
		tcc = []struct {
			in  string
			exp string
		}{
			{"a:b/c/d", "a:b"},
			{"a:b/c", "a:b"},
			{"a:b/", "a:b"},
			{"a:b", "a:b"},
			{"a/", "a"},
			{"a", "a"},
			{"", ""},
		}
	)

	for _, tc := range tcc {
		t.Run(tc.in, func(t *testing.T) {
			require.Equal(t, tc.exp, ResourceType(tc.in))
		})
	}
}

func TestResourceComponent(t *testing.T) {
	var (
		tcc = []struct {
			in  string
			exp string
		}{
			{"ns::cmp:r/1/2/3", "ns::cmp"},
			{"ns::cmp:r", "ns::cmp"},
			{"ns::cmp/", "ns::cmp"},
			{"ns::cmp", "ns::cmp"},
			{"cmp", "cmp"},
		}
	)

	for _, tc := range tcc {
		t.Run(tc.in, func(t *testing.T) {
			require.Equal(t, tc.exp, ResourceComponent(tc.in))
		})
	}
}

func TestResourceMatch(t *testing.T) {
	var (
		tcc = []struct {
			m string
			r string
			e bool
		}{
			{"::corteza:test/a/b/c", "::corteza:test/a/b/c", true},
			{"::corteza:test/a/b/*", "::corteza:test/a/b/c", true},
			{"::corteza:test/a/*/*", "::corteza:test/a/b/c", true},
			{"::corteza:test/*/*/*", "::corteza:test/a/b/c", true},
			{"::corteza:test/a/*/*", "::corteza:test/1/2/3", false},
		}
	)

	for _, tc := range tcc {
		t.Run(tc.m, func(t *testing.T) {
			require.Equal(t, tc.e, matchResource(tc.m, tc.r))
		})
	}
}

func TestLevel(t *testing.T) {
	var (
		tcc = []struct {
			r string
			l int
		}{
			{"corteza::test/a/b/c", 111},
			{"corteza::test/a/b/*", 11},
			{"corteza::test/a/*/*", 1},
			{"corteza::test/*/*/*", 0},
			{"corteza::test/a/*/123", 101},
		}
	)

	for _, tc := range tcc {
		t.Run(tc.r, func(t *testing.T) {
			require.Equal(t, tc.l, level(tc.r))
		})
	}
}

func TestIsSpecific(t *testing.T) {
	var (
		tcc = []struct {
			r string
			e bool
		}{
			{"corteza::test/a/b/c", true},
			{"corteza::test/a/b/*", true},
			{"corteza::test/a/*/*", true},
			{"corteza::test/a/*/123", true},
			{"corteza::test/*/*/*", false},
			{"corteza::test/*/*", false},
			{"corteza::test/*", false},
			{"corteza::test/", false},
		}
	)

	for _, tc := range tcc {
		t.Run(tc.r, func(t *testing.T) {
			require.Equal(t, tc.e, isSpecific(tc.r))
		})
	}
}

func TestParseResourceID(t *testing.T) {
	var (
		tcc = []struct {
			in      string
			outType string
			outIDs  []uint64
		}{
			{"corteza::test/*/*/*", "corteza::test", []uint64{0, 0, 0}},
			{"corteza::test/234/*/123", "corteza::test", []uint64{234, 0, 123}},
			{"corteza::test/3/2/1", "corteza::test", []uint64{3, 2, 1}},
		}
	)

	for _, tc := range tcc {
		t.Run(tc.in, func(t *testing.T) {
			outType, outIDs := ParseResourceID(tc.in)

			require.Equal(t, tc.outType, outType)
			require.Equal(t, tc.outIDs, outIDs)
		})
	}
}