Improve exception handling of PlayStation Network (#147792)

2026-09-28 02:18:10 -04:00 · 2025-06-30 19:53:46 +02:00
parent 5e3fc858d8
commit 2c30a5a14c
2 changed files with 118 additions and 5 deletions
@@ -0,0 +1,109 @@
+"""Tests for PlayStation Network."""
+
+from unittest.mock import MagicMock
+
+from psnawp_api.core import (
+    PSNAWPAuthenticationError,
+    PSNAWPClientError,
+    PSNAWPNotFoundError,
+    PSNAWPServerError,
+)
+import pytest
+
+from homeassistant.components.playstation_network.const import DOMAIN
+from homeassistant.config_entries import SOURCE_REAUTH, ConfigEntryState
+from homeassistant.core import HomeAssistant
+
+from tests.common import MockConfigEntry
+
+
+@pytest.mark.parametrize(
+    "exception", [PSNAWPNotFoundError, PSNAWPServerError, PSNAWPClientError]
+)
+async def test_config_entry_not_ready(
+    hass: HomeAssistant,
+    config_entry: MockConfigEntry,
+    mock_psnawpapi: MagicMock,
+    exception: Exception,
+) -> None:
+    """Test config entry not ready."""
+
+    mock_psnawpapi.user.side_effect = exception
+    config_entry.add_to_hass(hass)
+    await hass.config_entries.async_setup(config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    assert config_entry.state is ConfigEntryState.SETUP_RETRY
+
+
+async def test_config_entry_auth_failed(
+    hass: HomeAssistant,
+    config_entry: MockConfigEntry,
+    mock_psnawpapi: MagicMock,
+) -> None:
+    """Test config entry auth failed setup error."""
+
+    mock_psnawpapi.user.side_effect = PSNAWPAuthenticationError
+    config_entry.add_to_hass(hass)
+    await hass.config_entries.async_setup(config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    assert config_entry.state is ConfigEntryState.SETUP_ERROR
+
+    flows = hass.config_entries.flow.async_progress()
+    assert len(flows) == 1
+
+    flow = flows[0]
+    assert flow.get("step_id") == "reauth_confirm"
+    assert flow.get("handler") == DOMAIN
+
+    assert "context" in flow
+    assert flow["context"].get("source") == SOURCE_REAUTH
+    assert flow["context"].get("entry_id") == config_entry.entry_id
+
+
+@pytest.mark.parametrize(
+    "exception", [PSNAWPNotFoundError, PSNAWPServerError, PSNAWPClientError]
+)
+async def test_coordinator_update_data_failed(
+    hass: HomeAssistant,
+    config_entry: MockConfigEntry,
+    mock_psnawpapi: MagicMock,
+    exception: Exception,
+) -> None:
+    """Test coordinator data update failed."""
+
+    mock_psnawpapi.user.return_value.get_presence.side_effect = exception
+    config_entry.add_to_hass(hass)
+    await hass.config_entries.async_setup(config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    assert config_entry.state is ConfigEntryState.SETUP_RETRY
+
+
+async def test_coordinator_update_auth_failed(
+    hass: HomeAssistant,
+    config_entry: MockConfigEntry,
+    mock_psnawpapi: MagicMock,
+) -> None:
+    """Test coordinator update auth failed setup error."""
+
+    mock_psnawpapi.user.return_value.get_presence.side_effect = (
+        PSNAWPAuthenticationError
+    )
+    config_entry.add_to_hass(hass)
+    await hass.config_entries.async_setup(config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    assert config_entry.state is ConfigEntryState.SETUP_ERROR
+
+    flows = hass.config_entries.flow.async_progress()
+    assert len(flows) == 1
+
+    flow = flows[0]
+    assert flow.get("step_id") == "reauth_confirm"
+    assert flow.get("handler") == DOMAIN
+
+    assert "context" in flow
+    assert flow["context"].get("source") == SOURCE_REAUTH
+    assert flow["context"].get("entry_id") == config_entry.entry_id