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.""" """Set up my integration from a config entry."""
client = MyClient(entry.data[CONF_HOST]) client = MyClient(entry.data[CONF_HOST])
try: try:
await client.async_setup() await client.async_setup()
except OfflineException: except OfflineException as ex:
raise ConfigEntryNotReady("Device is offline") raise ConfigEntryNotReady("Device is offline") from ex
except InvalidAuthException: except InvalidAuthException as ex:
raise ConfigEntryAuthFailed("Invalid authentication") raise ConfigEntryAuthFailed("Invalid authentication") from ex
except AccountClosedException: except AccountClosedException as ex:
raise ConfigEntryError("Account closed") raise ConfigEntryError("Account closed") from ex
entry.runtime_data = client entry.runtime_data = client