From 08a850cfc7bf00dbc383ab045dac33ab964baf27 Mon Sep 17 00:00:00 2001 From: Simone Chemelli Date: Thu, 28 Aug 2025 11:57:32 +0200 Subject: [PATCH] Fix exception countries migration for Alexa Devices (#151292) --- .../components/alexa_devices/__init__.py | 2 +- .../components/alexa_devices/const.py | 18 +++++++++--------- tests/components/alexa_devices/const.py | 1 - tests/components/alexa_devices/test_init.py | 9 +++------ 4 files changed, 13 insertions(+), 17 deletions(-) diff --git a/homeassistant/components/alexa_devices/__init__.py b/homeassistant/components/alexa_devices/__init__.py index 7a4641bc51f..9407a2d8987 100644 --- a/homeassistant/components/alexa_devices/__init__.py +++ b/homeassistant/components/alexa_devices/__init__.py @@ -48,7 +48,7 @@ async def async_migrate_entry(hass: HomeAssistant, entry: AmazonConfigEntry) -> ) # Convert country in domain - country = entry.data[CONF_COUNTRY] + country = entry.data[CONF_COUNTRY].lower() domain = COUNTRY_DOMAINS.get(country, country) # Add site to login data diff --git a/homeassistant/components/alexa_devices/const.py b/homeassistant/components/alexa_devices/const.py index 3ade3ad3ecd..c60096bae57 100644 --- a/homeassistant/components/alexa_devices/const.py +++ b/homeassistant/components/alexa_devices/const.py @@ -7,21 +7,21 @@ _LOGGER = logging.getLogger(__package__) DOMAIN = "alexa_devices" CONF_LOGIN_DATA = "login_data" -DEFAULT_DOMAIN = {"domain": "com"} +DEFAULT_DOMAIN = "com" COUNTRY_DOMAINS = { "ar": DEFAULT_DOMAIN, "at": DEFAULT_DOMAIN, - "au": {"domain": "com.au"}, - "be": {"domain": "com.be"}, + "au": "com.au", + "be": "com.be", "br": DEFAULT_DOMAIN, - "gb": {"domain": "co.uk"}, + "gb": "co.uk", "il": DEFAULT_DOMAIN, - "jp": {"domain": "co.jp"}, - "mx": {"domain": "com.mx"}, + "jp": "co.jp", + "mx": "com.mx", "no": DEFAULT_DOMAIN, - "nz": {"domain": "com.au"}, + "nz": "com.au", "pl": DEFAULT_DOMAIN, - "tr": {"domain": "com.tr"}, + "tr": "com.tr", "us": DEFAULT_DOMAIN, - "za": {"domain": "co.za"}, + "za": "co.za", } diff --git a/tests/components/alexa_devices/const.py b/tests/components/alexa_devices/const.py index 6a4dff1c38d..ca701cd46e8 100644 --- a/tests/components/alexa_devices/const.py +++ b/tests/components/alexa_devices/const.py @@ -1,7 +1,6 @@ """Alexa Devices tests const.""" TEST_CODE = "023123" -TEST_COUNTRY = "IT" TEST_PASSWORD = "fake_password" TEST_SERIAL_NUMBER = "echo_test_serial_number" TEST_USERNAME = "fake_email@gmail.com" diff --git a/tests/components/alexa_devices/test_init.py b/tests/components/alexa_devices/test_init.py index 7055f8482cc..6c3faffd27b 100644 --- a/tests/components/alexa_devices/test_init.py +++ b/tests/components/alexa_devices/test_init.py @@ -11,7 +11,7 @@ from homeassistant.core import HomeAssistant from homeassistant.helpers import device_registry as dr from . import setup_integration -from .const import TEST_COUNTRY, TEST_PASSWORD, TEST_SERIAL_NUMBER, TEST_USERNAME +from .const import TEST_PASSWORD, TEST_SERIAL_NUMBER, TEST_USERNAME from tests.common import MockConfigEntry @@ -42,7 +42,7 @@ async def test_migrate_entry( domain=DOMAIN, title="Amazon Test Account", data={ - CONF_COUNTRY: TEST_COUNTRY, + CONF_COUNTRY: "US", # country should be in COUNTRY_DOMAINS exceptions CONF_USERNAME: TEST_USERNAME, CONF_PASSWORD: TEST_PASSWORD, CONF_LOGIN_DATA: {"session": "test-session"}, @@ -58,7 +58,4 @@ async def test_migrate_entry( assert len(hass.config_entries.async_entries(DOMAIN)) == 1 assert config_entry.state is ConfigEntryState.LOADED assert config_entry.minor_version == 2 - assert ( - config_entry.data[CONF_LOGIN_DATA]["site"] - == f"https://www.amazon.{TEST_COUNTRY}" - ) + assert config_entry.data[CONF_LOGIN_DATA]["site"] == "https://www.amazon.com"