提交代码

vendor/symfony/http-kernel/Tests/Exception/TooManyRequestsHttpExceptionTest.php

@@ -0,0 +1,42 @@
+<?php
+
+namespace Symfony\Component\HttpKernel\Tests\Exception;
+
+use Symfony\Component\HttpKernel\Exception\TooManyRequestsHttpException;
+
+class TooManyRequestsHttpExceptionTest extends HttpExceptionTest
+{
+    public function testHeadersDefaultRertyAfter()
+    {
+        $exception = new TooManyRequestsHttpException(10);
+        $this->assertSame(['Retry-After' => 10], $exception->getHeaders());
+    }
+
+    public function testWithHeaderConstruct()
+    {
+        $headers = [
+            'Cache-Control' => 'public, s-maxage=69',
+        ];
+
+        $exception = new TooManyRequestsHttpException(69, null, null, null, $headers);
+
+        $headers['Retry-After'] = 69;
+
+        $this->assertSame($headers, $exception->getHeaders());
+    }
+
+    /**
+     * @dataProvider headerDataProvider
+     */
+    public function testHeadersSetter($headers)
+    {
+        $exception = new TooManyRequestsHttpException(10);
+        $exception->setHeaders($headers);
+        $this->assertSame($headers, $exception->getHeaders());
+    }
+
+    protected function createException(string $message = null, \Throwable $previous = null, ?int $code = 0, array $headers = [])
+    {
+        return new TooManyRequestsHttpException(null, $message, $previous, $code, $headers);
+    }
+}