Change test API to allow multiple pathoc requests

Add simple unit test for websocket server

2 files changed, 9 insertions, 2 deletions

diff --git a/libpathod/pathod.py b/libpathod/pathod.py
index 56348fbb..c5ad942a 100644
--- a/libpathod/pathod.py
+++ b/libpathod/pathod.py
@@ -103,8 +103,15 @@ class PathodHandler(tcp.BaseHandler):
             while True:
                 try:
                     frm = websockets.Frame.from_file(self.rfile)
+                    retlog = dict(
+                        type="wsframe",
+                        frame=dict(
+                        ),
+                        cipher=None,
+                    )
+                    self.addlog(retlog)
                     break
-                except tcp.NetLibTimeout:
+                except tcp.NetLibTimeout: # pragma: no cover
                     pass
             lg(frm.human_readable())
         return self.handle_websocket, None
diff --git a/libpathod/test.py b/libpathod/test.py
index afd41f81..596fea9c 100644
--- a/libpathod/test.py
+++ b/libpathod/test.py
@@ -48,7 +48,7 @@ class Daemon:
         l = self.log()
         if not l:
             return None
-        return l[-1]
+        return l[0]
 
     def log(self):
         """