HttpClientExtensionsTests.cs

using System.IO.Compression;
using System.Net;
using System.Net.Http.Headers;
using System.Net.Sockets;
using System.Text;
using System.Text.Json;
using Microsoft.Extensions.Time.Testing;
using PostHog;
using PostHog.Api;
using PostHog.Library;
#if NETCOREAPP3_1
using TestLibrary.Fakes.Polyfills;
#endif

namespace HttpClientExtensionsTests;

public class ThePostJsonWithRetryAsyncMethod
{
    static readonly Uri BatchUrl = new("https://us.i.posthog.com/batch");

    static PostHogOptions CreateOptions(
        int maxRetries = 3,
        TimeSpan? initialRetryDelay = null,
        TimeSpan? maxRetryDelay = null,
        bool enableCompression = false) => new()
        {
            ProjectToken = "test-api-key",
            MaxRetries = maxRetries,
            InitialRetryDelay = initialRetryDelay ?? TimeSpan.FromMilliseconds(1),
            MaxRetryDelay = maxRetryDelay ?? TimeSpan.FromSeconds(30),
            EnableCompression = enableCompression
        };

    static HttpClient CreateHttpClient(FakeRetryHttpMessageHandler handler)
        => new(handler) { BaseAddress = new Uri("https://us.i.posthog.com") };

    [Fact]
    public async Task ReturnsSuccessOnFirstAttemptWithNoRetry()
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions();
        var timeProvider = new FakeTimeProvider();

        var result = await httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            new { api_key = "test", batch = Array.Empty<object>() },
            timeProvider,
            options,
            CancellationToken.None);

        Assert.NotNull(result);
        Assert.Equal(1, result.Status);
        Assert.Equal(1, handler.RequestCount);
    }

    [Theory]
    [InlineData(HttpStatusCode.InternalServerError)] // 500
    [InlineData(HttpStatusCode.BadGateway)] // 502
    [InlineData(HttpStatusCode.ServiceUnavailable)] // 503
    [InlineData(HttpStatusCode.GatewayTimeout)] // 504
    [InlineData(HttpStatusCode.RequestTimeout)] // 408
    [InlineData(HttpStatusCode.TooManyRequests)] // 429
    public async Task RetriesOnRetryableStatusCodeThenSucceeds(HttpStatusCode statusCode)
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddResponse(statusCode, new { error = "transient" });
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);
        var timeProvider = new FakeTimeProvider();

        // Start the request
        var task = httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            new { api_key = "test", batch = Array.Empty<object>() },
            timeProvider,
            options,
            CancellationToken.None);

        // Wait for first request to complete before advancing time
        await handler.WaitForRequestCountAsync(1);

        // Advance time to trigger the retry
        timeProvider.Advance(TimeSpan.FromSeconds(1));
        var result = await task;

        Assert.NotNull(result);
        Assert.Equal(1, result.Status);
        Assert.Equal(2, handler.RequestCount);
    }

    [Theory]
    [InlineData(HttpStatusCode.BadRequest)] // 400
    [InlineData(HttpStatusCode.Forbidden)] // 403
    public async Task DoesNotRetryOnNonRetryableStatusCodeAndThrowsApiException(HttpStatusCode statusCode)
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddResponse(statusCode, new { type = "error", detail = "Bad request" });
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 }); // Should never be reached
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);
        var timeProvider = new FakeTimeProvider();

        await Assert.ThrowsAsync<ApiException>(() =>
            httpClient.PostJsonWithRetryAsync<ApiResult>(
                BatchUrl,
                new { api_key = "test", batch = Array.Empty<object>() },
                timeProvider,
                options,
                CancellationToken.None));

        Assert.Equal(1, handler.RequestCount);
    }

    [Fact]
    public async Task DoesNotRetryOnUnauthorizedAndThrowsUnauthorizedAccessException()
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddResponse(HttpStatusCode.Unauthorized, new { type = "error", detail = "Invalid API key" });
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 }); // Should never be reached
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);
        var timeProvider = new FakeTimeProvider();

        await Assert.ThrowsAsync<UnauthorizedAccessException>(() =>
            httpClient.PostJsonWithRetryAsync<ApiResult>(
                BatchUrl,
                new { api_key = "test", batch = Array.Empty<object>() },
                timeProvider,
                options,
                CancellationToken.None));

        Assert.Equal(1, handler.RequestCount);
    }

    [Fact]
    public async Task DoesNotRetryOnNotFoundAndThrowsHttpRequestException()
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddResponse(HttpStatusCode.NotFound, new { type = "error", detail = "Not found" });
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 }); // Should never be reached
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);
        var timeProvider = new FakeTimeProvider();

        await Assert.ThrowsAsync<HttpRequestException>(() =>
            httpClient.PostJsonWithRetryAsync<ApiResult>(
                BatchUrl,
                new { api_key = "test", batch = Array.Empty<object>() },
                timeProvider,
                options,
                CancellationToken.None));

        Assert.Equal(1, handler.RequestCount);
    }

    [Fact]
    public async Task ThrowsAfterMaxRetriesWhenAllAttemptsFail()
    {
        var handler = new FakeRetryHttpMessageHandler();
        // Add 4 failures (1 initial + 3 retries)
        handler.AddResponse(HttpStatusCode.ServiceUnavailable, new { type = "error", detail = "Down" });
        handler.AddResponse(HttpStatusCode.ServiceUnavailable, new { type = "error", detail = "Down" });
        handler.AddResponse(HttpStatusCode.ServiceUnavailable, new { type = "error", detail = "Down" });
        handler.AddResponse(HttpStatusCode.ServiceUnavailable, new { type = "error", detail = "Down" });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);
        var timeProvider = new FakeTimeProvider();

        var task = httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            new { api_key = "test", batch = Array.Empty<object>() },
            timeProvider,
            options,
            CancellationToken.None);

        // Advance time for each retry attempt (1 initial + 3 retries)
        for (var i = 1; i <= 4 && !task.IsCompleted; i++)
        {
            await handler.WaitForRequestCountAsync(i);
            timeProvider.Advance(TimeSpan.FromSeconds(1));
        }

        await Assert.ThrowsAsync<ApiException>(() => task);
        Assert.Equal(4, handler.RequestCount); // 1 initial + 3 retries
    }

    [Fact]
    public async Task RespectsRetryAfterDeltaHeader()
    {
        var handler = new FakeRetryHttpMessageHandler();
        using var responseWithRetryAfter = new HttpResponseMessage(HttpStatusCode.TooManyRequests)
        {
            Content = new StringContent("{\"type\": \"error\", \"detail\": \"rate limited\"}")
        };
        responseWithRetryAfter.Headers.RetryAfter = new RetryConditionHeaderValue(TimeSpan.FromMilliseconds(100));
        handler.AddResponse(responseWithRetryAfter);
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);
        var timeProvider = new FakeTimeProvider();

        var task = httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            new { api_key = "test", batch = Array.Empty<object>() },
            timeProvider,
            options,
            CancellationToken.None);

        // Wait for first request to complete before advancing time
        await handler.WaitForRequestCountAsync(1);

