test compression edge case

mojolicious · Feb 2, 2014 · eb127ea · eb127ea
1 parent 5ebab41
commit eb127ea
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/lib/Test/Mojo.pm b/lib/Test/Mojo.pm
@@ -289,13 +289,13 @@ sub _build_ok {
 sub _json {
   my ($self, $method, $p) = @_;
   return Mojo::JSON::Pointer->new->$method(
-    Mojo::JSON->new->decode(@{$self->message}[1]), $p);
+    Mojo::JSON->new->decode(@{$self->message // []}[1]), $p);
 }
 
 sub _message {
   my ($self, $name, $value, $desc) = @_;
   local $Test::Builder::Level = $Test::Builder::Level + 1;
-  my ($type, $msg) = @{$self->message};
+  my ($type, $msg) = @{$self->message // []};
 
   # Type check
   if (ref $value eq 'HASH') {

diff --git a/t/mojolicious/websocket_lite_app.t b/t/mojolicious/websocket_lite_app.t
@@ -191,6 +191,13 @@ $t->websocket_ok(
   ->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 .. 262144;
+$t->websocket_ok(
+  '/echo' => {'Sec-WebSocket-Extensions' => 'permessage-deflate'})
+  ->send_ok({binary => $huge})->message_ok->message_is({binary => $huge})
+  ->finish_ok;
+
 # JSON roundtrips
 $t->websocket_ok('/json')->send_ok({json => {test => 23, snowman => '☃'}})
   ->message_ok->json_message_is('' => {test => 24, snowman => '☃'})