mirror of
https://github.com/home-assistant/core.git
synced 2025-07-19 11:17:21 +00:00
Use async_load_fixture in twitch tests (#146016)
This commit is contained in:
parent
89a40f1c48
commit
ebfbea39ff
@ -7,8 +7,9 @@ from twitchAPI.object.base import TwitchObject
|
|||||||
|
|
||||||
from homeassistant.components.twitch.const import DOMAIN
|
from homeassistant.components.twitch.const import DOMAIN
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
|
from homeassistant.util.json import JsonArrayType
|
||||||
|
|
||||||
from tests.common import MockConfigEntry, load_json_array_fixture
|
from tests.common import MockConfigEntry, async_load_json_array_fixture
|
||||||
|
|
||||||
|
|
||||||
async def setup_integration(hass: HomeAssistant, config_entry: MockConfigEntry) -> None:
|
async def setup_integration(hass: HomeAssistant, config_entry: MockConfigEntry) -> None:
|
||||||
@ -25,24 +26,35 @@ TwitchType = TypeVar("TwitchType", bound=TwitchObject)
|
|||||||
class TwitchIterObject(Generic[TwitchType]):
|
class TwitchIterObject(Generic[TwitchType]):
|
||||||
"""Twitch object iterator."""
|
"""Twitch object iterator."""
|
||||||
|
|
||||||
def __init__(self, fixture: str, target_type: type[TwitchType]) -> None:
|
raw_data: JsonArrayType
|
||||||
|
data: list
|
||||||
|
total: int
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self, hass: HomeAssistant, fixture: str, target_type: type[TwitchType]
|
||||||
|
) -> None:
|
||||||
"""Initialize object."""
|
"""Initialize object."""
|
||||||
self.raw_data = load_json_array_fixture(fixture, DOMAIN)
|
self.hass = hass
|
||||||
self.data = [target_type(**item) for item in self.raw_data]
|
self.fixture = fixture
|
||||||
self.total = len(self.raw_data)
|
|
||||||
self.target_type = target_type
|
self.target_type = target_type
|
||||||
|
|
||||||
async def __aiter__(self) -> AsyncIterator[TwitchType]:
|
async def __aiter__(self) -> AsyncIterator[TwitchType]:
|
||||||
"""Return async iterator."""
|
"""Return async iterator."""
|
||||||
|
if not hasattr(self, "raw_data"):
|
||||||
|
self.raw_data = await async_load_json_array_fixture(
|
||||||
|
self.hass, self.fixture, DOMAIN
|
||||||
|
)
|
||||||
|
self.data = [self.target_type(**item) for item in self.raw_data]
|
||||||
|
self.total = len(self.raw_data)
|
||||||
async for item in get_generator_from_data(self.raw_data, self.target_type):
|
async for item in get_generator_from_data(self.raw_data, self.target_type):
|
||||||
yield item
|
yield item
|
||||||
|
|
||||||
|
|
||||||
async def get_generator(
|
async def get_generator(
|
||||||
fixture: str, target_type: type[TwitchType]
|
hass: HomeAssistant, fixture: str, target_type: type[TwitchType]
|
||||||
) -> AsyncGenerator[TwitchType]:
|
) -> AsyncGenerator[TwitchType]:
|
||||||
"""Return async generator."""
|
"""Return async generator."""
|
||||||
data = load_json_array_fixture(fixture, DOMAIN)
|
data = await async_load_json_array_fixture(hass, fixture, DOMAIN)
|
||||||
async for item in get_generator_from_data(data, target_type):
|
async for item in get_generator_from_data(data, target_type):
|
||||||
yield item
|
yield item
|
||||||
|
|
||||||
|
@ -93,7 +93,7 @@ def mock_connection(aioclient_mock: AiohttpClientMocker) -> None:
|
|||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def twitch_mock() -> Generator[AsyncMock]:
|
def twitch_mock(hass: HomeAssistant) -> Generator[AsyncMock]:
|
||||||
"""Return as fixture to inject other mocks."""
|
"""Return as fixture to inject other mocks."""
|
||||||
with (
|
with (
|
||||||
patch(
|
patch(
|
||||||
@ -106,13 +106,13 @@ def twitch_mock() -> Generator[AsyncMock]:
|
|||||||
),
|
),
|
||||||
):
|
):
|
||||||
mock_client.return_value.get_users = lambda *args, **kwargs: get_generator(
|
mock_client.return_value.get_users = lambda *args, **kwargs: get_generator(
|
||||||
"get_users.json", TwitchUser
|
hass, "get_users.json", TwitchUser
|
||||||
)
|
)
|
||||||
mock_client.return_value.get_followed_channels.return_value = TwitchIterObject(
|
mock_client.return_value.get_followed_channels.return_value = TwitchIterObject(
|
||||||
"get_followed_channels.json", FollowedChannel
|
hass, "get_followed_channels.json", FollowedChannel
|
||||||
)
|
)
|
||||||
mock_client.return_value.get_followed_streams.return_value = get_generator(
|
mock_client.return_value.get_followed_streams.return_value = get_generator(
|
||||||
"get_followed_streams.json", Stream
|
hass, "get_followed_streams.json", Stream
|
||||||
)
|
)
|
||||||
mock_client.return_value.check_user_subscription.return_value = (
|
mock_client.return_value.check_user_subscription.return_value = (
|
||||||
UserSubscription(
|
UserSubscription(
|
||||||
|
@ -175,7 +175,7 @@ async def test_reauth_wrong_account(
|
|||||||
"""Check reauth flow."""
|
"""Check reauth flow."""
|
||||||
await setup_integration(hass, config_entry)
|
await setup_integration(hass, config_entry)
|
||||||
twitch_mock.return_value.get_users = lambda *args, **kwargs: get_generator(
|
twitch_mock.return_value.get_users = lambda *args, **kwargs: get_generator(
|
||||||
"get_users_2.json", TwitchUser
|
hass, "get_users_2.json", TwitchUser
|
||||||
)
|
)
|
||||||
result = await config_entry.start_reauth_flow(hass)
|
result = await config_entry.start_reauth_flow(hass)
|
||||||
assert result["type"] is FlowResultType.FORM
|
assert result["type"] is FlowResultType.FORM
|
||||||
|
@ -53,7 +53,7 @@ async def test_oauth_without_sub_and_follow(
|
|||||||
) -> None:
|
) -> None:
|
||||||
"""Test state with oauth."""
|
"""Test state with oauth."""
|
||||||
twitch_mock.return_value.get_followed_channels.return_value = TwitchIterObject(
|
twitch_mock.return_value.get_followed_channels.return_value = TwitchIterObject(
|
||||||
"empty_response.json", FollowedChannel
|
hass, "empty_response.json", FollowedChannel
|
||||||
)
|
)
|
||||||
twitch_mock.return_value.check_user_subscription.side_effect = (
|
twitch_mock.return_value.check_user_subscription.side_effect = (
|
||||||
TwitchResourceNotFound
|
TwitchResourceNotFound
|
||||||
@ -70,7 +70,7 @@ async def test_oauth_with_sub(
|
|||||||
) -> None:
|
) -> None:
|
||||||
"""Test state with oauth and sub."""
|
"""Test state with oauth and sub."""
|
||||||
twitch_mock.return_value.get_followed_channels.return_value = TwitchIterObject(
|
twitch_mock.return_value.get_followed_channels.return_value = TwitchIterObject(
|
||||||
"empty_response.json", FollowedChannel
|
hass, "empty_response.json", FollowedChannel
|
||||||
)
|
)
|
||||||
subscription = await async_load_json_object_fixture(
|
subscription = await async_load_json_object_fixture(
|
||||||
hass, "check_user_subscription_2.json", DOMAIN
|
hass, "check_user_subscription_2.json", DOMAIN
|
||||||
|
Loading…
x
Reference in New Issue
Block a user