wait for the connection

mojolicious · Dec 9, 2014 · 10b35cd · 10b35cd
1 parent 092b5da
commit 10b35cd
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/t/mojo/websocket.t b/t/mojo/websocket.t
@@ -459,10 +459,12 @@ $ua->websocket(
     $tx->on(
       frame => sub {
         my ($tx, $frame) = @_;
-        $pong = $frame->[5] if $frame->[4] == 10;
-        Mojo::IOLoop->stop;
+        return unless $frame->[4] == 10;
+        $pong = $frame->[5];
+        $tx->finish;
       }
     );
+    $tx->on(finish => sub { Mojo::IOLoop->stop });
     $tx->send([1, 0, 0, 0, 9, 'test']);
   }
 );