Add parameterized test for pydantic v1/v2 default value handling

Author: koxudaxi

tests/data/expected/main/jsonschema/has_default_value_pydantic_v2.py

@@ -0,0 +1,46 @@
+# generated by datamodel-codegen:
+#   filename:  has_default_value.json
+#   timestamp: 2019-07-26T00:00:00+00:00
+
+from __future__ import annotations
+
+from enum import Enum
+
+from pydantic import BaseModel, Field, RootModel
+
+
+class TeamType(Enum):
+    Department = 'Department'
+    Division = 'Division'
+    BusinessUnit = 'BusinessUnit'
+    Organization = 'Organization'
+
+
+class ID(RootModel[str]):
+    root: str
+
+
+class Pet(BaseModel):
+    name: str | None = None
+
+
+class Family(RootModel[list[ID]]):
+    root: list[ID] = Field(
+        default_factory=lambda: [ID.model_validate(v) for v in ['abc', 'efg']]
+    )
+
+
+class FamilyPets(RootModel[list[Pet]]):
+    root: list[Pet] = Field(
+        default_factory=lambda: [Pet.model_validate(v) for v in ['taro', 'shiro']]
+    )
+
+
+class Person(BaseModel):
+    id: ID | None = Field(default_factory=lambda: ID('abc'))
+    user: Pet | None = None
+    firstName: str | None = Field(None, description="The person's first name.")
+    team: TeamType | None = 'Department'
+    anotherTeam: TeamType | None = 'Department'
+    Family_1: Family | None = Field(None, alias='Family')
+    FamilyPets_1: FamilyPets | None = Field(None, alias='FamilyPets')

tests/main/jsonschema/test_main_jsonschema.py

@@ -3254,14 +3254,22 @@ def test_jsonschema_use_title_as_name_nested_titles_pydantic(output_file: Path)
     )
 
 
-def test_main_jsonschema_has_default_value(output_file: Path) -> None:
+@pytest.mark.parametrize(
+    ("output_model", "expected_file"),
+    [
+        ("pydantic.BaseModel", "has_default_value.py"),
+        ("pydantic_v2.BaseModel", "has_default_value_pydantic_v2.py"),
+    ],
+)
+def test_main_jsonschema_has_default_value(output_model: str, expected_file: str, output_file: Path) -> None:
     """Test default value handling."""
     run_main_and_assert(
         input_path=JSON_SCHEMA_DATA_PATH / "has_default_value.json",
         output_path=output_file,
         input_file_type="jsonschema",
         assert_func=assert_file_content,
-        expected_file="has_default_value.py",
+        expected_file=expected_file,
+        extra_args=["--output-model-type", output_model],
     )