fix(stream): remove start function

1 files changed, 5 insertions, 2 deletions

diff --git a/test/hello.ml b/test/hello.ml
index b397727..a00eb2e 100644
--- a/test/hello.ml
+++ b/test/hello.ml
@@ -4,11 +4,13 @@ open! Flesh
 
 let program (stream, push) config =
   let+ _auth = Sasl.authenticate (stream, push) config
-  in match _auth with
+  in begin match _auth with
      | Error (NotAuthorized, Some (_, text)) -> print_endline ("Not authorized: " ^ text)
      | Error (MalformedRequest, Some (_, text)) -> print_endline ("Malformed request: " ^ text)
      | Error _ -> print_endline "Error!"
      | Ok _ -> print_endline "Success!"
+     end;
+     Portal.close (stream, push)
 
 let main =
   let config : Sasl.auth_config = {
@@ -17,7 +19,8 @@ let main =
       preferred_mechanisms = [Stream.PLAIN]
     }
   in let domain = (List.nth (String.split_on_char '@' config.jid) 1) in
-     let* stream, push = Stream.start domain in
+     let* stream, push = Portal.connect domain in
+     let* _id = Portal.header domain (stream, push) in
      Lwt.catch
        (fun () -> program (stream, push) config >|= (fun () -> push None))
        (fun exn ->