Add more tests for invalid commands

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

1 files changed, 13 insertions, 1 deletions

diff --git a/tests/test_swiftstory.py b/tests/test_swiftstory.py
index c2780d5..6a1f639 100644
--- a/tests/test_swiftstory.py
+++ b/tests/test_swiftstory.py
@@ -5,8 +5,20 @@ from swiftstory.Status import error
 
 
 class TestSwiftStory(unittest.TestCase):
-    def test_invalid_message_received(self):
+    def test_receive_invalid_json(self):
         self.assertEqual(
                 error("badly formatted json"),
                 SwiftStory.message_received_handler(client=None, message="{invalid_json}")
                 )
+
+    def test_receive_unknown_command(self):
+        self.assertEqual(
+                error("invalid command"),
+                SwiftStory.message_received_handler(client=None, message='{"op": "unknown"}')
+                )
+
+    def test_receive_without_command(self):
+        self.assertEqual(
+                error("invalid command"),
+                SwiftStory.message_received_handler(client=None, message='{}')
+                )