Generate accept header from request key

amphp · Oct 22, 2023 · 4e0ee4d · 4e0ee4d
1 parent f5bb36b
commit 4e0ee4d
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/test/WebsocketTest.php b/test/WebsocketTest.php
@@ -21,6 +21,8 @@
 use Psr\Log\NullLogger;
 use Revolt\EventLoop;
 use function Amp\delay;
+use function Amp\Websocket\generateAcceptFromKey;
+use function Amp\Websocket\generateKey;
 
 class WebsocketTest extends AsyncTestCase
 {
@@ -145,7 +147,7 @@ public function createRequest(): Request
     {
         $headers = [
             "host" => ["localhost"],
-            "sec-websocket-key" => ["x3JJHMbDL1EzLkh9GBhXDw=="],
+            "sec-websocket-key" => [generateKey()],
             "sec-websocket-version" => ["13"],
             "upgrade" => ["websocket"],
             "connection" => ["upgrade"],
@@ -178,7 +180,7 @@ public function provideHandshakes(): iterable
             [
                 "upgrade" => ["websocket"],
                 "connection" => ["upgrade"],
-                "sec-websocket-accept" => ["HSmrc0sMlYUkAGmm5OPpG2HaGWk="],
+                "sec-websocket-accept" => [generateAcceptFromKey($request->getHeader('sec-websocket-key'))],
             ],
         ];
 
@@ -230,7 +232,7 @@ public function provideHandshakes(): iterable
             [
                 "upgrade" => ["websocket"],
                 "connection" => ["upgrade"],
-                "sec-websocket-accept" => ["HSmrc0sMlYUkAGmm5OPpG2HaGWk="],
+                "sec-websocket-accept" => [generateAcceptFromKey($request->getHeader('sec-websocket-key'))],
                 "sec-websocket-extensions" => ["permessage-deflate; client_max_window_bits=15"],
             ],
         ];
@@ -244,7 +246,7 @@ public function provideHandshakes(): iterable
             [
                 "upgrade" => ["websocket"],
                 "connection" => ["upgrade"],
-                "sec-websocket-accept" => ["HSmrc0sMlYUkAGmm5OPpG2HaGWk="],
+                "sec-websocket-accept" => [generateAcceptFromKey($request->getHeader('sec-websocket-key'))],
             ],
         ];
     }