@@ -1942,7 +1942,7 @@ func (r *RootCmd) scaletestNotifications() *serpent.Command {
19421942Short :"Simulate notification delivery by creating many users listening to notifications." ,
19431943Handler :func (inv * serpent.Invocation )error {
19441944ctx := inv .Context ()
1945- client ,err := r .TryInitClient (inv )
1945+ client ,err := r .InitClient (inv )
19461946if err != nil {
19471947return err
19481948}
@@ -2034,7 +2034,7 @@ func (r *RootCmd) scaletestNotifications() *serpent.Command {
20342034NotificationTimeout :notificationTimeout ,
20352035DialTimeout :dialTimeout ,
20362036DialBarrier :dialBarrier ,
2037- OwnerWatchBarrier : ownerWatchBarrier ,
2037+ ReceivingWatchBarrier : ownerWatchBarrier ,
20382038ExpectedNotifications :expectedNotifications ,
20392039Metrics :metrics ,
20402040}
@@ -2048,12 +2048,12 @@ func (r *RootCmd) scaletestNotifications() *serpent.Command {
20482048User : createusers.Config {
20492049OrganizationID :me .OrganizationIDs [0 ],
20502050},
2051- Roles : []string {},
2052- NotificationTimeout :notificationTimeout ,
2053- DialTimeout :dialTimeout ,
2054- DialBarrier :dialBarrier ,
2055- OwnerWatchBarrier : ownerWatchBarrier ,
2056- Metrics :metrics ,
2051+ Roles :string {},
2052+ NotificationTimeout :notificationTimeout ,
2053+ DialTimeout :dialTimeout ,
2054+ DialBarrier :dialBarrier ,
2055+ ReceivingWatchBarrier : ownerWatchBarrier ,
2056+ Metrics :metrics ,
20572057}
20582058if err := config .Validate ();err != nil {
20592059return xerrors .Errorf ("validate config: %w" ,err )
@@ -2418,7 +2418,6 @@ func triggerUserNotifications(
24182418slog .F ("email" ,triggerEmail ),
24192419slog .F ("org_id" ,orgID ))
24202420
2421- createTime := time .Now ()
24222421testUser ,err := client .CreateUserWithOrgs (ctx , codersdk.CreateUserRequestWithOrgs {
24232422OrganizationIDs : []uuid.UUID {orgID },
24242423Username :triggerUsername ,
@@ -2429,15 +2428,14 @@ func triggerUserNotifications(
24292428logger .Error (ctx ,"create test user" ,slog .Error (err ))
24302429return
24312430}
2432- expectedNotifications [notificationsLib .TemplateUserAccountCreated ]<- createTime
2431+ expectedNotifications [notificationsLib .TemplateUserAccountCreated ]<- time . Now ()
24332432
2434- deleteTime := time .Now ()
24352433err = client .DeleteUser (ctx ,testUser .ID )
24362434if err != nil {
24372435logger .Error (ctx ,"delete test user" ,slog .Error (err ))
24382436return
24392437}
2440- expectedNotifications [notificationsLib .TemplateUserAccountDeleted ]<- deleteTime
2438+ expectedNotifications [notificationsLib .TemplateUserAccountDeleted ]<- time . Now ()
24412439close (expectedNotifications [notificationsLib .TemplateUserAccountCreated ])
24422440close (expectedNotifications [notificationsLib .TemplateUserAccountDeleted ])
24432441}