8000 Use async_load_fixture in moehlenhoff_alpha2 tests (#146012) · home-assistant/core@664eb7a · GitHub
[go: up one dir, main page]
Skip to content

Commit 664eb7a

Browse files
epenetepenet
authored
Use async_load_fixture in moehlenhoff_alpha2 tests (#146012)
1 parent 33b99b6 commit 664eb7a

File tree

2 files changed

+12
-6
lines changed

2 files changed

+12
-6
lines changed

‎tests/components/moehlenhoff_alpha2/__init__.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,23 @@
11
"""Tests for the moehlenhoff_alpha2 integration."""
22

3+
from functools import partialmethod
34
from unittest.mock import patch
45

6+
from moehlenhoff_alpha2 import Alpha2Base
57
import xmltodict
68

79
from homeassistant.components.moehlenhoff_alpha2.const import DOMAIN
810
from homeassistant.const import CONF_HOST
911
from homeassistant.core import HomeAssistant
1012

11-
from tests.common import MockConfigEntry, load_fixture
13+
from tests.common import MockConfigEntry, async_load_fixture
1214

1315
MOCK_BASE_HOST = "fake-base-host"
1416

1517

16-
async def mock_update_data(self):
18+
async def mock_update_data(self: Alpha2Base, hass: HomeAssistant) -> None:
1719
"""Mock Alpha2Base.update_data."""
18-
data = xmltodict.parse(load_fixture("static2.xml", DOMAIN))
20+
data = xmltodict.parse(await async_load_fixture(hass, "static2.xml", DOMAIN))
1921
for _type in ("HEATAREA", "HEATCTRL", "IODEVICE"):
2022
if not isinstance(data["Devices"]["Device"][_type], list):
2123
data["Devices"]["Device"][_type] = [data["Devices"]["Device"][_type]]
@@ -26,7 +28,7 @@ async def init_integration(hass: HomeAssistant) -> MockConfigEntry:
2628
"""Mock integration setup."""
2729
with patch(
2830
"homeassistant.components.moehlenhoff_alpha2.coordinator.Alpha2Base.update_data",
29-
mock_update_data,
31+
partialmethod(mock_update_data, hass),
3032
):
3133
entry = MockConfigEntry(
3234
domain=DOMAIN,

‎tests/components/moehlenhoff_alpha2/test_config_flow.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
"""Test the moehlenhoff_alpha2 config flow."""
22

3+
from functools import partialmethod
34
from unittest.mock import patch
45

56
from homeassistant import config_entries
@@ -24,7 +25,7 @@ async def test_form(hass: HomeAssistant) -> None:
2425
with (
2526
patch(
2627
"homeassistant.components.moehlenhoff_alpha2.config_flow.Alpha2Base.update_data",
27-
mock_update_data,
28+
partialmethod(mock_update_data, hass),
2829
),
2930
patch(
3031
"homeassistant.components.moehlenhoff_alpha2.async_setup_entry",
@@ -54,7 +55,10 @@ async def test_form_duplicate_error(hass: HomeAssistant) -> None:
5455

5556
assert config_entry.data["host"] == MOCK_BASE_HOST
5657

57-
with patch("moehlenhoff_alpha2.Alpha2Base.update_data", mock_update_data):
58+
with patch(
59+
"moehlenhoff_alpha2.Alpha2Base.update_data",
60+
partialmethod(mock_update_data, hass),
61+
):
5862
result = await hass.config_entries.flow.async_init(
5963
DOMAIN,
6064
data={"host": MOCK_BASE_HOST},

0 commit comments

Comments
 (0)
0