Handle exception when JSON received is not an object

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

1 files changed, 18 insertions, 0 deletions

diff --git a/tests/test_swiftstory.py b/tests/test_swiftstory.py
index 6a1f639..51dc837 100644
--- a/tests/test_swiftstory.py
+++ b/tests/test_swiftstory.py
@@ -11,6 +11,24 @@ class TestSwiftStory(unittest.TestCase):
                 SwiftStory.message_received_handler(client=None, message="{invalid_json}")
                 )
 
+    def test_receive_json_array(self):
+        self.assertEqual(
+                error("invalid command"),
+                SwiftStory.message_received_handler(client=None, message='[]')
+                )
+
+    def test_receive_json_number(self):
+        self.assertEqual(
+                error("invalid command"),
+                SwiftStory.message_received_handler(client=None, message='2.3')
+                )
+
+    def test_receive_json_null(self):
+        self.assertEqual(
+                error("invalid command"),
+                SwiftStory.message_received_handler(client=None, message='null')
+                )
+
     def test_receive_unknown_command(self):
         self.assertEqual(
                 error("invalid command"),