add runtime type for wrapped records

Author: bentsku

src/py_avro_schema/_schemas.py

@@ -320,6 +320,9 @@ def _wrap_as_record(self, inner_schema: JSONObj, names: NamesType) -> JSONType:
                 {"name": REF_DATA_KEY, "type": inner_schema},
             ],
         }
+        if Option.ADD_RUNTIME_TYPE_FIELD in self.options:
+            record_schema["fields"].append({"name": RUNTIME_TYPE_KEY, "type": ["null", "string"]})
+
         if self.namespace:
             record_schema["namespace"] = self.namespace
         return record_schema

tests/test_avro_schema.py

@@ -80,3 +80,17 @@ class PyType:
         ],
     }
     assert_schema(PyType, expected, options=pas.Option.ADD_RUNTIME_TYPE_FIELD)
+
+
+def test_add_type_field_on_wrapped_record():
+    py_type = list[str]
+    expected = {
+        "type": "record",
+        "name": "StrList",
+        "fields": [
+            {"name": "__id", "type": ["null", "long"], "default": None},
+            {"name": "__data", "type": {"type": "array", "items": "string"}},
+            {"name": "_runtime_type", "type": ["null", "string"]},
+        ],
+    }
+    assert_schema(py_type, expected, options=pas.Option.WRAP_INTO_RECORDS | pas.Option.ADD_RUNTIME_TYPE_FIELD)