test(coderd): fix TestWorkspaceWatcher skipped pending state (#8584)

mafredri · web-flow · commit 611fbd8215ca · 2023-07-18T21:58:50.000+03:00
diff --git a/coderd/workspaces_test.go b/coderd/workspaces_test.go
@@ -2158,8 +2158,10 @@ func TestWorkspaceWatcher(t *testing.T) {
 	_ = coderdtest.CreateWorkspaceBuild(t, client, workspace, database.WorkspaceTransitionStart, func(req *codersdk.CreateWorkspaceBuildRequest) {
 		req.TemplateVersionID = updatedVersion.ID
 	})
-	wait("workspace build pending", func(w codersdk.Workspace) bool {
-		return w.LatestBuild.Status == codersdk.WorkspaceStatusPending
+	// We want to verify pending state here, but it's possible that we reach
+	// failed state fast enough that we never see pending.
+	wait("workspace build pending or failed", func(w codersdk.Workspace) bool {
+		return w.LatestBuild.Status == codersdk.WorkspaceStatusPending || w.LatestBuild.Status == codersdk.WorkspaceStatusFailed
 	})
 	wait("workspace build failed", func(w codersdk.Workspace) bool {
 		return w.LatestBuild.Status == codersdk.WorkspaceStatusFailed
