chore: Update Disruption model fields to be optional

Make the 'summary', 'additional_info', 'created', and 'last_update' fields in the Disruption model optional by setting their default value to None. This allows for more flexibility when creating instances of the Disruption class and handling data that may not always be available.
mnbf9rca · Jul 23, 2024 · 468fe35 · 468fe35
1 parent d41e815
commit 468fe35
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/pydantic_tfl_api/models/disruption.py b/pydantic_tfl_api/models/disruption.py
@@ -11,10 +11,10 @@ class Disruption(BaseModel):
     disruption_type: str = Field(alias='type')
     category_description: str = Field(alias='categoryDescription')
     description: str = Field(alias='description')
-    summary: str = Field(alias='summary')
-    additional_info: str = Field(alias='additionalInfo')
-    created: datetime = Field(alias='created')
-    last_update: datetime = Field(alias='lastUpdate')
+    summary: Optional[str] = Field(None, alias='summary')
+    additional_info: Optional[str] = Field(None, alias='additionalInfo')
+    created: Optional[datetime] = Field(None, alias='created')
+    last_update: Optional[datetime] = Field(None, alias='lastUpdate')
     affected_routes: List[AffectedRoute] = Field(alias='affectedRoutes')
     affected_stops: List[StopPoint] = Field(alias='affectedStops')
     closure_text: str = Field(alias='closureText')

diff --git a/tests/config_for_tests.py b/tests/config_for_tests.py
@@ -157,35 +157,35 @@
         "endpoint": "lineDisruptionsByLineId",
         "endpoint_args": ["victoria,northern"],
         "endpoint_params": {},
-        "model": "Line",
+        "model": "Disruption",
         "result_is_empty": True,
     },
     "lineDisruptionsByLineId_piccadilly_None_Line": {
         "endpoint": "lineDisruptionsByLineId",
         "endpoint_args": ["piccadilly"],
         "endpoint_params": {},
-        "model": "Line",
+        "model": "Disruption",
         "result_is_empty": False,
     },    
     "lineDisruptionsByLineId_14_None_Line": {
         "endpoint": "lineDisruptionsByLineId",
         "endpoint_args": ["14"],
         "endpoint_params": {},
-        "model": "Line",
+        "model": "Disruption",
         "result_is_empty": True,
     },
     "lineDisruptionsByMode_tube_None_Line": {
         "endpoint": "lineDisruptionsByMode",
         "endpoint_args": ["tube"],
         "endpoint_params": {},
-        "model": "Line",
+        "model": "Disruption",
         "result_is_empty": False,
     },
     "lineDisruptionsByMode_tube_overground_None_Line": {
         "endpoint": "lineDisruptionsByMode",
         "endpoint_args": ["tube,overground"],
         "endpoint_params": {},
-        "model": "Line",
+        "model": "Disruption",
         "result_is_empty": False,
     },
     "stopPointById_940GZZLUASL_None_StopPoint": {