Add unit test

Emyrk · Emyrk · commit bff451b8d631 · 2023-05-02T10:50:19.000-05:00
diff --git a/enterprise/cli/workspaceproxy.go b/enterprise/cli/workspaceproxy.go
@@ -2,7 +2,9 @@ package cli
 
 import (
 	"fmt"
+	"strings"
 
+	"github.com/fatih/color"
 	"golang.org/x/xerrors"
 
 	"github.com/coder/coder/cli/clibase"
@@ -152,7 +154,27 @@ func (r *RootCmd) listProxies() *clibase.Cmd {
 		formatter = cliui.NewOutputFormatter(
 			cliui.TableFormat([]codersdk.WorkspaceProxy{}, []string{"name", "url", "status status"}),
 			cliui.JSONFormat(),
-			cliui.TextFormat(),
+			cliui.ChangeFormatterData(cliui.TextFormat(), func(data any) (any, error) {
+				resp, ok := data.([]codersdk.WorkspaceProxy)
+				if !ok {
+					return nil, xerrors.Errorf("unexpected type %T", data)
+				}
+				var str strings.Builder
+				str.WriteString("Workspace Proxies:\n")
+				sep := ""
+				for i, proxy := range resp {
+					str.WriteString(sep)
+					str.WriteString(fmt.Sprintf("%d: %s %s %s", i, proxy.Name, proxy.URL, proxy.Status.Status))
+					for _, errMsg := range proxy.Status.Report.Errors {
+						str.WriteString(color.RedString("\n\tErr: %s", errMsg))
+					}
+					for _, warnMsg := range proxy.Status.Report.Errors {
+						str.WriteString(color.YellowString("\n\tWarn: %s", warnMsg))
+					}
+					sep = "\n"
+				}
+				return str.String(), nil
+			}),
 		)
 	)
 
diff --git a/enterprise/cli/workspaceproxy_test.go b/enterprise/cli/workspaceproxy_test.go
@@ -65,6 +65,21 @@ func Test_ProxyCRUD(t *testing.T) {
 		_, err = uuid.Parse(parts[0])
 		require.NoError(t, err, "expected token to be a uuid")
 
+		// Fetch proxies and check output
+		inv, conf = newCLI(
+			t,
+			"proxy", "ls",
+		)
+
+		pty = ptytest.New(t)
+		inv.Stdout = pty.Output()
+		clitest.SetupConfig(t, client, conf)
+
+		err = inv.WithContext(ctx).Run()
+		require.NoError(t, err)
+		pty.ExpectMatch(expectedName)
+
+		// Also check via the api
 		proxies, err := client.WorkspaceProxies(ctx)
 		require.NoError(t, err, "failed to get workspace proxies")
 		require.Len(t, proxies, 1, "expected 1 proxy")
