From b3410c68b8aaedb5998a4c9067bb23e7c25c59b9 Mon Sep 17 00:00:00 2001
From: meejah <meejah@meejah.ca>
Date: Tue, 23 Apr 2019 17:25:17 -0600
Subject: [PATCH] try some Autobahn websocket tests

---
 src/allmydata/test/test_websocket_logs.py | 63 +++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 src/allmydata/test/test_websocket_logs.py

diff --git a/src/allmydata/test/test_websocket_logs.py b/src/allmydata/test/test_websocket_logs.py
new file mode 100644
index 000000000..2dab3c22d
--- /dev/null
+++ b/src/allmydata/test/test_websocket_logs.py
@@ -0,0 +1,63 @@
+import json
+
+from twisted.trial import unittest
+from twisted.internet.defer import inlineCallbacks, Deferred
+
+from eliot import log_call
+
+from autobahn.twisted.testing import create_memory_agent, MemoryReactorClockResolver
+from autobahn.twisted.websocket import WebSocketServerProtocol
+from autobahn.twisted.websocket import WebSocketClientProtocol
+
+from allmydata.web.logs import TokenAuthenticatedWebSocketServerProtocol
+#_StreamingLogClientProtocol
+
+class TestStreamingLogs(unittest.TestCase):
+    """
+    Test websocket streaming of logs
+
+    Note: depends on un-merged Autobahn branch
+    """
+
+    def setUp(self):
+        self.reactor = MemoryReactorClockResolver()
+        self.agent = create_memory_agent(self.reactor, TokenAuthenticatedWebSocketServerProtocol)
+
+    @inlineCallbacks
+    def test_one_log(self):
+
+        class TestClient(WebSocketClientProtocol):
+            def onOpen(self):
+                print("opened")
+
+            def onMessage(self, msg, is_binary=False):
+                print("msg {}".format(msg))
+
+            def onClose(self, *args):
+                print("close: {}".format(args))
+
+        proto = yield self.agent.open(
+            transport_config=u"ws://localhost:1234/ws",
+            options={},
+            protocol_class=None,
+        )
+
+        messages = []
+        def got_message(msg, is_binary=False):
+            messages.append(json.loads(msg))
+        proto.on("message", got_message)
+
+
+        @log_call(action_type=u"test:cli:magic-folder:cleanup")
+        def do_a_thing():
+            pass
+
+        do_a_thing()
+
+        proto.transport.loseConnection()
+        self.agent.flush()
+        yield proto.is_closed
+
+        self.assertEqual(len(messages), 2)
+        self.assertEqual("started", messages[0]["action_status"])
+        self.assertEqual("succeeded", messages[1]["action_status"])