Keep original cause in test-before-setup exceptions (#2657)

This commit is contained in:
Ville Skyttä 2025-04-26 15:10:30 +03:00 committed by GitHub
parent 0a2cfa3fc1
commit 105df90176
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -25,15 +25,15 @@ async def async_setup_entry(hass: HomeAssistant, entry: MyIntegrationConfigEntry
"""Set up my integration from a config entry."""
client = MyClient(entry.data[CONF_HOST])
try:
await client.async_setup()
except OfflineException:
raise ConfigEntryNotReady("Device is offline")
except InvalidAuthException:
raise ConfigEntryAuthFailed("Invalid authentication")
except AccountClosedException:
raise ConfigEntryError("Account closed")
except OfflineException as ex:
raise ConfigEntryNotReady("Device is offline") from ex
except InvalidAuthException as ex:
raise ConfigEntryAuthFailed("Invalid authentication") from ex
except AccountClosedException as ex:
raise ConfigEntryError("Account closed") from ex
entry.runtime_data = client