将时间类型修改为datetime，然后通过自定义json_decoder来确保生成的时间格式正确性

Author: asiazhang

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "testsolar-testtool-sdk"
-version = "0.1"
+version = "0.1.2"
 description = "TestSolar测试工具SDK"
 authors = [
     {name = "asiazhang", email = "asiazhang2002@gmail.com"},

src/testsolar_testtool_sdk/model/testresult.py

@@ -1,5 +1,6 @@
 from enum import Enum
 from typing import List, Optional
+from datetime import datetime
 
 from pydantic import BaseModel, Field
 
@@ -54,29 +55,48 @@ class Attachment(BaseModel):
 class TestCaseLog(BaseModel):
     __test__ = False
 
-    time: str = Field(alias="Time")
+    time: datetime = Field(alias="Time")
     level: LogLevel = Field(alias="Level")
     content: str = Field(alias="Content")
     attachments: List[Attachment] = Field(alias="Attachments")
     assert_error: TestCaseAssertError = Field(alias="AssertError")
     runtime_error: Optional[TestCaseRuntimeError] = Field(None, alias="RuntimeError")
 
+    class Config:
+        json_encoders = {
+            datetime: lambda dt: _format_datetime(dt),
+        }
+
 
 class TestCaseStep(BaseModel):
     __test__ = False
 
-    start_time: str = Field(alias="StartTime")
-    end_time: Optional[str] = Field(alias="EndTime")
+    start_time: datetime = Field(alias="StartTime")
+    end_time: Optional[datetime] = Field(alias="EndTime")
     title: str = Field(alias="Title")
     logs: List[TestCaseLog] = Field(alias="Logs")
 
+    class Config:
+        json_encoders = {
+            datetime: lambda dt: _format_datetime(dt),
+        }
+
+
+def _format_datetime(t: datetime) -> str:
+    return t.strftime("%Y-%m-%dT%H:%M:%S.%f")[:-3] + "Z"
+
 
 class TestResult(BaseModel):
     __test__ = False
 
     test: TestCase = Field(alias="Test")
-    start_time: str = Field(alias="StartTime")
-    end_time: Optional[str] = Field(alias="EndTime")
+    start_time: datetime = Field(alias="StartTime")
+    end_time: Optional[datetime] = Field(alias="EndTime")
     result_type: ResultType = Field(alias="ResultType")
     message: str = Field(alias="Message")
     steps: List[TestCaseStep] = Field(alias="Steps")
+
+    class Config:
+        json_encoders = {
+            datetime: lambda dt: _format_datetime(dt),
+        }

tests/test_report.py

@@ -1,10 +1,11 @@
 import concurrent.futures
 import io
+import json
 import logging
 import random
 from datetime import datetime, timedelta
-from typing import BinaryIO
 from functools import partial
+from typing import BinaryIO
 
 from src.testsolar_testtool_sdk.model.load import LoadResult, LoadError
 from src.testsolar_testtool_sdk.model.test import TestCase
@@ -22,10 +23,6 @@
 logger.setLevel(logging.DEBUG)
 
 
-def _format_datetime(t: datetime) -> str:
-    return t.strftime("%Y-%m-%dT%H:%M:%S.%f")[:-3] + "Z"
-
-
 def generate_demo_load_result() -> LoadResult:
     r: LoadResult = LoadResult(Tests=[], LoadErrors=[])
 
@@ -62,8 +59,8 @@ def generate_test_result(index: int) -> TestResult:
     start: datetime = datetime.utcnow() - timedelta(seconds=40)
     _tr = TestResult(
         Test=TestCase(Name=f"mumu/mu.py/test_case_name_{index}_p1", Attributes={}),
-        StartTime=_format_datetime(start),
-        EndTime=_format_datetime(datetime.utcnow()),
+        StartTime=start,
+        EndTime=datetime.utcnow(),
         ResultType=ResultType.SUCCEED,
         Message="ファイルが見つかりません。ファイルパスを確認して、もう一度試してください。",
         Steps=[generate_testcase_step(f"{index}_{x}") for x in range(10)],
@@ -76,7 +73,7 @@ def generate_testcase_log(index: str) -> TestCaseLog:
     start: datetime = datetime.utcnow() - timedelta(seconds=15)
 
     return TestCaseLog(
-        Time=_format_datetime(start),
+        Time=start,
         Level=LogLevel.INFO,
         Attachments=[
             # Attachment(
@@ -95,8 +92,8 @@ def generate_testcase_log(index: str) -> TestCaseLog:
 def generate_testcase_step(index: str) -> TestCaseStep:
     start: datetime = datetime.utcnow() - timedelta(seconds=10)
     return TestCaseStep(
-        StartTime=_format_datetime(start),
-        EndTime=_format_datetime(datetime.utcnow()),
+        StartTime=start,
+        EndTime=datetime.utcnow(),
         Title=get_random_unicode(100),
         Logs=[generate_testcase_log(f"{index}_{x}") for x in range(100)],
     )
@@ -156,6 +153,19 @@ def send_test_result(pipe_io: BinaryIO):
     reporter.report_run_case_result(run_case_result)
 
 
+def test_datetime_formatted():
+    run_case_result = generate_test_result(0)
+    data = run_case_result.model_dump_json(by_alias=True, indent=2)
+    tr = json.loads(data)
+    assert tr['StartTime'].endswith("Z")
+    assert tr['EndTime'].endswith("Z")
+
+    assert tr['Steps'][0]['StartTime'].endswith("Z")
+    assert tr['Steps'][0]['EndTime'].endswith("Z")
+
+    assert tr['Steps'][0]['Logs'][0]['Time'].endswith("Z")
+
+
 def test_report_run_case_result():
     # 创建一个Reporter实例
     pipe_io = io.BytesIO()