#if NET8_0_OR_GREATER
        // Verify task is waiting for the Retry-After delay
        Assert.False(task.IsCompleted, "Task should be waiting for Retry-After delay");
#endif

        // Advance time by the Retry-After value
        timeProvider.Advance(TimeSpan.FromMilliseconds(100));
        var result = await task;

        Assert.NotNull(result);
        Assert.Equal(1, result.Status);
        Assert.Equal(2, handler.RequestCount);
    }

    [Fact]
    public async Task RespectsRetryAfterDateHeader()
    {
        var handler = new FakeRetryHttpMessageHandler();
        using var responseWithRetryAfter = new HttpResponseMessage(HttpStatusCode.TooManyRequests)
        {
            Content = new StringContent("{\"type\": \"error\", \"detail\": \"rate limited\"}")
        };
        var timeProvider = new FakeTimeProvider();
        // Set Retry-After to a date 100ms in the future
        var retryAfterDate = timeProvider.GetUtcNow().AddMilliseconds(100);
        responseWithRetryAfter.Headers.RetryAfter = new RetryConditionHeaderValue(retryAfterDate);
        handler.AddResponse(responseWithRetryAfter);
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);

        var task = httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            new { api_key = "test", batch = Array.Empty<object>() },
            timeProvider,
            options,
            CancellationToken.None);

        // Wait for first request to complete before advancing time
        await handler.WaitForRequestCountAsync(1);

#if NET8_0_OR_GREATER
        // Verify task is waiting for the Retry-After delay
        Assert.False(task.IsCompleted, "Task should be waiting for Retry-After date");
