forked from mirrors/ntfy
63 lines
1.7 KiB
Go
63 lines
1.7 KiB
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"])
|
|
|
|
}
|