fix: refresh entitlements after creating first user (#12285)

This commit is contained in:
Marcin Tojek
2024-02-23 17:48:24 +01:00
committed by GitHub
parent 2cb9bfd517
commit 90db6683c4
7 changed files with 64 additions and 15 deletions

View File

@ -75,10 +75,16 @@ func TestFirstUser(t *testing.T) {
t.Run("Trial", func(t *testing.T) {
t.Parallel()
called := make(chan struct{})
trialGenerated := make(chan struct{})
entitlementsRefreshed := make(chan struct{})
client := coderdtest.New(t, &coderdtest.Options{
TrialGenerator: func(context.Context, codersdk.LicensorTrialRequest) error {
close(called)
close(trialGenerated)
return nil
},
RefreshEntitlements: func(context.Context) error {
close(entitlementsRefreshed)
return nil
},
})
@ -94,7 +100,9 @@ func TestFirstUser(t *testing.T) {
}
_, err := client.CreateFirstUser(ctx, req)
require.NoError(t, err)
<-called
_ = testutil.RequireRecvCtx(ctx, t, trialGenerated)
_ = testutil.RequireRecvCtx(ctx, t, entitlementsRefreshed)
})
}