@@ -29,6 +29,7 @@ import (
2929
3030"github.com/coder/coder/v2/cli/cliui"
3131"github.com/coder/coder/v2/coderd/httpapi"
32+ notificationsLib"github.com/coder/coder/v2/coderd/notifications"
3233"github.com/coder/coder/v2/coderd/tracing"
3334"github.com/coder/coder/v2/codersdk"
3435"github.com/coder/coder/v2/codersdk/workspacesdk"
@@ -2018,17 +2019,23 @@ func (r *RootCmd) scaletestNotifications() *serpent.Command {
20182019ownerDialBarrier .Add (int (ownerUserCount ))
20192020regularDialBarrier .Add (int (regularUserCount ))
20202021
2022+ expectedNotifications := map [uuid.UUID ]chan time.Time {
2023+ notificationsLib .TemplateUserAccountCreated :make (chan time.Time ,1 ),
2024+ notificationsLib .TemplateUserAccountDeleted :make (chan time.Time ,1 ),
2025+ }
2026+
20212027configs := make ([]notifications.Config ,0 ,userCount )
20222028for range ownerUserCount {
20232029config := notifications.Config {
20242030User : createusers.Config {
20252031OrganizationID :me .OrganizationIDs [0 ],
20262032},
2027- IsOwner :true ,
2028- NotificationTimeout :notificationTimeout ,
2029- DialTimeout :dialTimeout ,
2030- DialBarrier :ownerDialBarrier ,
2031- Metrics :metrics ,
2033+ IsOwner :true ,
2034+ NotificationTimeout :notificationTimeout ,
2035+ DialTimeout :dialTimeout ,
2036+ DialBarrier :ownerDialBarrier ,
2037+ ExpectedNotifications :expectedNotifications ,
2038+ Metrics :metrics ,
20322039}
20332040if err := config .Validate ();err != nil {
20342041return xerrors .Errorf ("validate config: %w" ,err )
@@ -2113,6 +2120,7 @@ func (r *RootCmd) scaletestNotifications() *serpent.Command {
21132120slog .F ("email" ,triggerEmail ),
21142121slog .F ("org_id" ,me .OrganizationIDs [0 ]))
21152122
2123+ createTime := time .Now ()
21162124testUser ,err := client .CreateUserWithOrgs (ctx , codersdk.CreateUserRequestWithOrgs {
21172125OrganizationIDs : []uuid.UUID {me .OrganizationIDs [0 ]},
21182126Username :triggerUsername ,
@@ -2123,14 +2131,17 @@ func (r *RootCmd) scaletestNotifications() *serpent.Command {
21232131logger .Error (ctx ,"create test user" ,slog .Error (err ))
21242132return
21252133}
2134+ expectedNotifications [notificationsLib .TemplateUserAccountCreated ]<- createTime
21262135
2136+ deleteTime := time .Now ()
21272137err = client .DeleteUser (ctx ,testUser .ID )
21282138if err != nil {
21292139logger .Error (ctx ,"delete test user" ,slog .Error (err ))
21302140return
21312141}
2132-
2133- logger .Info (ctx ,"test user created and deleted successfully" )
2142+ expectedNotifications [notificationsLib .TemplateUserAccountDeleted ]<- deleteTime
2143+ close (expectedNotifications [notificationsLib .TemplateUserAccountCreated ])
2144+ close (expectedNotifications [notificationsLib .TemplateUserAccountDeleted ])
21342145}()
21352146
21362147th := harness .NewTestHarness (timeoutStrategy .wrapStrategy (harness.ConcurrentExecutionStrategy {}),cleanupStrategy .toStrategy ())