3e8d3083ac
* Refactor NextBus integration to use new API This removes the `messages`, `directions`, and `attribution` attributes from the sensor. Those may be added back in the future with additional refactoring. Some existing sensors may be broken today because of deprecated Agency names. This patch will not migrate them as the migration path is ambiguous. Setting up again should work though. * Move result indexing outside of try/except
116 lines
3.1 KiB
Python
116 lines
3.1 KiB
Python
"""Test helpers for NextBus tests."""
|
|
|
|
from typing import Any
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture(
|
|
params=[
|
|
[
|
|
{
|
|
"name": "Outbound",
|
|
"shortName": "Outbound",
|
|
"useForUi": True,
|
|
"stops": ["5184"],
|
|
},
|
|
{
|
|
"name": "Outbound - Hidden",
|
|
"shortName": "Outbound - Hidden",
|
|
"useForUi": False,
|
|
"stops": ["5651"],
|
|
},
|
|
],
|
|
[
|
|
{
|
|
"name": "Outbound",
|
|
"shortName": "Outbound",
|
|
"useForUi": True,
|
|
"stops": ["5184"],
|
|
},
|
|
{
|
|
"name": "Inbound",
|
|
"shortName": "Inbound",
|
|
"useForUi": True,
|
|
"stops": ["5651"],
|
|
},
|
|
],
|
|
]
|
|
)
|
|
def route_config_direction(request: pytest.FixtureRequest) -> Any:
|
|
"""Generate alternative directions values.
|
|
|
|
When only on edirection is returned, it is not returned as a list, but instead an object.
|
|
"""
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_nextbus_lists(
|
|
mock_nextbus: MagicMock, route_config_direction: Any
|
|
) -> MagicMock:
|
|
"""Mock all list functions in nextbus to test validate logic."""
|
|
instance = mock_nextbus.return_value
|
|
instance.agencies.return_value = [
|
|
{
|
|
"id": "sfmta-cis",
|
|
"name": "San Francisco Muni CIS",
|
|
"shortName": "SF Muni CIS",
|
|
"region": "",
|
|
"website": "",
|
|
"logo": "",
|
|
"nxbs2RedirectUrl": "",
|
|
}
|
|
]
|
|
|
|
instance.routes.return_value = [
|
|
{
|
|
"id": "F",
|
|
"rev": 1057,
|
|
"title": "F Market & Wharves",
|
|
"description": "7am-10pm daily",
|
|
"color": "",
|
|
"textColor": "",
|
|
"hidden": False,
|
|
"timestamp": "2024-06-23T03:06:58Z",
|
|
},
|
|
]
|
|
|
|
instance.route_details.return_value = {
|
|
"id": "F",
|
|
"rev": 1057,
|
|
"title": "F Market & Wharves",
|
|
"description": "7am-10pm daily",
|
|
"color": "",
|
|
"textColor": "",
|
|
"hidden": False,
|
|
"boundingBox": {},
|
|
"stops": [
|
|
{
|
|
"id": "5184",
|
|
"lat": 37.8071299,
|
|
"lon": -122.41732,
|
|
"name": "Jones St & Beach St",
|
|
"code": "15184",
|
|
"hidden": False,
|
|
"showDestinationSelector": True,
|
|
"directions": ["F_0_var1", "F_0_var0"],
|
|
},
|
|
{
|
|
"id": "5651",
|
|
"lat": 37.8071299,
|
|
"lon": -122.41732,
|
|
"name": "Jones St & Beach St",
|
|
"code": "15651",
|
|
"hidden": False,
|
|
"showDestinationSelector": True,
|
|
"directions": ["F_0_var1", "F_0_var0"],
|
|
},
|
|
],
|
|
"directions": route_config_direction,
|
|
"paths": [],
|
|
"timestamp": "2024-06-23T03:06:58Z",
|
|
}
|
|
|
|
return instance
|