Restore panicking behavior in test code

informalsystems · Jan 18, 2021 · f8a37a6 · f8a37a6
1 parent 4778810
commit f8a37a6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/rpc/src/client/transport/websocket.rs b/rpc/src/client/transport/websocket.rs
@@ -589,7 +589,10 @@ mod test {
             loop {
                 tokio::select! {
                     Some(ev) = self.event_rx.recv() => self.publish_event(ev),
-                    Ok((stream, _)) = self.listener.accept() => self.handle_incoming(stream).await,
+                    res = self.listener.accept() => {
+                        let (stream, _) = res.unwrap();
+                        self.handle_incoming(stream).await
+                    }
                     Some(res) = self.terminate_rx.recv() => {
                         self.terminate().await;
                         return res;