#endif

        // Advance time past the Retry-After date
        timeProvider.Advance(TimeSpan.FromMilliseconds(100));
        var result = await task;

        Assert.NotNull(result);
        Assert.Equal(1, result.Status);
        Assert.Equal(2, handler.RequestCount);
    }

    [Fact]
    public async Task RespectsRetryAfterDateInThePastByUsingZeroDelay()
    {
        var handler = new FakeRetryHttpMessageHandler();
        var timeProvider = new FakeTimeProvider();

        using var responseWithPastRetryAfter = new HttpResponseMessage(HttpStatusCode.TooManyRequests)
        {
            Content = new StringContent("{\"type\": \"error\", \"detail\": \"rate limited\"}")
        };
        // Set Retry-After to a date 100ms in the PAST (simulates clock skew between client and server)
        var retryAfterDate = timeProvider.GetUtcNow().AddMilliseconds(-100);
        responseWithPastRetryAfter.Headers.RetryAfter = new RetryConditionHeaderValue(retryAfterDate);
        handler.AddResponse(responseWithPastRetryAfter);
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);

        var task = httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            new { api_key = "test", batch = Array.Empty<object>() },
            timeProvider,
            options,
            CancellationToken.None);

        // Wait for first request to complete
        await handler.WaitForRequestCountAsync(1);

        // With date in past, delay should be clamped to 0 - even minimal time advancement should trigger retry
        timeProvider.Advance(TimeSpan.FromMilliseconds(1));
        var result = await task;

        Assert.NotNull(result);
        Assert.Equal(1, result.Status);
        Assert.Equal(2, handler.RequestCount);
    }

#if NET8_0_OR_GREATER
    [Fact]
    public async Task ThrowsOperationCanceledExceptionWhenCancellationRequestedDuringDelay()
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddResponse(HttpStatusCode.ServiceUnavailable, new { type = "error" });
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 }); // Should never be reached
        using var httpClient = CreateHttpClient(handler);
        // Use a long delay so we can cancel during it
        var options = CreateOptions(maxRetries: 3, initialRetryDelay: TimeSpan.FromMinutes(1));
        var timeProvider = new FakeTimeProvider();
        using var cts = new CancellationTokenSource();

        var task = httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            new { api_key = "test", batch = Array.Empty<object>() },
            timeProvider,
            options,
            cts.Token);

        // Wait for first request to complete (the one that returns 503)
        await handler.WaitForRequestCountAsync(1);

        // Cancel while waiting for retry delay
        await cts.CancelAsync();

        // TaskCanceledException inherits from OperationCanceledException
        await Assert.ThrowsAnyAsync<OperationCanceledException>(() => task);
        Assert.Equal(1, handler.RequestCount); // Should not have retried after cancellation
    }
#endif

    [Fact]
    public async Task CapsRetryDelayAtMaxRetryDelay()
    {
        var handler = new FakeRetryHttpMessageHandler();
        using var responseWithLargeRetryAfter = new HttpResponseMessage(HttpStatusCode.TooManyRequests)
        {
            Content = new StringContent("{\"type\": \"error\", \"detail\": \"rate limited\"}")
        };
        // Server requests 500ms delay, but our max is 50ms
        responseWithLargeRetryAfter.Headers.RetryAfter = new RetryConditionHeaderValue(TimeSpan.FromMilliseconds(500));
        handler.AddResponse(responseWithLargeRetryAfter);
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3, maxRetryDelay: TimeSpan.FromMilliseconds(50));
        var timeProvider = new FakeTimeProvider();

        var task = httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            new { api_key = "test", batch = Array.Empty<object>() },
            timeProvider,
            options,
            CancellationToken.None);

        // Wait for first request to complete before advancing time
        await handler.WaitForRequestCountAsync(1);

#if NET8_0_OR_GREATER
        // Verify task is waiting (delay was capped, not skipped)
        Assert.False(task.IsCompleted, "Task should be waiting for capped delay");
