Use is in ConfigEntryState enum comparison in tests (A-M) (#114925)

This commit is contained in:
epenet
2024-04-05 17:16:55 +02:00
committed by GitHub
parent dd8de14cc5
commit 0b01326f9f
99 changed files with 302 additions and 295 deletions

View File

@@ -134,12 +134,12 @@ async def test_setup_and_unload_entry(
):
assert await hass.config_entries.async_setup(java_mock_config_entry.entry_id)
await hass.async_block_till_done()
assert java_mock_config_entry.state == ConfigEntryState.LOADED
assert java_mock_config_entry.state is ConfigEntryState.LOADED
assert await hass.config_entries.async_unload(java_mock_config_entry.entry_id)
await hass.async_block_till_done()
assert not hass.data.get(DOMAIN)
assert java_mock_config_entry.state == ConfigEntryState.NOT_LOADED
assert java_mock_config_entry.state is ConfigEntryState.NOT_LOADED
async def test_setup_entry_lookup_failure(
@@ -157,7 +157,7 @@ async def test_setup_entry_lookup_failure(
)
await hass.async_block_till_done()
assert java_mock_config_entry.state == ConfigEntryState.SETUP_RETRY
assert java_mock_config_entry.state is ConfigEntryState.SETUP_RETRY
async def test_setup_entry_init_failure(
@@ -175,7 +175,7 @@ async def test_setup_entry_init_failure(
)
await hass.async_block_till_done()
assert java_mock_config_entry.state == ConfigEntryState.SETUP_RETRY
assert java_mock_config_entry.state is ConfigEntryState.SETUP_RETRY
async def test_setup_entry_not_ready(
@@ -199,7 +199,7 @@ async def test_setup_entry_not_ready(
)
await hass.async_block_till_done()
assert java_mock_config_entry.state == ConfigEntryState.SETUP_RETRY
assert java_mock_config_entry.state is ConfigEntryState.SETUP_RETRY
async def test_entry_migration(
@@ -246,7 +246,7 @@ async def test_entry_migration(
CONF_ADDRESS: TEST_ADDRESS,
}
assert migrated_config_entry.version == 3
assert migrated_config_entry.state == ConfigEntryState.LOADED
assert migrated_config_entry.state is ConfigEntryState.LOADED
# Test migrated device entry.
device_entry = device_registry.async_get(device_entry_id)
@@ -305,7 +305,7 @@ async def test_entry_migration_host_only(
CONF_ADDRESS: TEST_HOST,
}
assert v1_mock_config_entry.version == 3
assert v1_mock_config_entry.state == ConfigEntryState.LOADED
assert v1_mock_config_entry.state is ConfigEntryState.LOADED
async def test_entry_migration_v3_failure(
@@ -335,4 +335,4 @@ async def test_entry_migration_v3_failure(
# Test config entry.
assert v1_mock_config_entry.version == 2
assert v1_mock_config_entry.state == ConfigEntryState.MIGRATION_ERROR
assert v1_mock_config_entry.state is ConfigEntryState.MIGRATION_ERROR