lint

johnstcn · johnstcn · commit 2611f519c720 · 2023-04-27T15:35:11.000+01:00
diff --git a/cli/trafficgen.go b/cli/trafficgen.go
@@ -8,11 +8,12 @@ import (
 	"sync/atomic"
 	"time"
 
+	"github.com/google/uuid"
+	"golang.org/x/xerrors"
+
 	"github.com/coder/coder/cli/clibase"
 	"github.com/coder/coder/codersdk"
 	"github.com/coder/coder/cryptorand"
-	"github.com/google/uuid"
-	"golang.org/x/xerrors"
 )
 
 func (r *RootCmd) trafficGen() *clibase.Cmd {
@@ -76,9 +77,12 @@ func (r *RootCmd) trafficGen() *clibase.Cmd {
 				Data: "#",
 			})
 			if err != nil {
-				return xerrors.Errorf("write comment to pty: %w", err)
+				return xerrors.Errorf("serialize request: %w", err)
 			}
 			_, err = crw.Write(data)
+			if err != nil {
+				return xerrors.Errorf("write comment to pty: %w", err)
+			}
 			// Now we begin writing random data to the pty.
 			writeSize := int(bps / 10)
 			rch := make(chan error)
@@ -176,7 +180,7 @@ func copyContext(ctx context.Context, dst io.Writer, src []byte) error {
 		default:
 			_, err := dst.Write(src[idx : idx+1])
 			if err != nil {
-				if err == io.EOF {
+				if xerrors.Is(err, io.EOF) {
 					return nil
 				}
 				return err
diff --git a/cli/trafficgen_test.go b/cli/trafficgen_test.go
@@ -6,15 +6,16 @@ import (
 	"strings"
 	"testing"
 
+	"github.com/google/uuid"
+	"github.com/stretchr/testify/require"
+
 	"github.com/coder/coder/agent"
 	"github.com/coder/coder/cli/clitest"
 	"github.com/coder/coder/coderd/coderdtest"
 	"github.com/coder/coder/codersdk/agentsdk"
 	"github.com/coder/coder/provisioner/echo"
 	"github.com/coder/coder/provisionersdk/proto"
 	"github.com/coder/coder/testutil"
-	"github.com/google/uuid"
-	"github.com/stretchr/testify/require"
 )
 
 // This test pretends to stand up a workspace and run a no-op traffic generation test.
