mirror of
https://github.com/coder/coder.git
synced 2025-07-06 15:41:45 +00:00
af125c3795
* chore: refactor entitlements to be passable as an argument Previously, all usage of entitlements requires mutex usage on the api struct directly. This prevents passing the entitlements to a sub package. It also creates the possibility for misuse.
64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
package entitlements_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/v2/coderd/entitlements"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
)
|
|
|
|
func TestUpdate(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
set := entitlements.New()
|
|
require.False(t, set.Enabled(codersdk.FeatureMultipleOrganizations))
|
|
|
|
set.Update(func(entitlements *codersdk.Entitlements) {
|
|
entitlements.Features[codersdk.FeatureMultipleOrganizations] = codersdk.Feature{
|
|
Enabled: true,
|
|
Entitlement: codersdk.EntitlementEntitled,
|
|
}
|
|
})
|
|
require.True(t, set.Enabled(codersdk.FeatureMultipleOrganizations))
|
|
}
|
|
|
|
func TestAllowRefresh(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
now := time.Now()
|
|
set := entitlements.New()
|
|
set.Update(func(entitlements *codersdk.Entitlements) {
|
|
entitlements.RefreshedAt = now
|
|
})
|
|
|
|
ok, wait := set.AllowRefresh(now)
|
|
require.False(t, ok)
|
|
require.InDelta(t, time.Minute.Seconds(), wait.Seconds(), 5)
|
|
|
|
set.Update(func(entitlements *codersdk.Entitlements) {
|
|
entitlements.RefreshedAt = now.Add(time.Minute * -2)
|
|
})
|
|
|
|
ok, wait = set.AllowRefresh(now)
|
|
require.True(t, ok)
|
|
require.Equal(t, time.Duration(0), wait)
|
|
}
|
|
|
|
func TestReplace(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
set := entitlements.New()
|
|
require.False(t, set.Enabled(codersdk.FeatureMultipleOrganizations))
|
|
set.Replace(codersdk.Entitlements{
|
|
Features: map[codersdk.FeatureName]codersdk.Feature{
|
|
codersdk.FeatureMultipleOrganizations: {
|
|
Enabled: true,
|
|
},
|
|
},
|
|
})
|
|
require.True(t, set.Enabled(codersdk.FeatureMultipleOrganizations))
|
|
}
|