From c54514b283c47f774b0b6e9e22449ee6876ffc51 Mon Sep 17 00:00:00 2001 From: Olivier Gayot Date: Sun, 13 Sep 2020 00:16:32 +0200 Subject: Handle exception when JSON received is not an object Signed-off-by: Olivier Gayot --- tests/test_swiftstory.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'tests') 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"), -- cgit v1.2.3