fix functional tests for batch

Author: sthulb

tests/functional/batch/sample_models.py

@@ -38,7 +38,7 @@ class OrderDynamoDB(BaseModel):
     @field_validator("Message", mode="before")
     def transform_message_to_dict(cls, value: Dict[Literal["S"], str]):
         try:
-            return json.loads(value["S"])
+            return json.loads(value)
         except TypeError:
             raise ValueError
 

tests/functional/test_utilities_batch.py

@@ -526,7 +526,7 @@ class OrderDynamoDB(BaseModel):
         # so Pydantic can auto-initialize nested Order model
         @field_validator("Message", mode="before")
         def transform_message_to_dict(cls, value: Dict[Literal["S"], str]):
-            return json.loads(value["S"])
+            return json.loads(value)
 
     class OrderDynamoDBChangeRecord(DynamoDBStreamChangedRecordModel):
         NewImage: Optional[OrderDynamoDB] = None
@@ -570,7 +570,7 @@ class OrderDynamoDB(BaseModel):
         # so Pydantic can auto-initialize nested Order model
         @field_validator("Message", mode="before")
         def transform_message_to_dict(cls, value: Dict[Literal["S"], str]):
-            return json.loads(value["S"])
+            return json.loads(value)
 
     class OrderDynamoDBChangeRecord(DynamoDBStreamChangedRecordModel):
         NewImage: Optional[OrderDynamoDB] = None