From 690f9b10e88bb60fd6dc0f7c42821dbc06ac87c9 Mon Sep 17 00:00:00 2001
From: Veesh Goldman <rabbiveesh@gmail.com>
Date: Mon, 26 Oct 2020 14:32:14 +0200
Subject: [PATCH] Websocket Lite app test uses subtests

---
 t/mojolicious/websocket_lite_app.t | 443 ++++++++++++++++-------------
 1 file changed, 240 insertions(+), 203 deletions(-)

diff --git a/t/mojolicious/websocket_lite_app.t b/t/mojolicious/websocket_lite_app.t
index bf03d36b6f..8323aca40e 100644
--- a/t/mojolicious/websocket_lite_app.t
+++ b/t/mojolicious/websocket_lite_app.t
@@ -134,208 +134,245 @@ post {data => 'plain nested too!'};
 
 my $t = Test::Mojo->new;
 
-# Simple roundtrip
-$t->websocket_ok('/echo')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
-
-# Multiple roundtrips
-$t->websocket_ok('/echo')->send_ok('hello again')->message_ok->message_is('echo: hello again')
-  ->send_ok('and one more time')->message_ok->message_is('echo: and one more time')->finish_ok;
-
-# Simple roundtrip with redirect
-$t->get_ok('/not_echo/308')->status_is(308);
-$t->ua->max_redirects(10);
-$t->get_ok('/not_echo/302')->status_is(200)->content_is('plain echo!');
-$t->websocket_ok('/not_echo/308')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
-$t->websocket_ok('/not_echo/307')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
-$t->websocket_ok('/not_echo/303')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
-$t->websocket_ok('/not_echo/302')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
-$t->websocket_ok('/not_echo/301')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
-
-# Custom headers and protocols
-my $headers = {DNT => 1, 'Sec-WebSocket-Key' => 'NTA2MDAyMDU1NjMzNjkwMg=='};
-$t->websocket_ok('/echo' => $headers => ['foo', 'bar', 'baz'])
-  ->header_is('Sec-WebSocket-Accept' => 'I+x5C3/LJxrmDrWw42nMP4pCSes=')->header_is('Sec-WebSocket-Protocol' => undef)
-  ->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
-is $t->tx->req->headers->dnt,                    1,               'right "DNT" value';
-is $t->tx->req->headers->sec_websocket_protocol, 'foo, bar, baz', 'right "Sec-WebSocket-Protocol" value';
-
-# Bytes
-$t->websocket_ok('/echo')->send_ok({binary => 'bytes!'})->message_ok->message_is({binary => 'bytes!'})
-  ->send_ok({binary => 'bytes!'})->message_ok->message_isnt({text => 'bytes!'})->finish_ok;
-
-# Bytes in multiple frames
-$t->websocket_ok('/echo')->send_ok([0, 0, 0, 0, 2, 'a'])->send_ok([0, 0, 0, 0, 0, 'b'])->send_ok([1, 0, 0, 0, 0, 'c'])
-  ->message_ok->message_is({binary => 'abc'})->finish_ok;
-
-# Zero
-$t->websocket_ok('/echo')->send_ok(0)->message_ok->message_is('echo: 0')->send_ok(0)
-  ->message_ok->message_like({text => qr/0/})->finish_ok(1000)->finished_ok(1000);
-
-# 64-bit binary message
-$t->request_ok($t->ua->build_websocket_tx('/echo'));
-is $t->tx->max_websocket_size, 262144, 'right size';
-$t->tx->max_websocket_size(65538);
-$t->send_ok({binary => 'a' x 65538})->message_ok->message_is({binary => 'a' x 65538})->finish_ok->finished_ok(1005);
-
-# 64-bit binary message (too large for server)
-$t->websocket_ok('/echo')->send_ok({binary => 'b' x 65539})->finished_ok(1009);
-
-# 64-bit binary message (too large for client)
-$t->websocket_ok('/echo');
-$t->tx->max_websocket_size(65536);
-$t->send_ok({binary => 'c' x 65537})->finished_ok(1009);
-
-# Binary message in two frames without FIN bit (too large for server)
-$t->websocket_ok('/echo')->send_ok([0, 0, 0, 0, 2, 'd' x 30000])->send_ok([0, 0, 0, 0, 0, 'd' x 35539])
-  ->finished_ok(1009);
-
-# Plain alternative
-$t->get_ok('/echo')->status_is(200)->content_is('plain echo!');
-
-# Compression denied by the server
-$t->websocket_ok('/no_compression' => {'Sec-WebSocket-Extensions' => 'permessage-deflate'});
-is $t->tx->req->headers->sec_websocket_extensions, 'permessage-deflate', 'right "Sec-WebSocket-Extensions" value';
-ok !$t->tx->compressed, 'WebSocket has no compression';
-$t->send_ok({binary => 'a' x 500})->message_ok->message_is({binary => 'a' x 500})->finish_ok;
-
-# Compressed message ("permessage-deflate")
-$t->websocket_ok('/echo' => {'Sec-WebSocket-Extensions' => 'permessage-deflate'});
-ok $t->tx->compressed, 'WebSocket has compression';
-$t->send_ok({binary => 'a' x 10000})->header_is('Sec-WebSocket-Extensions' => 'permessage-deflate');
-is $t->tx->req->headers->sec_websocket_extensions, 'permessage-deflate', 'right "Sec-WebSocket-Extensions" value';
-my $payload;
-$t->tx->once(
-  frame => sub {
-    my ($tx, $frame) = @_;
-    $payload = $frame->[5];
-  }
-);
-$t->message_ok->message_is({binary => 'a' x 10000});
-ok length $payload < 10000, 'message has been compressed';
-$t->finish_ok->finished_ok(1005);
-
-# Timeout
-$t->websocket_ok('/timeout')->send_ok('timeout')->message_ok->message_is('timeout: 30')->send_ok('0')
-  ->message_ok->message_is('0: 0')->send_ok('120')->message_ok->message_is('120: 120')->finish_ok;
-
-# Compressed message exceeding the limit when decompressed
-$t->websocket_ok('/echo' => {'Sec-WebSocket-Extensions' => 'permessage-deflate'})
-  ->header_is('Sec-WebSocket-Extensions' => 'permessage-deflate')->send_ok({binary => 'a' x 1000000})
-  ->finished_ok(1009);
-
-# Huge message that doesn't compress very well
-my $huge = join '', map { int rand(9) } 1 .. 65538;
-$t->websocket_ok('/echo' => {'Sec-WebSocket-Extensions' => 'permessage-deflate'})->send_ok({binary => $huge})
-  ->message_ok->message_is({binary => $huge})->finish_ok;
-
-# Protocol negotiation
-$t->websocket_ok('/protocols' => ['bar'])->message_ok->message_is('bar')->message_ok->message_is('bar')->finish_ok;
-is $t->tx->protocol, 'bar', 'right protocol';
-is $t->tx->res->headers->sec_websocket_protocol, 'bar', 'right "Sec-WebSocket-Protocol" value';
-$t->websocket_ok('/protocols' => ['baz', 'bar', 'foo'])->message_ok->message_is('foo')->message_ok->message_is('foo')
-  ->finish_ok;
-is $t->tx->protocol, 'foo', 'right protocol';
-is $t->tx->res->headers->sec_websocket_protocol, 'foo', 'right "Sec-WebSocket-Protocol" value';
-$t->websocket_ok('/protocols' => ['0'])->message_ok->message_is('0')->message_ok->message_is('0')->finish_ok;
-is $t->tx->protocol, '0', 'right protocol';
-is $t->tx->res->headers->sec_websocket_protocol, '0', 'right "Sec-WebSocket-Protocol" value';
-$t->websocket_ok('/protocols' => [''])->message_ok->message_is('none')->message_ok->message_is('none')->finish_ok;
-is $t->tx->protocol, undef, 'no protocol';
-$t->websocket_ok('/protocols' => ['', '', ''])->message_ok->message_is('none')->message_ok->message_is('none')
-  ->finish_ok;
-is $t->tx->protocol, undef, 'no protocol';
-$t->websocket_ok('/protocols')->message_ok->message_is('none')->message_ok->message_is('none')->finish_ok;
-is $t->tx->protocol, undef, 'no protocol';
-
-# JSON roundtrips (with a lot of different tests)
-$t->websocket_ok('/json')->send_ok({json => {test => 23, snowman => '☃'}})
-  ->message_ok->json_message_is('' => {test => 24, snowman => '☃'})
-  ->json_message_is('' => {test => 24, snowman => '☃'})->json_message_has('/test')->json_message_hasnt('/test/2')
-  ->send_ok({binary => encode_json([1, 2, 3])}, 'with description')->message_ok('with description')
-  ->message_is('[1,2,3,4]')->message_is('[1,2,3,4]', 'with description')->message_isnt('[1,2,3]')
-  ->message_isnt('[1,2,3]', 'with description')->message_like(qr/3/)->message_like(qr/3/, 'with description')
-  ->message_unlike(qr/5/)->message_unlike(qr/5/, 'with description')->json_message_is([1, 2, 3, 4])
-  ->json_message_is([1, 2, 3, 4])->send_ok({binary => encode_json([1, 2, 3])})->message_ok->json_message_has('/2')
-  ->json_message_has('/2', 'with description')->json_message_hasnt('/5')->json_message_hasnt('/5', 'with description')
-  ->json_message_is('/2' => 3)->json_message_is('/2' => 3, 'with description')->send_ok({json => {'☃' => [1, 2, 3]}})
-  ->message_ok->json_message_is('/☃', [1, 2, 3])->json_message_like('/☃/1' => qr/\d/)
-  ->json_message_like('/☃/2' => qr/3/, 'with description')->json_message_unlike('/☃/1' => qr/[a-z]/)
-  ->json_message_unlike('/☃/2' => qr/2/, 'with description')->send_ok({json => 'works'})
-  ->message_ok->json_message_is('works')->send_ok({json => undef})->message_ok->json_message_is(undef)->finish_ok;
-
-# Plain request
-$t->get_ok('/plain')->status_is(200)->content_is('Nothing to see here!');
-
-# Server push
-$t->websocket_ok('/push')->message_ok->message_is('push')->message_ok->message_is('push')
-  ->message_ok->message_is('push')->finish_ok;
-$t->websocket_ok('/push')->message_ok->message_unlike(qr/shift/)->message_ok->message_isnt('shift')
-  ->message_ok->message_like(qr/us/)->message_ok->message_unlike({binary => qr/push/})->finish_ok;
-
-# Another plain request
-$t->get_ok('/plain')->status_is(200)->content_is('Nothing to see here!');
-
-# Multiple roundtrips
-$t->websocket_ok('/echo')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
-$t->websocket_ok('/echo')->send_ok('this')->send_ok('just')->send_ok('works')->message_ok->message_is('echo: this')
-  ->message_ok->message_is('echo: just')->message_ok->message_is('echo: works')->message_like(qr/orks/)->finish_ok;
-
-# Another plain request
-$t->get_ok('/plain')->status_is(200)->content_is('Nothing to see here!');
-
-# Unicode roundtrips
-$t->websocket_ok('/unicode')->send_ok('hello')->message_ok->message_is('♥: hello')->finish_ok;
-$t->websocket_ok('/unicode')->send_ok('hello again')->message_ok->message_is('♥: hello again')
-  ->send_ok('and one ☃ more time')->message_ok->message_is('♥: and one ☃ more time')->finish_ok;
-
-# Binary frame and events
-my $bytes = b("I ♥ Mojolicious")->encode('UTF-16LE')->to_string;
-$t->websocket_ok('/bytes');
-my $binary;
-$t->tx->on(
-  frame => sub {
-    my ($ws, $frame) = @_;
-    $binary++ if $frame->[4] == 2;
-  }
-);
-my $close;
-$t->tx->on(finish => sub { shift; $close = [@_] });
-$t->send_ok({binary => $bytes})->message_ok->message_is($bytes);
-ok $binary, 'received binary frame';
-$binary = undef;
-$t->send_ok({text => $bytes})->message_ok->message_is($bytes);
-ok !$binary, 'received text frame';
-$t->finish_ok(1000 => 'Have a nice day!');
-is_deeply $close, [1000, 'Have a nice day!'], 'right status and message';
-
-# Binary roundtrips
-$t->request_ok($t->ua->build_websocket_tx('/bytes'))->send_ok({binary => $bytes})->message_ok->message_is($bytes)
-  ->send_ok({binary => $bytes})->message_ok->message_is($bytes)->finish_ok;
-
-# Two responses
-$t->websocket_ok('/once')->send_ok('hello')->message_ok->message_is('ONE: hello')->message_ok->message_is('TWO: hello')
-  ->send_ok('hello')->message_ok->message_is('ONE: hello')->send_ok('hello')->message_ok->message_is('ONE: hello')
-  ->finish_ok;
-
-# WebSocket connection gets closed right away
-$t->websocket_ok('/close')->finished_ok(1001);
-
-# WebSocket connection gets closed after one message
-$t->websocket_ok('/one_sided')->message_ok->message_is('I ♥ Mojolicious!')->finished_ok(1005);
-
-# Nested WebSocket
-$t->websocket_ok('/nested')->send_ok('hello')->message_ok->message_is('nested echo: hello')->finished_ok(1000);
-
-# Test custom message
-$t->message([binary => 'foobarbaz'])->message_like(qr/bar/)->message_is({binary => 'foobarbaz'});
-
-# Nested WebSocket with cookie
-$t->websocket_ok('/nested')->send_ok('hello')->message_ok->message_is('nested echo: helloagain')->finished_ok(1000);
-
-# Nested plain request
-$t->get_ok('/nested')->status_is(200)->content_is('plain nested!');
-
-# Another nested plain request
-$t->post_ok('/nested')->status_is(200)->content_is('plain nested too!');
+subtest 'Simple roundtrip' => sub {
+  $t->websocket_ok('/echo')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
+};
+
+subtest 'Multiple roundtrips' => sub {
+  $t->websocket_ok('/echo')->send_ok('hello again')->message_ok->message_is('echo: hello again')
+    ->send_ok('and one more time')->message_ok->message_is('echo: and one more time')->finish_ok;
+};
+
+subtest 'Simple roundtrip with redirect' => sub {
+  $t->get_ok('/not_echo/308')->status_is(308);
+  $t->ua->max_redirects(10);
+  $t->get_ok('/not_echo/302')->status_is(200)->content_is('plain echo!');
+  $t->websocket_ok('/not_echo/308')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
+  $t->websocket_ok('/not_echo/307')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
+  $t->websocket_ok('/not_echo/303')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
+  $t->websocket_ok('/not_echo/302')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
+  $t->websocket_ok('/not_echo/301')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
+};
+
+subtest 'Custom headers and protocols' => sub {
+  my $headers = {DNT => 1, 'Sec-WebSocket-Key' => 'NTA2MDAyMDU1NjMzNjkwMg=='};
+  $t->websocket_ok('/echo' => $headers => ['foo', 'bar', 'baz'])
+    ->header_is('Sec-WebSocket-Accept' => 'I+x5C3/LJxrmDrWw42nMP4pCSes=')->header_is('Sec-WebSocket-Protocol' => undef)
+    ->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
+  is $t->tx->req->headers->dnt,                    1,               'right "DNT" value';
+  is $t->tx->req->headers->sec_websocket_protocol, 'foo, bar, baz', 'right "Sec-WebSocket-Protocol" value';
+};
+
+subtest 'Bytes' => sub {
+  $t->websocket_ok('/echo')->send_ok({binary => 'bytes!'})->message_ok->message_is({binary => 'bytes!'})
+    ->send_ok({binary => 'bytes!'})->message_ok->message_isnt({text => 'bytes!'})->finish_ok;
+};
+
+subtest 'Bytes in multiple frames' => sub {
+  $t->websocket_ok('/echo')->send_ok([0, 0, 0, 0, 2, 'a'])->send_ok([0, 0, 0, 0, 0, 'b'])
+    ->send_ok([1, 0, 0, 0, 0, 'c'])->message_ok->message_is({binary => 'abc'})->finish_ok;
+};
+
+subtest 'Zero' => sub {
+  $t->websocket_ok('/echo')->send_ok(0)->message_ok->message_is('echo: 0')->send_ok(0)
+    ->message_ok->message_like({text => qr/0/})->finish_ok(1000)->finished_ok(1000);
+};
+
+subtest '64-bit binary message' => sub {
+  $t->request_ok($t->ua->build_websocket_tx('/echo'));
+  is $t->tx->max_websocket_size, 262144, 'right size';
+  $t->tx->max_websocket_size(65538);
+  $t->send_ok({binary => 'a' x 65538})->message_ok->message_is({binary => 'a' x 65538})->finish_ok->finished_ok(1005);
+};
+
+subtest '64-bit binary message (too large for server)' => sub {
+  $t->websocket_ok('/echo')->send_ok({binary => 'b' x 65539})->finished_ok(1009);
+};
+
+subtest '64-bit binary message (too large for client)' => sub {
+  $t->websocket_ok('/echo');
+  $t->tx->max_websocket_size(65536);
+  $t->send_ok({binary => 'c' x 65537})->finished_ok(1009);
+};
+
+subtest 'Binary message in two frames without FIN bit (too large for server)' => sub {
+  $t->websocket_ok('/echo')->send_ok([0, 0, 0, 0, 2, 'd' x 30000])->send_ok([0, 0, 0, 0, 0, 'd' x 35539])
+    ->finished_ok(1009);
+};
+
+subtest 'Plain alternative' => sub {
+  $t->get_ok('/echo')->status_is(200)->content_is('plain echo!');
+};
+
+subtest 'Compression denied by the server' => sub {
+  $t->websocket_ok('/no_compression' => {'Sec-WebSocket-Extensions' => 'permessage-deflate'});
+  is $t->tx->req->headers->sec_websocket_extensions, 'permessage-deflate', 'right "Sec-WebSocket-Extensions" value';
+  ok !$t->tx->compressed, 'WebSocket has no compression';
+  $t->send_ok({binary => 'a' x 500})->message_ok->message_is({binary => 'a' x 500})->finish_ok;
+};
+
+subtest 'Compressed message ("permessage-deflate")' => sub {
+  $t->websocket_ok('/echo' => {'Sec-WebSocket-Extensions' => 'permessage-deflate'});
+  ok $t->tx->compressed, 'WebSocket has compression';
+  $t->send_ok({binary => 'a' x 10000})->header_is('Sec-WebSocket-Extensions' => 'permessage-deflate');
+  is $t->tx->req->headers->sec_websocket_extensions, 'permessage-deflate', 'right "Sec-WebSocket-Extensions" value';
+  my $payload;
+  $t->tx->once(
+    frame => sub {
+      my ($tx, $frame) = @_;
+      $payload = $frame->[5];
+    }
+  );
+  $t->message_ok->message_is({binary => 'a' x 10000});
+  ok length $payload < 10000, 'message has been compressed';
+  $t->finish_ok->finished_ok(1005);
+};
+
+subtest 'Timeout' => sub {
+  $t->websocket_ok('/timeout')->send_ok('timeout')->message_ok->message_is('timeout: 30')->send_ok('0')
+    ->message_ok->message_is('0: 0')->send_ok('120')->message_ok->message_is('120: 120')->finish_ok;
+};
+
+subtest 'Compressed message exceeding the limit when decompressed' => sub {
+  $t->websocket_ok('/echo' => {'Sec-WebSocket-Extensions' => 'permessage-deflate'})
+    ->header_is('Sec-WebSocket-Extensions' => 'permessage-deflate')->send_ok({binary => 'a' x 1000000})
+    ->finished_ok(1009);
+};
+
+subtest "Huge message that doesn't compress very well" => sub {
+  my $huge = join '', map { int rand(9) } 1 .. 65538;
+  $t->websocket_ok('/echo' => {'Sec-WebSocket-Extensions' => 'permessage-deflate'})->send_ok({binary => $huge})
+    ->message_ok->message_is({binary => $huge})->finish_ok;
+};
+
+subtest 'Protocol negotiation' => sub {
+  $t->websocket_ok('/protocols' => ['bar'])->message_ok->message_is('bar')->message_ok->message_is('bar')->finish_ok;
+  is $t->tx->protocol, 'bar', 'right protocol';
+  is $t->tx->res->headers->sec_websocket_protocol, 'bar', 'right "Sec-WebSocket-Protocol" value';
+  $t->websocket_ok('/protocols' => ['baz', 'bar', 'foo'])->message_ok->message_is('foo')->message_ok->message_is('foo')
+    ->finish_ok;
+  is $t->tx->protocol, 'foo', 'right protocol';
+  is $t->tx->res->headers->sec_websocket_protocol, 'foo', 'right "Sec-WebSocket-Protocol" value';
+  $t->websocket_ok('/protocols' => ['0'])->message_ok->message_is('0')->message_ok->message_is('0')->finish_ok;
+  is $t->tx->protocol, '0', 'right protocol';
+  is $t->tx->res->headers->sec_websocket_protocol, '0', 'right "Sec-WebSocket-Protocol" value';
+  $t->websocket_ok('/protocols' => [''])->message_ok->message_is('none')->message_ok->message_is('none')->finish_ok;
+  is $t->tx->protocol, undef, 'no protocol';
+  $t->websocket_ok('/protocols' => ['', '', ''])->message_ok->message_is('none')->message_ok->message_is('none')
+    ->finish_ok;
+  is $t->tx->protocol, undef, 'no protocol';
+  $t->websocket_ok('/protocols')->message_ok->message_is('none')->message_ok->message_is('none')->finish_ok;
+  is $t->tx->protocol, undef, 'no protocol';
+};
+
+subtest 'JSON roundtrips (with a lot of different tests)' => sub {
+  $t->websocket_ok('/json')->send_ok({json => {test => 23, snowman => '☃'}})
+    ->message_ok->json_message_is('' => {test => 24, snowman => '☃'})
+    ->json_message_is('' => {test => 24, snowman => '☃'})->json_message_has('/test')->json_message_hasnt('/test/2')
+    ->send_ok({binary => encode_json([1, 2, 3])}, 'with description')->message_ok('with description')
+    ->message_is('[1,2,3,4]')->message_is('[1,2,3,4]', 'with description')->message_isnt('[1,2,3]')
+    ->message_isnt('[1,2,3]', 'with description')->message_like(qr/3/)->message_like(qr/3/, 'with description')
+    ->message_unlike(qr/5/)->message_unlike(qr/5/, 'with description')->json_message_is([1, 2, 3, 4])
+    ->json_message_is([1, 2, 3, 4])->send_ok({binary => encode_json([1, 2, 3])})->message_ok->json_message_has('/2')
+    ->json_message_has('/2', 'with description')->json_message_hasnt('/5')
+    ->json_message_hasnt('/5', 'with description')->json_message_is('/2' => 3)
+    ->json_message_is('/2' => 3, 'with description')->send_ok({json => {'☃' => [1, 2, 3]}})
+    ->message_ok->json_message_is('/☃', [1, 2, 3])->json_message_like('/☃/1' => qr/\d/)
+    ->json_message_like('/☃/2' => qr/3/, 'with description')->json_message_unlike('/☃/1' => qr/[a-z]/)
+    ->json_message_unlike('/☃/2' => qr/2/, 'with description')->send_ok({json => 'works'})
+    ->message_ok->json_message_is('works')->send_ok({json => undef})->message_ok->json_message_is(undef)->finish_ok;
+};
+
+subtest 'Plain request' => sub {
+  $t->get_ok('/plain')->status_is(200)->content_is('Nothing to see here!');
+};
+
+subtest 'Server push' => sub {
+  $t->websocket_ok('/push')->message_ok->message_is('push')->message_ok->message_is('push')
+    ->message_ok->message_is('push')->finish_ok;
+  $t->websocket_ok('/push')->message_ok->message_unlike(qr/shift/)->message_ok->message_isnt('shift')
+    ->message_ok->message_like(qr/us/)->message_ok->message_unlike({binary => qr/push/})->finish_ok;
+};
+
+subtest 'Another plain request' => sub {
+  $t->get_ok('/plain')->status_is(200)->content_is('Nothing to see here!');
+};
+
+subtest 'Multiple roundtrips' => sub {
+  $t->websocket_ok('/echo')->send_ok('hello')->message_ok->message_is('echo: hello')->finish_ok;
+  $t->websocket_ok('/echo')->send_ok('this')->send_ok('just')->send_ok('works')->message_ok->message_is('echo: this')
+    ->message_ok->message_is('echo: just')->message_ok->message_is('echo: works')->message_like(qr/orks/)->finish_ok;
+};
+
+subtest 'Another plain request' => sub {
+  $t->get_ok('/plain')->status_is(200)->content_is('Nothing to see here!');
+};
+
+subtest 'Unicode roundtrips' => sub {
+  $t->websocket_ok('/unicode')->send_ok('hello')->message_ok->message_is('♥: hello')->finish_ok;
+  $t->websocket_ok('/unicode')->send_ok('hello again')->message_ok->message_is('♥: hello again')
+    ->send_ok('and one ☃ more time')->message_ok->message_is('♥: and one ☃ more time')->finish_ok;
+};
+
+subtest 'Binary frame and events' => sub {
+  my $bytes = b("I ♥ Mojolicious")->encode('UTF-16LE')->to_string;
+  $t->websocket_ok('/bytes');
+  my $binary;
+  $t->tx->on(
+    frame => sub {
+      my ($ws, $frame) = @_;
+      $binary++ if $frame->[4] == 2;
+    }
+  );
+  my $close;
+  $t->tx->on(finish => sub { shift; $close = [@_] });
+  $t->send_ok({binary => $bytes})->message_ok->message_is($bytes);
+  ok $binary, 'received binary frame';
+  $binary = undef;
+  $t->send_ok({text => $bytes})->message_ok->message_is($bytes);
+  ok !$binary, 'received text frame';
+  $t->finish_ok(1000 => 'Have a nice day!');
+  is_deeply $close, [1000, 'Have a nice day!'], 'right status and message';
+};
+
+subtest 'Binary roundtrips' => sub {
+  my $bytes = b("I ♥ Mojolicious")->encode('UTF-16LE')->to_string;
+  $t->request_ok($t->ua->build_websocket_tx('/bytes'))->send_ok({binary => $bytes})->message_ok->message_is($bytes)
+    ->send_ok({binary => $bytes})->message_ok->message_is($bytes)->finish_ok;
+};
+
+subtest 'Two responses' => sub {
+  $t->websocket_ok('/once')->send_ok('hello')->message_ok->message_is('ONE: hello')
+    ->message_ok->message_is('TWO: hello')->send_ok('hello')->message_ok->message_is('ONE: hello')->send_ok('hello')
+    ->message_ok->message_is('ONE: hello')->finish_ok;
+};
+
+subtest 'WebSocket connection gets closed right away' => sub {
+  $t->websocket_ok('/close')->finished_ok(1001);
+};
+
+subtest 'WebSocket connection gets closed after one message' => sub {
+  $t->websocket_ok('/one_sided')->message_ok->message_is('I ♥ Mojolicious!')->finished_ok(1005);
+};
+
+subtest 'Nested WebSocket' => sub {
+  $t->websocket_ok('/nested')->send_ok('hello')->message_ok->message_is('nested echo: hello')->finished_ok(1000);
+};
+
+subtest 'Test custom message' => sub {
+  $t->message([binary => 'foobarbaz'])->message_like(qr/bar/)->message_is({binary => 'foobarbaz'});
+};
+
+subtest 'Nested WebSocket with cookie' => sub {
+  $t->websocket_ok('/nested')->send_ok('hello')->message_ok->message_is('nested echo: helloagain')->finished_ok(1000);
+};
+
+subtest 'Nested plain request' => sub {
+  $t->get_ok('/nested')->status_is(200)->content_is('plain nested!');
+};
+
+subtest 'Another nested plain request' => sub {
+  $t->post_ok('/nested')->status_is(200)->content_is('plain nested too!');
+};
 
 done_testing();