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)