dify

dify/api/tests/unit_tests/libs/test_orjson.py

@@ -0,0 +1,25 @@
+import orjson
+import pytest
+
+from libs.orjson import orjson_dumps
+
+
+def test_orjson_dumps_round_trip_basic():
+    obj = {"a": 1, "b": [1, 2, 3], "c": {"d": True}}
+    s = orjson_dumps(obj)
+    assert orjson.loads(s) == obj
+
+
+def test_orjson_dumps_with_unicode_and_indent():
+    obj = {"msg": "你好，Dify"}
+    s = orjson_dumps(obj, option=orjson.OPT_INDENT_2)
+    # contains indentation newline/spaces
+    assert "\n" in s
+    assert orjson.loads(s) == obj
+
+
+def test_orjson_dumps_non_utf8_encoding_fails():
+    obj = {"msg": "你好"}
+    # orjson.dumps() always produces UTF-8 bytes; decoding with non-UTF8 fails.
+    with pytest.raises(UnicodeDecodeError):
+        orjson_dumps(obj, encoding="ascii")