33from collections .abc import Generator
44from unittest .mock import AsyncMock , patch
55
6- from airos .airos8 import AirOSData
6+ from airos .airos8 import AirOS8Data
77import pytest
88
99from homeassistant .components .airos .const import DOMAIN
1616def ap_fixture ():
1717 """Load fixture data for AP mode."""
1818 json_data = load_json_object_fixture ("airos_loco5ac_ap-ptp.json" , DOMAIN )
19- return AirOSData .from_dict (json_data )
19+ return AirOS8Data .from_dict (json_data )
2020
2121
2222@pytest .fixture
@@ -30,15 +30,15 @@ def mock_setup_entry() -> Generator[AsyncMock]:
3030
3131@pytest .fixture
3232def mock_airos_client (
33- request : pytest .FixtureRequest , ap_fixture : AirOSData
33+ request : pytest .FixtureRequest , ap_fixture : AirOS8Data
3434) -> Generator [AsyncMock ]:
3535 """Fixture to mock the AirOS API client."""
3636 with (
3737 patch (
38- "homeassistant.components.airos.config_flow.AirOS " , autospec = True
38+ "homeassistant.components.airos.config_flow.AirOS8 " , autospec = True
3939 ) as mock_airos ,
40- patch ("homeassistant.components.airos.coordinator.AirOS " , new = mock_airos ),
41- patch ("homeassistant.components.airos.AirOS " , new = mock_airos ),
40+ patch ("homeassistant.components.airos.coordinator.AirOS8 " , new = mock_airos ),
41+ patch ("homeassistant.components.airos.AirOS8 " , new = mock_airos ),
4242 ):
4343 client = mock_airos .return_value
4444 client .status .return_value = ap_fixture
0 commit comments