mirror of
https://github.com/home-assistant/core.git
synced 2025-06-24 17:11:53 +02:00
111 lines
3.2 KiB
Python
111 lines
3.2 KiB
Python
"""Common fixtures for the Playstation Network tests."""
|
|
|
|
from collections.abc import Generator
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.playstation_network.const import CONF_NPSSO, DOMAIN
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
NPSSO_TOKEN: str = "npsso-token"
|
|
NPSSO_TOKEN_INVALID_JSON: str = "{'npsso': 'npsso-token'"
|
|
PSN_ID: str = "my-psn-id"
|
|
|
|
|
|
@pytest.fixture(name="config_entry")
|
|
def mock_config_entry() -> MockConfigEntry:
|
|
"""Mock PlayStation Network configuration entry."""
|
|
return MockConfigEntry(
|
|
domain=DOMAIN,
|
|
title="test-user",
|
|
data={
|
|
CONF_NPSSO: NPSSO_TOKEN,
|
|
},
|
|
unique_id=PSN_ID,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_setup_entry() -> Generator[AsyncMock]:
|
|
"""Override async_setup_entry."""
|
|
with patch(
|
|
"homeassistant.components.playstation_network.async_setup_entry",
|
|
return_value=True,
|
|
) as mock_setup_entry:
|
|
yield mock_setup_entry
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_user() -> Generator[MagicMock]:
|
|
"""Mock psnawp_api User object."""
|
|
|
|
with patch(
|
|
"homeassistant.components.playstation_network.helpers.User",
|
|
autospec=True,
|
|
) as mock_client:
|
|
client = mock_client.return_value
|
|
|
|
client.account_id = PSN_ID
|
|
client.online_id = "testuser"
|
|
|
|
client.get_presence.return_value = {
|
|
"basicPresence": {
|
|
"availability": "availableToPlay",
|
|
"primaryPlatformInfo": {"onlineStatus": "online", "platform": "PS5"},
|
|
"gameTitleInfoList": [
|
|
{
|
|
"npTitleId": "PPSA07784_00",
|
|
"titleName": "STAR WARS Jedi: Survivor™",
|
|
"format": "PS5",
|
|
"launchPlatform": "PS5",
|
|
"conceptIconUrl": "https://image.api.playstation.com/vulcan/ap/rnd/202211/2222/l8QTN7ThQK3lRBHhB3nX1s7h.png",
|
|
}
|
|
],
|
|
}
|
|
}
|
|
|
|
yield client
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_psnawpapi(mock_user: MagicMock) -> Generator[MagicMock]:
|
|
"""Mock psnawp_api."""
|
|
|
|
with patch(
|
|
"homeassistant.components.playstation_network.helpers.PSNAWP",
|
|
autospec=True,
|
|
) as mock_client:
|
|
client = mock_client.return_value
|
|
|
|
client.user.return_value = mock_user
|
|
client.me.return_value.get_account_devices.return_value = [
|
|
{
|
|
"deviceId": "1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234",
|
|
"deviceType": "PS5",
|
|
"activationType": "PRIMARY",
|
|
"activationDate": "2021-01-14T18:00:00.000Z",
|
|
"accountDeviceVector": "abcdefghijklmnopqrstuv",
|
|
}
|
|
]
|
|
yield client
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_psnawp_npsso(mock_user: MagicMock) -> Generator[MagicMock]:
|
|
"""Mock psnawp_api."""
|
|
|
|
with patch(
|
|
"homeassistant.components.playstation_network.config_flow.parse_npsso_token",
|
|
side_effect=lambda token: token,
|
|
) as npsso:
|
|
yield npsso
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_token() -> Generator[MagicMock]:
|
|
"""Mock token generator."""
|
|
with patch("secrets.token_hex", return_value="123456789") as token:
|
|
yield token
|