feat(cli): make minor improvements to speedtest (#6266)

- Remove mostly redundant "Transferred" column
- Rename "Bandwidth" to "Throughput"
- Replace "--reverse" (which has an ambiguous starting state) with "--direction=(up|down)"
- Tolerate AgentStartErrors which may be caused by failing startup
  script

cli/speedtest.go

@@ -21,7 +21,7 @@ func speedtest() *cobra.Command {
 	var (
 		direct    bool
 		duration  time.Duration
-		reverse  bool
+		direction string
 	)
 	cmd := &cobra.Command{
 		Annotations: workspaceCommand,
@@ -48,7 +48,7 @@ func speedtest() *cobra.Command {
 					return client.WorkspaceAgent(ctx, workspaceAgent.ID)
 				},
 			})
-			if err != nil {
+			if err != nil && !xerrors.Is(err, cliui.AgentStartError) {
 				return xerrors.Errorf("await agent: %w", err)
 			}
 			logger := slog.Make(sloghuman.Sink(cmd.ErrOrStderr()))
@@ -94,17 +94,22 @@ func speedtest() *cobra.Command {
 			} else {
 				conn.AwaitReachable(ctx)
 			}
-			dir := tsspeedtest.Download
+			var tsDir tsspeedtest.Direction
-			if reverse {
+			switch direction {
-				dir = tsspeedtest.Upload
+			case "up":
+				tsDir = tsspeedtest.Upload
+			case "down":
+				tsDir = tsspeedtest.Download
+			default:
+				return xerrors.Errorf("invalid direction: %q", direction)
 			}
-			cmd.Printf("Starting a %ds %s test...\n", int(duration.Seconds()), dir)
+			cmd.Printf("Starting a %ds %s test...\n", int(duration.Seconds()), tsDir)
-			results, err := conn.Speedtest(ctx, dir, duration)
+			results, err := conn.Speedtest(ctx, tsDir, duration)
 			if err != nil {
 				return err
 			}
 			tableWriter := cliui.Table()
-			tableWriter.AppendHeader(table.Row{"Interval", "Transfer", "Bandwidth"})
+			tableWriter.AppendHeader(table.Row{"Interval", "Throughput"})
 			startTime := results[0].IntervalStart
 			for _, r := range results {
 				if r.Total {
@@ -112,7 +117,6 @@ func speedtest() *cobra.Command {
 				}
 				tableWriter.AppendRow(table.Row{
 					fmt.Sprintf("%.2f-%.2f sec", r.IntervalStart.Sub(startTime).Seconds(), r.IntervalEnd.Sub(startTime).Seconds()),
-					fmt.Sprintf("%.4f MBits", r.MegaBits()),
 					fmt.Sprintf("%.4f Mbits/sec", r.MBitsPerSecond()),
 				})
 			}
@@ -122,8 +126,9 @@ func speedtest() *cobra.Command {
 	}
 	cliflag.BoolVarP(cmd.Flags(), &direct, "direct", "d", "", false,
 		"Specifies whether to wait for a direct connection before testing speed.")
-	cliflag.BoolVarP(cmd.Flags(), &reverse, "reverse", "r", "", false,
+	cliflag.StringVarP(cmd.Flags(), &direction, "direction", "", "", "down",
-		"Specifies whether to run in reverse mode where the client receives and the server sends.")
+		"Specifies whether to run in reverse mode where the client receives and the server sends. (up|down)",
+	)
 	cmd.Flags().DurationVarP(&duration, "time", "t", tsspeedtest.DefaultDuration,
 		"Specifies the duration to monitor traffic.")
 	return cmd

cli/testdata/coder_speedtest_--help.golden

@@ -5,9 +5,9 @@ Usage:
 
 Flags:
   -d, --direct             Specifies whether to wait for a direct connection before testing speed.
+      --direction string   Specifies whether to run in reverse mode where the client receives
+                           and the server sends. (up|down) (default "down")
   -h, --help               help for speedtest
-  -r, --reverse         Specifies whether to run in reverse mode where the client receives and
-                        the server sends.
   -t, --time duration      Specifies the duration to monitor traffic. (default 5s)
 
 Global Flags:

docs/cli/coder_speedtest.md

@@ -10,8 +10,8 @@ coder speedtest <workspace> [flags]
 
 ```
   -d, --direct             Specifies whether to wait for a direct connection before testing speed.
+      --direction string   Specifies whether to run in reverse mode where the client receives and the server sends. (up|down) (default "down")
   -h, --help               help for speedtest
-  -r, --reverse         Specifies whether to run in reverse mode where the client receives and the server sends.
   -t, --time duration      Specifies the duration to monitor traffic. (default 5s)
 ```