Add parameterized e2e test for class-decorators across all output types

Author: koxudaxi

tests/data/expected/main/class_decorators_dataclasses_dataclass.py

@@ -0,0 +1,17 @@
+# generated by datamodel-codegen:
+#   filename:  simple_frozen_test.json
+#   timestamp: 1985-10-26T08:21:00+00:00
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+from my_module import my_decorator
+
+
+@my_decorator
+@dataclass
+class User:
+    name: str
+    age: int
+    email: str | None = None

tests/data/expected/main/class_decorators_msgspec_Struct.py

@@ -0,0 +1,15 @@
+# generated by datamodel-codegen:
+#   filename:  simple_frozen_test.json
+#   timestamp: 1985-10-26T08:21:00+00:00
+
+from __future__ import annotations
+
+from msgspec import UNSET, Struct, UnsetType
+from my_module import my_decorator
+
+
+@my_decorator
+class User(Struct):
+    name: str
+    age: int
+    email: str | UnsetType = UNSET

tests/data/expected/main/class_decorators_pydantic_BaseModel.py

@@ -0,0 +1,15 @@
+# generated by datamodel-codegen:
+#   filename:  simple_frozen_test.json
+#   timestamp: 1985-10-26T08:21:00+00:00
+
+from __future__ import annotations
+
+from my_module import my_decorator
+from pydantic import BaseModel, EmailStr
+
+
+@my_decorator
+class User(BaseModel):
+    name: str
+    age: int
+    email: EmailStr | None = None

tests/data/expected/main/class_decorators_pydantic_v2_BaseModel.py

@@ -0,0 +1,15 @@
+# generated by datamodel-codegen:
+#   filename:  simple_frozen_test.json
+#   timestamp: 1985-10-26T08:21:00+00:00
+
+from __future__ import annotations
+
+from my_module import my_decorator
+from pydantic import BaseModel, EmailStr
+
+
+@my_decorator
+class User(BaseModel):
+    name: str
+    age: int
+    email: EmailStr | None = None

tests/data/expected/main/class_decorators_pydantic_v2_dataclass.py

@@ -0,0 +1,17 @@
+# generated by datamodel-codegen:
+#   filename:  simple_frozen_test.json
+#   timestamp: 1985-10-26T08:21:00+00:00
+
+from __future__ import annotations
+
+from my_module import my_decorator
+from pydantic import EmailStr
+from pydantic.dataclasses import dataclass
+
+
+@my_decorator
+@dataclass
+class User:
+    name: str
+    age: int
+    email: EmailStr | None = None

tests/main/test_main_general.py

@@ -284,6 +284,46 @@ def test_class_decorators_with_empty_entries(output_file: Path) -> None:
     )
 
 
+@freeze_time(TIMESTAMP)
+@pytest.mark.parametrize(
+    ("output_model_type", "expected_file"),
+    [
+        ("pydantic.BaseModel", "class_decorators_pydantic_BaseModel.py"),
+        ("pydantic_v2.BaseModel", "class_decorators_pydantic_v2_BaseModel.py"),
+        ("pydantic_v2.dataclass", "class_decorators_pydantic_v2_dataclass.py"),
+        ("dataclasses.dataclass", "class_decorators_dataclasses_dataclass.py"),
+        ("msgspec.Struct", "class_decorators_msgspec_Struct.py"),
+        # Note: TypedDict is excluded because its template doesn't support decorators
+    ],
+    ids=[
+        "pydantic_v1",
+        "pydantic_v2",
+        "pydantic_v2_dataclass",
+        "dataclasses",
+        "msgspec",
+    ],
+)
+def test_class_decorators_all_output_types(
+    output_file: Path, output_model_type: str, expected_file: str
+) -> None:
+    """Test --class-decorators works with all output model types that support decorators."""
+    run_main_and_assert(
+        input_path=DATA_PATH / "jsonschema" / "simple_frozen_test.json",
+        output_path=output_file,
+        input_file_type="jsonschema",
+        assert_func=assert_file_content,
+        expected_file=expected_file,
+        extra_args=[
+            "--output-model-type",
+            output_model_type,
+            "--class-decorators",
+            "@my_decorator",
+            "--additional-imports",
+            "my_module.my_decorator",
+        ],
+    )
+
+
 @freeze_time(TIMESTAMP)
 def test_use_attribute_docstrings(tmp_path: Path) -> None:
     """Test --use-attribute-docstrings flag functionality."""