<?php

namespace AsyncAws\Core\Tests\HttpClient;

use AsyncAws\Core\HttpClient\AwsRetryStrategy;

use AsyncAws\Core\Test\TestCase;

use PHPUnit\Framework\Attributes\DataProvider;

use Symfony\Component\HttpClient\MockHttpClient;

use Symfony\Component\HttpClient\Response\AsyncContext;

use Symfony\Component\HttpClient\Response\MockResponse;

use Symfony\Component\HttpClient\Retry\GenericRetryStrategy;

class AwsRetryStrategyTest extends TestCase

{

#[DataProvider('provideRetries')]

public function testShouldRetry(?bool $expected, int $statusCode, ?string $response): void

{

if (!class_exists(GenericRetryStrategy::class)) {

self::markTestSkipped('AwsRetryStrategy requires symfony/http-client 5.2');

}

$strategy = new AwsRetryStrategy();

$context = $this->getContext($statusCode);

self::assertSame($expected, $strategy->shouldRetry($context, $response, null));

}

public static function provideRetries(): iterable

{

yield [false, 200, null];

yield [true, 429, null];

yield [null, 400, null];

yield [false, 400, 'this is invalid'];

yield [false, 400, '{"__type": "RandomError"}'];

yield [true, 400, '{"__type": "RequestLimitExceeded"}'];

}

private function getContext(int $statusCode): AsyncContext

{

$passthru = null;

$info = [

'http_code' => $statusCode,

];

$response = new MockResponse('', $info);

return new AsyncContext($passthru, new MockHttpClient(), $response, $info, null, 0);

}

}