Fix KeyError in friends coordinator (#149684)

homeassistant/components/playstation_network/coordinator.py

@@ -6,7 +6,7 @@ from abc import abstractmethod
 from dataclasses import dataclass
 from datetime import timedelta
 import logging
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from psnawp_api.core.psnawp_exceptions import (
     PSNAWPAuthenticationError,
@@ -29,7 +29,7 @@ from homeassistant.exceptions import (
 )
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
-from .const import CONF_ACCOUNT_ID, DOMAIN
+from .const import DOMAIN
 from .helpers import PlaystationNetwork, PlaystationNetworkData
 
 _LOGGER = logging.getLogger(__name__)
@@ -176,7 +176,9 @@ class PlaystationNetworkFriendDataCoordinator(
 
     def _setup(self) -> None:
         """Set up the coordinator."""
-        self.user = self.psn.psn.user(account_id=self.subentry.data[CONF_ACCOUNT_ID])
+        if TYPE_CHECKING:
+            assert self.subentry.unique_id
+        self.user = self.psn.psn.user(account_id=self.subentry.unique_id)
         self.profile = self.user.profile()
 
     async def _async_setup(self) -> None:

tests/components/playstation_network/conftest.py

@@ -14,11 +14,7 @@ from psnawp_api.models.trophies import (
 )
 import pytest
 
-from homeassistant.components.playstation_network.const import (
+from homeassistant.components.playstation_network.const import CONF_NPSSO, DOMAIN
-    CONF_ACCOUNT_ID,
-    CONF_NPSSO,
-    DOMAIN,
-)
 from homeassistant.config_entries import ConfigSubentryData
 
 from tests.common import MockConfigEntry
@@ -40,7 +36,7 @@ def mock_config_entry() -> MockConfigEntry:
         unique_id=PSN_ID,
         subentries_data=[
             ConfigSubentryData(
-                data={CONF_ACCOUNT_ID: "fren-psn-id"},
+                data={},
                 subentry_id="ABCDEF",
                 subentry_type="friend",
                 title="PublicUniversalFriend",