#endif

        // Advance time by max delay (50ms), not the full 500ms - should be enough due to capping
        timeProvider.Advance(TimeSpan.FromMilliseconds(50));
        var result = await task;

        Assert.NotNull(result);
        Assert.Equal(1, result.Status);
    }

    [Fact]
    public async Task RetriesOnHttpRequestException()
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddException(new HttpRequestException("Network error"));
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);
        var timeProvider = new FakeTimeProvider();

        var task = httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            new { api_key = "test", batch = Array.Empty<object>() },
            timeProvider,
            options,
            CancellationToken.None);

        // Wait for first request to complete before advancing time
        await handler.WaitForRequestCountAsync(1);
        timeProvider.Advance(TimeSpan.FromSeconds(1));
        var result = await task;

        Assert.NotNull(result);
        Assert.Equal(1, result.Status);
        Assert.Equal(2, handler.RequestCount);
    }

    [Fact]
    public async Task RetriesUntilSuccessAfterMultipleServiceUnavailableResponses()
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddResponse(HttpStatusCode.ServiceUnavailable, new { type = "error" });
        handler.AddResponse(HttpStatusCode.ServiceUnavailable, new { type = "error" });
        handler.AddResponse(HttpStatusCode.ServiceUnavailable, new { type = "error" });
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 });
        using var httpClient = CreateHttpClient(handler);
        // Use small delays for fast tests
        var options = CreateOptions(maxRetries: 3, initialRetryDelay: TimeSpan.FromMilliseconds(10));
        var timeProvider = new FakeTimeProvider();

        var task = httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            new { api_key = "test", batch = Array.Empty<object>() },
            timeProvider,
            options,
            CancellationToken.None);

        // Advance time for each retry with exponential backoff
        // Delays: 10ms, 20ms, 40ms (doubled each time)
        for (var i = 1; i <= 4 && !task.IsCompleted; i++)
        {
            await handler.WaitForRequestCountAsync(i);
            timeProvider.Advance(TimeSpan.FromMilliseconds(50));
        }

        var result = await task;

        Assert.NotNull(result);
        Assert.Equal(1, result.Status);
        Assert.Equal(4, handler.RequestCount); // 1 initial + 3 retries
    }

    [Fact]
    public async Task MaxRetriesZeroMeansNoRetry()
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddResponse(HttpStatusCode.ServiceUnavailable, new { type = "error", detail = "Down" });
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 }); // Should never be reached
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 0);
        var timeProvider = new FakeTimeProvider();

        await Assert.ThrowsAsync<ApiException>(() =>
            httpClient.PostJsonWithRetryAsync<ApiResult>(
                BatchUrl,
                new { api_key = "test", batch = Array.Empty<object>() },
                timeProvider,
                options,
                CancellationToken.None));

        Assert.Equal(1, handler.RequestCount); // Only the initial attempt
    }

    [Fact]
    public async Task ClampsNegativeRetryAfterDeltaToZero()
    {
        var handler = new FakeRetryHttpMessageHandler();
        using var responseWithNegativeRetryAfter = new HttpResponseMessage(HttpStatusCode.TooManyRequests)
        {
            Content = new StringContent("{\"type\": \"error\", \"detail\": \"rate limited\"}")
        };
        // Negative delta (malformed server response)
        responseWithNegativeRetryAfter.Headers.RetryAfter = new RetryConditionHeaderValue(TimeSpan.FromMilliseconds(-100));
        handler.AddResponse(responseWithNegativeRetryAfter);
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);
        var timeProvider = new FakeTimeProvider();

        var task = httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            new { api_key = "test", batch = Array.Empty<object>() },
            timeProvider,
            options,
            CancellationToken.None);

        // Wait for first request to complete
        await handler.WaitForRequestCountAsync(1);

        // With negative delta clamped to 0, minimal time advancement triggers retry
        timeProvider.Advance(TimeSpan.FromMilliseconds(1));
        var result = await task;

        Assert.NotNull(result);
        Assert.Equal(1, result.Status);
        Assert.Equal(2, handler.RequestCount);
    }

    [Fact]
    public async Task RetriesOnTaskCanceledExceptionFromTimeout()
    {
        var handler = new FakeRetryHttpMessageHandler();
        // Simulate HttpClient timeout (throws TaskCanceledException with non-canceled token)
        handler.AddException(new TaskCanceledException("The request timed out."));
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);
        var timeProvider = new FakeTimeProvider();

        var task = httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            new { api_key = "test", batch = Array.Empty<object>() },
            timeProvider,
            options,
            CancellationToken.None);

        // Wait for first request to complete before advancing time
        await handler.WaitForRequestCountAsync(1);
        timeProvider.Advance(TimeSpan.FromSeconds(1));
        var result = await task;

        Assert.NotNull(result);
        Assert.Equal(1, result.Status);
        Assert.Equal(2, handler.RequestCount);
    }

#if NET8_0_OR_GREATER
    [Fact]
    public async Task DoesNotRetryOnUserCancellation()
    {
        var handler = new FakeRetryHttpMessageHandler();
        using var cts = new CancellationTokenSource();
        // Cancel the token before starting
        await cts.CancelAsync();
        // The TaskCanceledException will be thrown with a canceled token
        handler.AddException(new TaskCanceledException("Operation was canceled.", null, cts.Token));
        handler.AddResponse(HttpStatusCode.OK, new { status = 1 }); // Should never be reached
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);
        var timeProvider = new FakeTimeProvider();

        // Should throw immediately, not retry
        await Assert.ThrowsAnyAsync<OperationCanceledException>(() =>
            httpClient.PostJsonWithRetryAsync<ApiResult>(
                BatchUrl,
                new { api_key = "test", batch = Array.Empty<object>() },
                timeProvider,
                options,
                cts.Token));

        Assert.Equal(1, handler.RequestCount); // Only the initial attempt
    }
#endif
}

public class ThePostJsonWithNetworkRetryAsyncMethod
{
    static readonly Uri FlagsUrl = new("https://us.i.posthog.com/flags/?v=2");

