simple test

mtojek · mtojek · commit 127aa472996c · 2024-08-26T14:39:48.000+02:00
diff --git a/coderd/provisionerdserver/provisionerdserver_test.go b/coderd/provisionerdserver/provisionerdserver_test.go
@@ -1804,6 +1804,88 @@ func TestNotifications(t *testing.T) {
 			})
 		}
 	})
+
+	t.Run("Manual build failed, template admins notified", func(t *testing.T) {
+		t.Parallel()
+
+		ctx := context.Background()
+		notifEnq := &testutil.FakeNotificationsEnqueuer{}
+
+		//	Otherwise `(*Server).FailJob` fails with:
+		// audit log - get build {"error": "sql: no rows in result set"}
+		ignoreLogErrors := true
+		srv, db, ps, pd := setup(t, ignoreLogErrors, &overrides{
+			notificationEnqueuer: notifEnq,
+		})
+
+		templateAdmin := dbgen.User(t, db, database.User{RBACRoles: []string{codersdk.RoleTemplateAdmin}})
+		user := dbgen.User(t, db, database.User{})
+		initiator := user
+
+		template := dbgen.Template(t, db, database.Template{
+			Name:           "template",
+			Provisioner:    database.ProvisionerTypeEcho,
+			OrganizationID: pd.OrganizationID,
+		})
+		template, err := db.GetTemplateByID(ctx, template.ID)
+		require.NoError(t, err)
+		file := dbgen.File(t, db, database.File{CreatedBy: user.ID})
+		workspace := dbgen.Workspace(t, db, database.Workspace{
+			TemplateID:     template.ID,
+			OwnerID:        user.ID,
+			OrganizationID: pd.OrganizationID,
+		})
+		version := dbgen.TemplateVersion(t, db, database.TemplateVersion{
+			OrganizationID: pd.OrganizationID,
+			TemplateID: uuid.NullUUID{
+				UUID:  template.ID,
+				Valid: true,
+			},
+			JobID: uuid.New(),
+		})
+		build := dbgen.WorkspaceBuild(t, db, database.WorkspaceBuild{
+			WorkspaceID:       workspace.ID,
+			TemplateVersionID: version.ID,
+			InitiatorID:       initiator.ID,
+			Transition:        database.WorkspaceTransitionDelete,
+			Reason:            database.BuildReasonInitiator,
+		})
+		job := dbgen.ProvisionerJob(t, db, ps, database.ProvisionerJob{
+			FileID: file.ID,
+			Type:   database.ProvisionerJobTypeWorkspaceBuild,
+			Input: must(json.Marshal(provisionerdserver.WorkspaceProvisionJob{
+				WorkspaceBuildID: build.ID,
+			})),
+			OrganizationID: pd.OrganizationID,
+		})
+		_, err = db.AcquireProvisionerJob(ctx, database.AcquireProvisionerJobParams{
+			OrganizationID: pd.OrganizationID,
+			WorkerID: uuid.NullUUID{
+				UUID:  pd.ID,
+				Valid: true,
+			},
+			Types: []database.ProvisionerType{database.ProvisionerTypeEcho},
+		})
+		require.NoError(t, err)
+
+		_, err = srv.FailJob(ctx, &proto.FailedJob{
+			JobId: job.ID.String(),
+			Type: &proto.FailedJob_WorkspaceBuild_{
+				WorkspaceBuild: &proto.FailedJob_WorkspaceBuild{
+					State: []byte{},
+				},
+			},
+		})
+		require.NoError(t, err)
+
+		require.Len(t, notifEnq.Sent, 1)
+		require.Equal(t, notifEnq.Sent[0].UserID, templateAdmin.ID)
+		require.Equal(t, notifEnq.Sent[0].TemplateID, notifications.TemplateWorkspaceManualBuildFailed)
+		require.Contains(t, notifEnq.Sent[0].Targets, template.ID)
+		require.Contains(t, notifEnq.Sent[0].Targets, workspace.ID)
+		require.Contains(t, notifEnq.Sent[0].Targets, workspace.OrganizationID)
+		require.Contains(t, notifEnq.Sent[0].Targets, user.ID)
+	})
 }
 
 type overrides struct {
