Function TestIsRetryableError

cli/ssh_internal_test.go:489–536

(t *testing.T)

}

func TestIsRetryableError(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name      string
		err       error
		retryable bool
	}{
		{"Nil", nil, false},
		{"ContextCanceled", context.Canceled, false},
		{"ContextDeadlineExceeded", context.DeadlineExceeded, false},
		{"WrappedContextCanceled", xerrors.Errorf("wrapped: %w", context.Canceled), false},
		{"DNSError", &net.DNSError{Err: "no such host", Name: "example.com", IsNotFound: true}, true},
		{"OpError", &net.OpError{Op: "dial", Net: "tcp", Err: &os.SyscallError{}}, true},
		{"WrappedDNSError", xerrors.Errorf("connect: %w", &net.DNSError{Err: "no such host", Name: "example.com"}), true},
		{"SDKError_500", codersdk.NewTestError(http.StatusInternalServerError, "GET", "/api"), true},
		{"SDKError_502", codersdk.NewTestError(http.StatusBadGateway, "GET", "/api"), true},
		{"SDKError_503", codersdk.NewTestError(http.StatusServiceUnavailable, "GET", "/api"), true},
		{"SDKError_401", codersdk.NewTestError(http.StatusUnauthorized, "GET", "/api"), false},
		{"SDKError_403", codersdk.NewTestError(http.StatusForbidden, "GET", "/api"), false},
		{"SDKError_404", codersdk.NewTestError(http.StatusNotFound, "GET", "/api"), false},
		{"GenericError", xerrors.New("something went wrong"), false},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()
			assert.Equal(t, tt.retryable, isRetryableError(tt.err))
		})
	}

	// net.Dialer.Timeout produces *net.OpError that matches both
	// IsConnectionError and context.DeadlineExceeded. Verify it is retryable.
	t.Run("DialTimeout", func(t *testing.T) {
		t.Parallel()
		ctx, cancel := context.WithDeadline(context.Background(), time.Now())
		defer cancel()
		<-ctx.Done() // ensure deadline has fired
		_, err := (&net.Dialer{}).DialContext(ctx, "tcp", "127.0.0.1:1")
		require.Error(t, err)
		// Proves the ambiguity: this error matches BOTH checks.
		require.ErrorIs(t, err, context.DeadlineExceeded)
		require.ErrorAs(t, err, new(*net.OpError))
		assert.True(t, isRetryableError(err))
		// Also when wrapped, as runCoderConnectStdio does.
		assert.True(t, isRetryableError(xerrors.Errorf("dial coder connect: %w", err)))
	})
}

func TestRetryWithInterval(t *testing.T) {
	t.Parallel()