    static PostHogOptions CreateOptions(int maxRetries = 3) => new()
    {
        ProjectToken = "test-api-key",
        MaxRetries = maxRetries,
        FeatureFlagRequestMaxRetries = maxRetries,
        InitialRetryDelay = TimeSpan.FromMilliseconds(1),
        MaxRetryDelay = TimeSpan.FromSeconds(30)
    };

    static HttpClient CreateHttpClient(FakeRetryHttpMessageHandler handler)
        => new(handler) { BaseAddress = new Uri("https://us.i.posthog.com") };

    [Fact]
    public async Task RetriesOnConnectionResetThenSucceeds()
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddException(new HttpRequestException("Connection reset", new SocketException((int)SocketError.ConnectionReset)));
        handler.AddResponse(HttpStatusCode.OK, new { flags = new { } });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions();
        var timeProvider = new FakeTimeProvider();

        var task = httpClient.PostJsonWithNetworkRetryAsync<FlagsApiResult>(
            FlagsUrl,
            new { api_key = "test", distinct_id = "user-1" },
            timeProvider,
            options,
            CancellationToken.None);

        await handler.WaitForRequestCountAsync(1);
        timeProvider.Advance(TimeSpan.FromSeconds(1));
        var result = await task;

        Assert.NotNull(result);
        Assert.Equal(2, handler.RequestCount);
    }

    [Fact]
    public async Task RetriesUntilSuccessAfterMultipleConnectionResetErrors()
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddException(new HttpRequestException("Connection reset", new SocketException((int)SocketError.ConnectionReset)));
        handler.AddException(new HttpRequestException("Connection reset", new SocketException((int)SocketError.ConnectionReset)));
        handler.AddException(new HttpRequestException("Connection reset", new SocketException((int)SocketError.ConnectionReset)));
        handler.AddResponse(HttpStatusCode.OK, new { flags = new { } });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);
        var timeProvider = new FakeTimeProvider();

        var task = httpClient.PostJsonWithNetworkRetryAsync<FlagsApiResult>(
            FlagsUrl,
            new { api_key = "test", distinct_id = "user-1" },
            timeProvider,
            options,
            CancellationToken.None);

        for (var i = 1; i <= 4 && !task.IsCompleted; i++)
        {
            await handler.WaitForRequestCountAsync(i);
            timeProvider.Advance(TimeSpan.FromMilliseconds(50));
        }

        var result = await task;

        Assert.NotNull(result);
        Assert.Equal(4, handler.RequestCount);
    }

    [Fact]
    public async Task ThrowsAfterMaxRetriesWhenConnectionResetPersists()
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddException(new HttpRequestException("Connection reset", new SocketException((int)SocketError.ConnectionReset)));
        handler.AddException(new HttpRequestException("Connection reset", new SocketException((int)SocketError.ConnectionReset)));
        handler.AddException(new HttpRequestException("Connection reset", new SocketException((int)SocketError.ConnectionReset)));
        handler.AddException(new HttpRequestException("Connection reset", new SocketException((int)SocketError.ConnectionReset)));
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 3);
        var timeProvider = new FakeTimeProvider();

        var task = httpClient.PostJsonWithNetworkRetryAsync<FlagsApiResult>(
            FlagsUrl,
            new { api_key = "test", distinct_id = "user-1" },
            timeProvider,
            options,
            CancellationToken.None);

        for (var i = 1; i <= 4 && !task.IsCompleted; i++)
        {
            await handler.WaitForRequestCountAsync(i);
            timeProvider.Advance(TimeSpan.FromMilliseconds(50));
        }

        await Assert.ThrowsAsync<HttpRequestException>(() => task);
        Assert.Equal(4, handler.RequestCount);
    }

    [Fact]
    public async Task DoesNotRetryWhenFeatureFlagRequestMaxRetriesIsZero()
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddException(new HttpRequestException("Connection reset", new SocketException((int)SocketError.ConnectionReset)));
        handler.AddResponse(HttpStatusCode.OK, new { flags = new { } });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions(maxRetries: 0);
        var timeProvider = new FakeTimeProvider();

        await Assert.ThrowsAsync<HttpRequestException>(() =>
            httpClient.PostJsonWithNetworkRetryAsync<FlagsApiResult>(
                FlagsUrl,
                new { api_key = "test", distinct_id = "user-1" },
                timeProvider,
                options,
                CancellationToken.None));

        Assert.Equal(1, handler.RequestCount);
    }

