diff --git a/pydantic_tfl_api/models/disruption.py b/pydantic_tfl_api/models/disruption.py index 424fd37..cb67f33 100644 --- 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 index 682c866..9da0f0d 100644 --- 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": {