libentitlement_test.go

package libentitlement

import (
	"fmt"
	"testing"

	"github.com/docker/libentitlement/defaults"
	"github.com/docker/libentitlement/entitlement"
	secprofile "github.com/docker/libentitlement/secprofile"
	"github.com/docker/libentitlement/testutils"
	"github.com/stretchr/testify/require"
)

func TestRegisterDummyEntitlement(t *testing.T) {
	spec := testutils.TestSpec()
	profile := secprofile.NewOCIProfile(spec, "dummy-profile-default")

	entMgr := NewEntitlementsManager(profile)

	// Add a dummy "foo.bar.cap-sys-admin" void entitlement that adds CAP_SYS_ADMIN
	capSysAdminVoidEntCallback := func(profile secprofile.Profile) (secprofile.Profile, error) {
		ociProfile, ok := profile.(*secprofile.OCIProfile)
		if !ok {
			return nil, fmt.Errorf("%s: error converting to OCI profile", defaults.NetworkNoneEntFullID)
		}

		ociProfile.AddCaps("CAP_SYS_ADMIN")

		return profile, nil
	}

	capSysAdminVoidEntFullName := "foo-bar.meh.cap-sys-admin"

	capSysAdminVoidEnt := entitlement.NewVoidEntitlement(capSysAdminVoidEntFullName, capSysAdminVoidEntCallback)

	err := entMgr.Add(capSysAdminVoidEnt)
	require.NoError(t, err, "Entitlement %s should have been added and enforced", capSysAdminVoidEntFullName)

	require.Contains(t, profile.OCI.Process.Capabilities.Bounding, "CAP_SYS_ADMIN", "Capability is missing after entitlement enforcement")
	require.Contains(t, profile.OCI.Process.Capabilities.Effective, "CAP_SYS_ADMIN", "Capability is missing after entitlement enforcement")
	require.Contains(t, profile.OCI.Process.Capabilities.Permitted, "CAP_SYS_ADMIN", "Capability is missing after entitlement enforcement")
	require.Contains(t, profile.OCI.Process.Capabilities.Inheritable, "CAP_SYS_ADMIN", "Capability is missing after entitlement enforcement")
}