import unittest

from flask import Flask

from app.utils.response import success


class ResponseMessageTest(unittest.TestCase):
    def test_success_message_defaults_to_chinese(self):
        app = Flask(__name__)

        with app.app_context():
            response, status_code = success({"ok": True})

        self.assertEqual(status_code, 200)
        self.assertEqual(response.get_json()["message"], "操作成功")


if __name__ == "__main__":
    unittest.main()