Add unit tests for swiftstory.Status

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

1 files changed, 21 insertions, 0 deletions

diff --git a/tests/test_status.py b/tests/test_status.py
new file mode 100644
index 0000000..5c48cd0
--- /dev/null
+++ b/tests/test_status.py
@@ -0,0 +1,21 @@
+import json
+import unittest
+
+import swiftstory.Status as Status
+
+
+class TestStatus(unittest.TestCase):
+    def test_error(self):
+        error_text = "Some error with utf-8 characters ùðú!"
+        error_code = 520
+        output = json.loads(Status.error(error_text, error_code))
+        self.assertEqual(output["type"], "response")
+        self.assertEqual(output["content"]["info"], error_text)
+        self.assertEqual(output["content"]["status"], error_code)
+
+    def test_success(self):
+        text = "Some message with utf-8 characters ùðú!"
+        output = json.loads(Status.success(text))
+        self.assertEqual(output["type"], "response")
+        self.assertEqual(output["content"]["status"], 0)
+        self.assertEqual(output["content"]["result"], text)