chore(log): add unit test to validate that retries happen

Author: johnstcn

log/coder_internal_test.go

@@ -170,6 +170,55 @@ func TestCoder(t *testing.T) {
 		require.ErrorIs(t, err, context.DeadlineExceeded)
 		<-handlerDone
 	})
+
+	// In this test, we validate that a 401 error on the initial connect
+	// results in a retry. When envbuilder initially attempts to connect
+	// using the Coder agent token, the workspace build may not yet have
+	// completed.
+	t.Run("V2Retry", func(t *testing.T) {
+		t.Parallel()
+		ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+		defer cancel()
+
+		token := uuid.NewString()
+		done := make(chan struct{})
+		handlerSend := make(chan int)
+		handler := func(w http.ResponseWriter, r *http.Request) {
+			t.Logf("test handler: %s", r.URL.Path)
+			if r.URL.Path == "/api/v2/buildinfo" {
+				w.Header().Set("Content-Type", "application/json")
+				_, _ = w.Write([]byte(`{"version": "v2.9.0"}`))
+				return
+			}
+			code := <-handlerSend
+			t.Logf("test handler response: %d", code)
+			w.WriteHeader(code)
+		}
+		srv := httptest.NewServer(http.HandlerFunc(handler))
+		defer srv.Close()
+
+		u, err := url.Parse(srv.URL)
+		require.NoError(t, err)
+		var connectError error
+		go func() {
+			defer close(handlerSend)
+			defer close(done)
+			_, _, connectError = Coder(ctx, u, token)
+		}()
+
+		// Initial: unauthorized
+		handlerSend <- http.StatusUnauthorized
+		// 2nd try: still unauthorized
+		handlerSend <- http.StatusUnauthorized
+		// 3rd try: authorized
+		handlerSend <- http.StatusOK
+
+		cancel()
+
+		<-done
+		require.ErrorContains(t, connectError, "failed to WebSocket dial")
+		require.ErrorIs(t, connectError, context.Canceled)
+	})
 }
 
 type fakeLogDest struct {