feat: add ws endpoint detection function

1 files changed, 4 insertions, 4 deletions

diff --git a/test/js/websockets_hello.ml b/test/js/websockets_hello.ml
index 41487b7..6b74d96 100644
--- a/test/js/websockets_hello.ml
+++ b/test/js/websockets_hello.ml
@@ -2,19 +2,19 @@ open Lwt.Syntax
 
 let () =
   (* Echo is a websocket that... echoes you stuff. *)
-  let stream, push = Portal_ws.lwt_ws "wss://echo.websocket.org" in
+  let stream, push = Portal_ws.ws_stream "wss://echo.websocket.org" in
   push (Some "great text");
   push (Some "other text");
   push (Some "yet another text");
   push (Some "BYE");
   Lwt.async @@
     fun () ->
-    let+ s = Lwt_stream.iter
+    let* server = Portal_ws.ws_endpoint "egregore.fun"
+    in let+ _ = Lwt_stream.iter
                (fun greetings ->
                  match greetings with
                  (* When the websocket sends "BYE", we close. *)
                  | "BYE" -> print_endline "CLOSING BYE"; push None
                  | hello -> print_endline ("> " ^ hello))
                stream
-    in s
-
+    in print_endline server