diff options
| -rw-r--r-- | test/js/websockets_hello.ml | 17 |
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) |