#if NET8_0_OR_GREATER
    [Fact]
    public async Task DoesNotRetryOnUserCancellation()
    {
        var handler = new FakeRetryHttpMessageHandler();
        using var cts = new CancellationTokenSource();
        await cts.CancelAsync();
        handler.AddException(new TaskCanceledException("Operation was canceled.", null, cts.Token));
        handler.AddResponse(HttpStatusCode.OK, new { flags = new { } });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions();
        var timeProvider = new FakeTimeProvider();

        await Assert.ThrowsAnyAsync<OperationCanceledException>(() =>
            httpClient.PostJsonWithNetworkRetryAsync<FlagsApiResult>(
                FlagsUrl,
                new { api_key = "test", distinct_id = "user-1" },
                timeProvider,
                options,
                cts.Token));

        Assert.Equal(1, handler.RequestCount);
    }
#endif

    [Fact]
    public async Task DoesNotRetryConnectionRefused()
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddException(new HttpRequestException("Connection refused", new SocketException((int)SocketError.ConnectionRefused)));
        handler.AddResponse(HttpStatusCode.OK, new { flags = new { } });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions();
        var timeProvider = new FakeTimeProvider();

        await Assert.ThrowsAsync<HttpRequestException>(() =>
            httpClient.PostJsonWithNetworkRetryAsync<FlagsApiResult>(
                FlagsUrl,
                new { api_key = "test", distinct_id = "user-1" },
                timeProvider,
                options,
                CancellationToken.None));

        Assert.Equal(1, handler.RequestCount);
    }

    [Fact]
    public async Task RetriesOnTaskCanceledExceptionFromTimeoutThenSucceeds()
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddException(new TaskCanceledException("The request timed out."));
        handler.AddResponse(HttpStatusCode.OK, new { flags = new { } });
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions();
        var timeProvider = new FakeTimeProvider();

        var task = httpClient.PostJsonWithNetworkRetryAsync<FlagsApiResult>(
            FlagsUrl,
            new { api_key = "test", distinct_id = "user-1" },
            timeProvider,
            options,
            CancellationToken.None);

        await handler.WaitForRequestCountAsync(1);
        timeProvider.Advance(TimeSpan.FromSeconds(1));
        var result = await task;

        Assert.NotNull(result);
        Assert.Equal(2, handler.RequestCount);
    }

    [Theory]
    [InlineData(HttpStatusCode.RequestTimeout)] // 408
    [InlineData(HttpStatusCode.TooManyRequests)] // 429
    [InlineData(HttpStatusCode.InternalServerError)] // 500
    [InlineData(HttpStatusCode.BadGateway)] // 502
    [InlineData(HttpStatusCode.ServiceUnavailable)] // 503
    [InlineData(HttpStatusCode.GatewayTimeout)] // 504
    public async Task DoesNotRetryOnHttpErrorStatusCodes(HttpStatusCode statusCode)
    {
        var handler = new FakeRetryHttpMessageHandler();
        handler.AddResponse(statusCode, new { type = "error", detail = "server error" });
        handler.AddResponse(HttpStatusCode.OK, new { flags = new { } }); // Should never be reached
        using var httpClient = CreateHttpClient(handler);
        var options = CreateOptions();
        var timeProvider = new FakeTimeProvider();

        await Assert.ThrowsAsync<ApiException>(() =>
            httpClient.PostJsonWithNetworkRetryAsync<FlagsApiResult>(
                FlagsUrl,
                new { api_key = "test", distinct_id = "user-1" },
                timeProvider,
                options,
                CancellationToken.None));

        Assert.Equal(1, handler.RequestCount);
    }
}

public class ThePostCompressedJsonAsyncMethod
{
    static readonly Uri BatchUrl = new("https://us.i.posthog.com/batch");

    [Fact]
    public async Task CompressesRequestBodyWithGzip()
    {
        byte[]? capturedBody = null;
        IEnumerable<string>? capturedContentEncoding = null;
        string? capturedContentType = null;

        var handler = new LambdaHttpMessageHandler(async request =>
        {
            capturedContentType = request.Content?.Headers.ContentType?.MediaType;
            capturedContentEncoding = request.Content?.Headers.ContentEncoding;
            if (request.Content != null)
            {
                capturedBody = await request.Content.ReadAsByteArrayAsync();
            }
            return new HttpResponseMessage(HttpStatusCode.OK)
            {
                Content = new StringContent("{\"status\": 1}")
            };
        });

        using var httpClient = new HttpClient(handler);
        var options = new PostHogOptions
        {
            ProjectToken = "test-api-key",
            EnableCompression = true
        };
        var timeProvider = new FakeTimeProvider();
        var payload = new { api_key = "test", batch = new[] { new { @event = "test-event" } } };

        var result = await httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            payload,
            timeProvider,
            options,
            CancellationToken.None);

        Assert.NotNull(result);
        Assert.NotNull(capturedBody);
        Assert.Equal("application/json", capturedContentType);
        Assert.Contains("gzip", capturedContentEncoding!);

