Support Raw in convert

Previously `Raw` inputs weren't supported in `convert`, now they are.
jcrist · Oct 20, 2024 · 74ebfe0 · 74ebfe0
1 parent c80616a
commit 74ebfe0
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/msgspec/_core.c b/msgspec/_core.c
@@ -20560,6 +20560,17 @@ convert_immutable(
     return ms_validation_error(expected, type, path);
 }
 
+static PyObject *
+convert_raw(
+    ConvertState *self, PyObject *obj, TypeNode *type, PathNode *path
+) {
+    if (type->types == 0) {
+        Py_INCREF(obj);
+        return obj;
+    }
+    return ms_validation_error("raw", type, path);
+}
+
 static PyObject *
 convert_seq_to_list(
     ConvertState *self, PyObject **items, Py_ssize_t size,
@@ -21670,6 +21681,9 @@ convert(
     else if (pytype == &Ext_Type) {
         return convert_immutable(self, MS_TYPE_EXT, "ext", obj, type, path);
     }
+    else if (pytype == &Raw_Type) {
+        return convert_raw(self, obj, type, path);
+    }
     else if (PyAnySet_Check(obj)) {
         return convert_any_set(self, obj, type, path);
     }

diff --git a/tests/test_convert.py b/tests/test_convert.py
@@ -2600,3 +2600,14 @@ def dec_hook(typ, x):
 
         assert rec.value.__cause__ is rec.value.__context__
         assert type(rec.value.__cause__) is TypeError
+
+
+class TestRaw:
+    def test_raw(self):
+        raw = msgspec.Raw(b"123")
+
+        class Ex(Struct):
+            x: msgspec.Raw
+
+        sol = Ex(x=raw)
+        assert convert({"x": raw}, type=Ex) == sol