@@ -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"
@@ -1783,17 +1784,23 @@ func (r *RootCmd) scaletestNotifications() *serpent.Command {
17831784ownerDialBarrier .Add (int (ownerUserCount ))
17841785regularDialBarrier .Add (int (regularUserCount ))
17851786
1787+ expectedNotifications := map [uuid.UUID ]chan time.Time {
1788+ notificationsLib .TemplateUserAccountCreated :make (chan time.Time ,1 ),
1789+ notificationsLib .TemplateUserAccountDeleted :make (chan time.Time ,1 ),
1790+ }
1791+
17861792configs := make ([]notifications.Config ,0 ,userCount )
17871793for range ownerUserCount {
17881794config := notifications.Config {
17891795User : createusers.Config {
17901796OrganizationID :me .OrganizationIDs [0 ],
17911797},
1792- IsOwner :true ,
1793- NotificationTimeout :notificationTimeout ,
1794- DialTimeout :dialTimeout ,
1795- DialBarrier :ownerDialBarrier ,
1796- Metrics :metrics ,
1798+ IsOwner :true ,
1799+ NotificationTimeout :notificationTimeout ,
1800+ DialTimeout :dialTimeout ,
1801+ DialBarrier :ownerDialBarrier ,
1802+ ExpectedNotifications :expectedNotifications ,
1803+ Metrics :metrics ,
17971804}
17981805if err := config .Validate ();err != nil {
17991806return xerrors .Errorf ("validate config: %w" ,err )
@@ -1878,6 +1885,7 @@ func (r *RootCmd) scaletestNotifications() *serpent.Command {
18781885slog .F ("email" ,triggerEmail ),
18791886slog .F ("org_id" ,me .OrganizationIDs [0 ]))
18801887
1888+ createTime := time .Now ()
18811889testUser ,err := client .CreateUserWithOrgs (ctx , codersdk.CreateUserRequestWithOrgs {
18821890OrganizationIDs : []uuid.UUID {me .OrganizationIDs [0 ]},
18831891Username :triggerUsername ,
@@ -1888,14 +1896,17 @@ func (r *RootCmd) scaletestNotifications() *serpent.Command {
18881896logger .Error (ctx ,"create test user" ,slog .Error (err ))
18891897return
18901898}
1899+ expectedNotifications [notificationsLib .TemplateUserAccountCreated ]<- createTime
18911900
1901+ deleteTime := time .Now ()
18921902err = client .DeleteUser (ctx ,testUser .ID )
18931903if err != nil {
18941904logger .Error (ctx ,"delete test user" ,slog .Error (err ))
18951905return
18961906}
1897-
1898- logger .Info (ctx ,"test user created and deleted successfully" )
1907+ expectedNotifications [notificationsLib .TemplateUserAccountDeleted ]<- deleteTime
1908+ close (expectedNotifications [notificationsLib .TemplateUserAccountCreated ])
1909+ close (expectedNotifications [notificationsLib .TemplateUserAccountDeleted ])
18991910}()
19001911
19011912th := harness .NewTestHarness (timeoutStrategy .wrapStrategy (harness.ConcurrentExecutionStrategy {}),cleanupStrategy .toStrategy ())