| 28 | } |
| 29 | |
| 30 | func TestAllowRefresh(t *testing.T) { |
| 31 | t.Parallel() |
| 32 | |
| 33 | now := time.Now() |
| 34 | set := entitlements.New() |
| 35 | set.Modify(func(entitlements *codersdk.Entitlements) { |
| 36 | entitlements.RefreshedAt = now |
| 37 | }) |
| 38 | |
| 39 | ok, wait := set.AllowRefresh(now) |
| 40 | require.False(t, ok) |
| 41 | require.InDelta(t, time.Minute.Seconds(), wait.Seconds(), 5) |
| 42 | |
| 43 | set.Modify(func(entitlements *codersdk.Entitlements) { |
| 44 | entitlements.RefreshedAt = now.Add(time.Minute * -2) |
| 45 | }) |
| 46 | |
| 47 | ok, wait = set.AllowRefresh(now) |
| 48 | require.True(t, ok) |
| 49 | require.Equal(t, time.Duration(0), wait) |
| 50 | } |
| 51 | |
| 52 | func TestUpdate(t *testing.T) { |
| 53 | t.Parallel() |