        // Decompress and verify content
        using var compressedStream = new MemoryStream(capturedBody);
        using var gzipStream = new GZipStream(compressedStream, CompressionMode.Decompress);
        using var reader = new StreamReader(gzipStream, Encoding.UTF8);
        var decompressedJson = await reader.ReadToEndAsync();

        Assert.Contains("test-event", decompressedJson, StringComparison.Ordinal);
        Assert.Contains("api_key", decompressedJson, StringComparison.Ordinal);
    }

    [Fact]
    public async Task DoesNotCompressWhenCompressionDisabled()
    {
        IEnumerable<string>? capturedContentEncoding = null;

        var handler = new LambdaHttpMessageHandler(request =>
        {
            capturedContentEncoding = request.Content?.Headers.ContentEncoding;
            // Response disposal is handled by PostJsonWithRetryAsync via using declaration
            return Task.FromResult(new HttpResponseMessage(HttpStatusCode.OK)
            {
                Content = new StringContent("{\"status\": 1}")
            });
        });

        using var httpClient = new HttpClient(handler);
        var options = new PostHogOptions
        {
            ProjectToken = "test-api-key",
            EnableCompression = false
        };
        var timeProvider = new FakeTimeProvider();
        var payload = new { api_key = "test", batch = new[] { new { @event = "test-event" } } };

        await httpClient.PostJsonWithRetryAsync<ApiResult>(
            BatchUrl,
            payload,
            timeProvider,
            options,
            CancellationToken.None);

        Assert.Empty(capturedContentEncoding ?? Enumerable.Empty<string>());
    }
}

/// <summary>
/// A fake HTTP message handler for testing retry logic.
/// Queues responses that are returned in order.
/// </summary>
sealed class FakeRetryHttpMessageHandler : HttpMessageHandler
{
    readonly Queue<Func<Task<HttpResponseMessage>>> _responses = new();
    int _requestCount;

    public int RequestCount => _requestCount;

    /// <summary>
    /// Waits until the request count reaches the specified value.
    /// Use this instead of Task.Delay for deterministic test synchronization.
    /// </summary>
    public async Task WaitForRequestCountAsync(int count, int timeoutMs = 5000)
    {
        var start = DateTime.UtcNow;
        while (Volatile.Read(ref _requestCount) < count)
        {
            if ((DateTime.UtcNow - start).TotalMilliseconds > timeoutMs)
            {
                throw new TimeoutException($"Timed out waiting for request count {count}. Current: {_requestCount}");
            }
            await Task.Yield();
        }
    }

    // Note: HttpResponseMessage disposal is handled by PostJsonWithRetryAsync via using declaration.
    // The handler creates responses that are returned to and disposed by the calling code.
    public void AddResponse(HttpStatusCode statusCode, object body)
    {
        var json = JsonSerializer.Serialize(body);
        _responses.Enqueue(() => Task.FromResult(new HttpResponseMessage(statusCode)
        {
            Content = new StringContent(json, Encoding.UTF8, "application/json")
        }));
    }

    public void AddResponse(HttpResponseMessage response)
    {
        _responses.Enqueue(() => Task.FromResult(response));
    }

    public void AddException(Exception exception)
    {
        _responses.Enqueue(() => throw exception);
    }

    protected override async Task<HttpResponseMessage> SendAsync(
        HttpRequestMessage request,
        CancellationToken cancellationToken)
    {
        Interlocked.Increment(ref _requestCount);

        if (_responses.Count == 0)
        {
            return new HttpResponseMessage(HttpStatusCode.NotFound);
        }

        return await _responses.Dequeue()();
    }
}

public class TheDoubleWithCapMethod
{
    [Fact]
    public void DoublesValueWhenBelowMax()
    {
        var current = TimeSpan.FromMilliseconds(100);
        var max = TimeSpan.FromMilliseconds(1000);

        var result = HttpClientExtensions.DoubleWithCap(current, max);

        Assert.Equal(TimeSpan.FromMilliseconds(200), result);
    }

    [Fact]
    public void CapsAtMaxWhenDoublingWouldExceedMax()
    {
        var current = TimeSpan.FromMilliseconds(600);
        var max = TimeSpan.FromMilliseconds(1000);

        var result = HttpClientExtensions.DoubleWithCap(current, max);

        // 600 * 2 = 1200, which exceeds 1000, so cap at 1000
        Assert.Equal(TimeSpan.FromMilliseconds(1000), result);
    }

    [Fact]
    public void ReturnsMaxWhenCurrentEqualsMax()
    {
        var current = TimeSpan.FromMilliseconds(1000);
        var max = TimeSpan.FromMilliseconds(1000);

        var result = HttpClientExtensions.DoubleWithCap(current, max);

        Assert.Equal(max, result);
    }

