Fix resources not reading in TTY

kylecarbs · kylecarbs · commit 00a999469746 · 2022-04-11T18:07:17.000-05:00
diff --git a/cli/cliui/resources_test.go b/cli/cliui/resources_test.go
@@ -16,71 +16,75 @@ func TestWorkspaceResources(t *testing.T) {
 	t.Run("SingleAgentSSH", func(t *testing.T) {
 		t.Parallel()
 		ptty := ptytest.New(t)
-		err := cliui.WorkspaceResources(ptty.Output(), []codersdk.WorkspaceResource{{
-			Type:       "google_compute_instance",
-			Name:       "dev",
-			Transition: database.WorkspaceTransitionStart,
-			Agents: []codersdk.WorkspaceAgent{{
-				Name:            "dev",
-				Status:          codersdk.WorkspaceAgentConnected,
-				Architecture:    "amd64",
-				OperatingSystem: "linux",
-			}},
-		}}, cliui.WorkspaceResourcesOptions{
-			WorkspaceName: "example",
-		})
-		require.NoError(t, err)
+		go func() {
+			err := cliui.WorkspaceResources(ptty.Output(), []codersdk.WorkspaceResource{{
+				Type:       "google_compute_instance",
+				Name:       "dev",
+				Transition: database.WorkspaceTransitionStart,
+				Agents: []codersdk.WorkspaceAgent{{
+					Name:            "dev",
+					Status:          codersdk.WorkspaceAgentConnected,
+					Architecture:    "amd64",
+					OperatingSystem: "linux",
+				}},
+			}}, cliui.WorkspaceResourcesOptions{
+				WorkspaceName: "example",
+			})
+			require.NoError(t, err)
+		}()
 		ptty.ExpectMatch("coder ssh example")
 	})
 
 	t.Run("MultipleStates", func(t *testing.T) {
 		t.Parallel()
 		ptty := ptytest.New(t)
 		disconnected := database.Now().Add(-4 * time.Second)
-		err := cliui.WorkspaceResources(ptty.Output(), []codersdk.WorkspaceResource{{
-			Address:    "disk",
-			Transition: database.WorkspaceTransitionStart,
-			Type:       "google_compute_disk",
-			Name:       "root",
-		}, {
-			Address:    "disk",
-			Transition: database.WorkspaceTransitionStop,
-			Type:       "google_compute_disk",
-			Name:       "root",
-		}, {
-			Address:    "another",
-			Transition: database.WorkspaceTransitionStart,
-			Type:       "google_compute_instance",
-			Name:       "dev",
-			Agents: []codersdk.WorkspaceAgent{{
-				CreatedAt:       database.Now().Add(-10 * time.Second),
-				Status:          codersdk.WorkspaceAgentConnecting,
-				Name:            "dev",
-				OperatingSystem: "linux",
-				Architecture:    "amd64",
-			}},
-		}, {
-			Transition: database.WorkspaceTransitionStart,
-			Type:       "kubernetes_pod",
-			Name:       "dev",
-			Agents: []codersdk.WorkspaceAgent{{
-				Status:          codersdk.WorkspaceAgentConnected,
-				Name:            "go",
-				Architecture:    "amd64",
-				OperatingSystem: "linux",
+		go func() {
+			err := cliui.WorkspaceResources(ptty.Output(), []codersdk.WorkspaceResource{{
+				Address:    "disk",
+				Transition: database.WorkspaceTransitionStart,
+				Type:       "google_compute_disk",
+				Name:       "root",
 			}, {
-				DisconnectedAt:  &disconnected,
-				Status:          codersdk.WorkspaceAgentDisconnected,
-				Name:            "postgres",
-				Architecture:    "amd64",
-				OperatingSystem: "linux",
-			}},
-		}}, cliui.WorkspaceResourcesOptions{
-			WorkspaceName:  "dev",
-			HideAgentState: false,
-			HideAccess:     false,
-		})
-		require.NoError(t, err)
+				Address:    "disk",
+				Transition: database.WorkspaceTransitionStop,
+				Type:       "google_compute_disk",
+				Name:       "root",
+			}, {
+				Address:    "another",
+				Transition: database.WorkspaceTransitionStart,
+				Type:       "google_compute_instance",
+				Name:       "dev",
+				Agents: []codersdk.WorkspaceAgent{{
+					CreatedAt:       database.Now().Add(-10 * time.Second),
+					Status:          codersdk.WorkspaceAgentConnecting,
+					Name:            "dev",
+					OperatingSystem: "linux",
+					Architecture:    "amd64",
+				}},
+			}, {
+				Transition: database.WorkspaceTransitionStart,
+				Type:       "kubernetes_pod",
+				Name:       "dev",
+				Agents: []codersdk.WorkspaceAgent{{
+					Status:          codersdk.WorkspaceAgentConnected,
+					Name:            "go",
+					Architecture:    "amd64",
+					OperatingSystem: "linux",
+				}, {
+					DisconnectedAt:  &disconnected,
+					Status:          codersdk.WorkspaceAgentDisconnected,
+					Name:            "postgres",
+					Architecture:    "amd64",
+					OperatingSystem: "linux",
+				}},
+			}}, cliui.WorkspaceResourcesOptions{
+				WorkspaceName:  "dev",
+				HideAgentState: false,
+				HideAccess:     false,
+			})
+			require.NoError(t, err)
+		}()
 		ptty.ExpectMatch("google_compute_disk.root")
 		ptty.ExpectMatch("google_compute_instance.dev")
 		ptty.ExpectMatch("coder ssh dev.postgres")
