feat(test): catch lwt exceptions

1 files changed, 10 insertions, 7 deletions

diff --git a/test/js/websockets_hello.ml b/test/js/websockets_hello.ml
index 8bc7487..8af5a5e 100644
--- a/test/js/websockets_hello.ml
+++ b/test/js/websockets_hello.ml
@@ -16,14 +16,17 @@ let rec run t =
         then next_tick (fun () -> run t)
         else ()
 
+let main (stream, push) =
+  let* _stream = Lwt_stream.get stream
+  in let+ _auth = Auth.send_auth_stanza (stream, push)
+                    "test@example.com" "password"
+                    Auth.PLAIN
+     in push None
+
 let () =
   run @@
     let domain = "egregore.fun" in
     let* stream, push = Stream.start domain
-    in let* _stream = Lwt_stream.get stream
-       in let* _auth = Auth.send_auth_stanza (stream, push)
-                        "test@example.com" "password"
-                        Auth.PLAIN
-          in push (Some "<iq>wrong</iq>");
-             let+ _recv = Lwt_stream.get stream in
-             push None
+    in Lwt.catch
+         (fun () -> main (stream, push))
+         (fun exn -> push None; Lwt.fail exn)