go-libs/pkg/authn/jwt/control_plane_agent_test.go

package jwt_test

import (
	"testing"

	"github.com/stretchr/testify/assert"

	auth "github.com/hanzo-fi/go-libs/v5/pkg/authn/jwt"
	"github.com/hanzo-fi/go-libs/v5/pkg/authn/oidc"
)

func TestDefaultControlPlaneAgent_GetScopes(t *testing.T) {
	t.Parallel()
	claims := oidc.AccessTokenClaims{
		Scopes: []string{"scope1", "scope2"},
	}
	agent := auth.NewDefaultControlPlaneAgent(claims)

	assert.Equal(t, []string{"scope1", "scope2"}, agent.GetScopes())
}

func TestDefaultControlPlaneAgent_GetOrganizationID(t *testing.T) {
	t.Parallel()
	claims := oidc.AccessTokenClaims{
		Claims: map[string]interface{}{"organization_id": "org123"},
	}
	agent := auth.NewDefaultControlPlaneAgent(claims)

	assert.Equal(t, "org123", agent.GetOrganizationID())
}

func TestDefaultControlPlaneAgent_HasScope(t *testing.T) {
	t.Parallel()
	claims := oidc.AccessTokenClaims{
		Scopes: []string{"scope1", "scope2"},
	}
	agent := auth.NewDefaultControlPlaneAgent(claims)

	assert.True(t, agent.HasScope("scope1"))
	assert.False(t, agent.HasScope("scope3"))
}

func TestDefaultControlPlaneAgent_Subject(t *testing.T) {
	t.Parallel()
	claims := oidc.AccessTokenClaims{
		TokenClaims: oidc.TokenClaims{
			Subject: "subject123",
		},
	}
	agent := auth.NewDefaultControlPlaneAgent(claims)

	assert.Equal(t, "subject123", agent.Subject())
}

func TestDefaultControlPlaneAgent_GetClientID(t *testing.T) {
	t.Parallel()
	claims := oidc.AccessTokenClaims{
		TokenClaims: oidc.TokenClaims{
			ClientID: "client123",
		},
	}
	agent := auth.NewDefaultControlPlaneAgent(claims)

	assert.Equal(t, "client123", agent.GetClientID())
}