Move WS code outside the main into swiftstory.interface.ws.py

Also, the GameManager object is not created globally anymore when
importing the module.

Signed-off-by: Olivier Gayot <olivier.gayot@sigexec.com>

1 files changed, 61 insertions, 0 deletions

diff --git a/tests/test_ws_interface.py b/tests/test_ws_interface.py
new file mode 100644
index 0000000..6776334
--- /dev/null
+++ b/tests/test_ws_interface.py
@@ -0,0 +1,61 @@
+import unittest
+
+from swiftstory.interface.ws import WebsocketsInterface
+from swiftstory.status import error
+
+
+class TestWebsocketsInterface(unittest.TestCase):
+    def setUp(self):
+        # XXX Passing None as the GameManager. Only OK if we don't try to join
+        # a game.
+        self.interface = WebsocketsInterface(None)
+
+    def test_receive_invalid_json(self):
+        self.assertEqual(
+                error("badly formatted json"),
+                self.interface.message_received_handler(client=None, message="{invalid_json}")
+                )
+
+    def test_receive_json_array(self):
+        self.assertEqual(
+                error("invalid command"),
+                self.interface.message_received_handler(client=None, message='[]')
+                )
+
+    def test_receive_json_number(self):
+        self.assertEqual(
+                error("invalid command"),
+                self.interface.message_received_handler(client=None, message='2.3')
+                )
+
+    def test_receive_json_null(self):
+        self.assertEqual(
+                error("invalid command"),
+                self.interface.message_received_handler(client=None, message='null')
+                )
+
+    def test_receive_unknown_command(self):
+        self.assertEqual(
+                error("invalid command"),
+                self.interface.message_received_handler(client=None, message='{"op": "unknown"}')
+                )
+
+    def test_receive_without_command(self):
+        self.assertEqual(
+                error("invalid command"),
+                self.interface.message_received_handler(client=None, message='{}')
+                )
+
+    def test_play_card_not_specified(self):
+        payload = '{"op": "play_white_card"}'
+        self.assertEqual(
+                error("field `card_id' is required"),
+                self.interface.message_received_handler(client=None, message=payload)
+                )
+
+    def test_join_game_not_specified(self):
+        payload = '{"op": "join_game"}'
+        self.assertEqual(
+                error("field `game_name' is required"),
+                self.interface.message_received_handler(client=None, message=payload)
+                )