package user

import (
	"testing"

	commonLogic "github.com/perfect-panel/server/internal/logic/common"
	"github.com/perfect-panel/server/internal/types"
	"github.com/stretchr/testify/require"
)

func TestFillUserSubscribeEntitlementFields(t *testing.T) {
	sub := &types.UserSubscribe{}
	entitlement := &commonLogic.EntitlementContext{
		EffectiveUserID: 2001,
		Source:          commonLogic.EntitlementSourceFamilyOwner,
		OwnerUserID:     2001,
		ReadOnly:        true,
	}

	fillUserSubscribeEntitlementFields(sub, entitlement)

	require.Equal(t, commonLogic.EntitlementSourceFamilyOwner, sub.EntitlementSource)
	require.Equal(t, int64(2001), sub.EntitlementOwnerUserId)
	require.True(t, sub.ReadOnly)
}