    [Fact]
    public void ReturnsMaxWhenCurrentExceedsMax()
    {
        var current = TimeSpan.FromMilliseconds(1500);
        var max = TimeSpan.FromMilliseconds(1000);

        var result = HttpClientExtensions.DoubleWithCap(current, max);

        Assert.Equal(max, result);
    }

    [Fact]
    public void HandlesOverflowProtectionForLargeValues()
    {
        // Use a value that would overflow if doubled without protection
        var current = TimeSpan.FromTicks(long.MaxValue / 2 + 1);
        var max = TimeSpan.MaxValue;

        var result = HttpClientExtensions.DoubleWithCap(current, max);

        // Should cap at max instead of overflowing
        Assert.Equal(max, result);
    }

    [Fact]
    public void DoublesCorrectlyAtBoundaryJustBelowHalfMax()
    {
        var max = TimeSpan.FromMilliseconds(1000);
        var current = TimeSpan.FromMilliseconds(499); // Just below half of max

        var result = HttpClientExtensions.DoubleWithCap(current, max);

        // 499 * 2 = 998, which is below 1000
        Assert.Equal(TimeSpan.FromMilliseconds(998), result);
    }

    [Fact]
    public void CapsAtMaxWhenCurrentIsExactlyHalfOfMax()
    {
        var max = TimeSpan.FromMilliseconds(1000);
        var current = TimeSpan.FromMilliseconds(500); // Exactly half of max

        var result = HttpClientExtensions.DoubleWithCap(current, max);

        // 500 * 2 = 1000, equals max
        Assert.Equal(max, result);
    }

    [Fact]
    public void CapsAtMaxWhenCurrentIsJustAboveHalfOfMax()
    {
        var max = TimeSpan.FromMilliseconds(1000);
        var current = TimeSpan.FromMilliseconds(501); // Just above half of max

        var result = HttpClientExtensions.DoubleWithCap(current, max);

        // 501 > 1000/2, so cap at max to avoid exceeding
        Assert.Equal(max, result);
    }
}

public class TheExponentialBackoffBehavior
{
    [Fact]
    public void DelaysDoubleWithEachRetry()
    {
        // Verify the exponential backoff sequence: 100ms -> 200ms -> 400ms -> 800ms
        var initialDelay = TimeSpan.FromMilliseconds(100);
        var maxDelay = TimeSpan.FromSeconds(30);

        var delay1 = initialDelay;
        var delay2 = HttpClientExtensions.DoubleWithCap(delay1, maxDelay);
        var delay3 = HttpClientExtensions.DoubleWithCap(delay2, maxDelay);
        var delay4 = HttpClientExtensions.DoubleWithCap(delay3, maxDelay);

        Assert.Equal(TimeSpan.FromMilliseconds(100), delay1);
        Assert.Equal(TimeSpan.FromMilliseconds(200), delay2);
        Assert.Equal(TimeSpan.FromMilliseconds(400), delay3);
        Assert.Equal(TimeSpan.FromMilliseconds(800), delay4);
    }

    [Fact]
    public void DelaysCappedAtMaxAfterMultipleDoublings()
    {
        // Start with 1 second, max of 5 seconds
        // Sequence: 1s -> 2s -> 4s -> 5s (capped) -> 5s (stays at cap)
        var initialDelay = TimeSpan.FromSeconds(1);
        var maxDelay = TimeSpan.FromSeconds(5);

        var delay1 = initialDelay;
        var delay2 = HttpClientExtensions.DoubleWithCap(delay1, maxDelay);
        var delay3 = HttpClientExtensions.DoubleWithCap(delay2, maxDelay);
        var delay4 = HttpClientExtensions.DoubleWithCap(delay3, maxDelay);
        var delay5 = HttpClientExtensions.DoubleWithCap(delay4, maxDelay);

        Assert.Equal(TimeSpan.FromSeconds(1), delay1);
        Assert.Equal(TimeSpan.FromSeconds(2), delay2);
        Assert.Equal(TimeSpan.FromSeconds(4), delay3);
        Assert.Equal(TimeSpan.FromSeconds(5), delay4); // Capped
        Assert.Equal(TimeSpan.FromSeconds(5), delay5); // Stays at cap
    }
}

/// <summary>
/// A simple lambda-based HTTP message handler for testing.
/// </summary>
sealed class LambdaHttpMessageHandler(
    Func<HttpRequestMessage, Task<HttpResponseMessage>> handler) : HttpMessageHandler
{
    protected override Task<HttpResponseMessage> SendAsync(
        HttpRequestMessage request,
        CancellationToken cancellationToken)
        => handler(request);
}