ntfy/user/types_test.go

package user

import (
	"github.com/stretchr/testify/require"
	"testing"
)

func TestPermission(t *testing.T) {
	require.Equal(t, PermissionReadWrite, NewPermission(true, true))
	require.Equal(t, PermissionRead, NewPermission(true, false))
	require.Equal(t, PermissionWrite, NewPermission(false, true))
	require.Equal(t, PermissionDenyAll, NewPermission(false, false))
	require.True(t, PermissionReadWrite.IsReadWrite())
	require.True(t, PermissionReadWrite.IsRead())
	require.True(t, PermissionReadWrite.IsWrite())
	require.True(t, PermissionRead.IsRead())
	require.True(t, PermissionWrite.IsWrite())
}

func TestParsePermission(t *testing.T) {
	_, err := ParsePermission("no")
	require.NotNil(t, err)

	p, err := ParsePermission("read-write")
	require.Nil(t, err)
	require.Equal(t, PermissionReadWrite, p)

	p, err = ParsePermission("rw")
	require.Nil(t, err)
	require.Equal(t, PermissionReadWrite, p)

	p, err = ParsePermission("read-only")
	require.Nil(t, err)
	require.Equal(t, PermissionRead, p)

	p, err = ParsePermission("WRITE")
	require.Nil(t, err)
	require.Equal(t, PermissionWrite, p)

	p, err = ParsePermission("deny-all")
	require.Nil(t, err)
	require.Equal(t, PermissionDenyAll, p)
}

func TestAllowedTier(t *testing.T) {
	require.False(t, AllowedTier("  no"))
	require.True(t, AllowedTier("yes"))
}

func TestTierContext(t *testing.T) {
	tier := &Tier{
		ID:                   "ti_abc",
		Code:                 "pro",
		StripeMonthlyPriceID: "price_123",
		StripeYearlyPriceID:  "price_456",
	}
	context := tier.Context()
	require.Equal(t, "ti_abc", context["tier_id"])
	require.Equal(t, "pro", context["tier_code"])
	require.Equal(t, "price_123", context["stripe_monthly_price_id"])
	require.Equal(t, "price_456", context["stripe_yearly_price_id"])

}

func TestUsernameRegex(t *testing.T) {
	username := "phil"
	usernameEmail := "phil@ntfy.sh"
	usernameEmailAlias := "phil+alias@ntfy.sh"
	usernameInvalid := "phil\rocks"

	require.True(t, AllowedUsername(username))
	require.True(t, AllowedUsername(usernameEmail))
	require.True(t, AllowedUsername(usernameEmailAlias))
	require.False(t, AllowedUsername(usernameInvalid))
}