fix(coderd): fix flake in TestAPI/ModifyAutostopWithRunningWorkspace

DanielleMaywood · DanielleMaywood · commit a3f64f74f794 · 2025-07-21T10:49:26.000Z
diff --git a/coderd/workspaces_test.go b/coderd/workspaces_test.go
@@ -2875,13 +2875,18 @@ func TestWorkspaceUpdateTTL(t *testing.T) {
 				ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
 				defer cancel()
 
-				err := client.UpdateWorkspaceTTL(ctx, workspace.ID, codersdk.UpdateWorkspaceTTLRequest{
-					TTLMillis: testCase.toTTL,
-				})
+				// Re-fetch the workspace build. This is required because
+				// `AwaitWorkspaceBuildJobCompleted` can return stale data.
+				build, err := client.WorkspaceBuild(ctx, build.ID)
 				require.NoError(t, err)
 
 				deadlineBefore := build.Deadline
 
+				err = client.UpdateWorkspaceTTL(ctx, workspace.ID, codersdk.UpdateWorkspaceTTLRequest{
+					TTLMillis: testCase.toTTL,
+				})
+				require.NoError(t, err)
+
 				build, err = client.WorkspaceBuild(ctx, build.ID)
 				require.NoError(t, err)
 
