diff --git a/tests/components/blink/test_init.py b/tests/components/blink/test_init.py index 5521303ac8b..454195c4d60 100644 --- a/tests/components/blink/test_init.py +++ b/tests/components/blink/test_init.py @@ -85,10 +85,10 @@ async def test_migrate_V0( mock_config_entry: MockConfigEntry, ) -> None: """Test migration script version 0.""" + mock_config_entry.add_to_hass(hass) hass.config_entries.async_update_entry(mock_config_entry, version=0) - mock_config_entry.add_to_hass(hass) assert await hass.config_entries.async_setup(mock_config_entry.entry_id) await hass.async_block_till_done() entry = hass.config_entries.async_get_entry(mock_config_entry.entry_id) @@ -104,6 +104,7 @@ async def test_migrate( version, ) -> None: """Test migration scripts.""" + mock_config_entry.add_to_hass(hass) hass.config_entries.async_update_entry( mock_config_entry, @@ -111,7 +112,6 @@ async def test_migrate( data={**mock_config_entry.data, "login_response": "Blah"}, ) - mock_config_entry.add_to_hass(hass) assert not await hass.config_entries.async_setup(mock_config_entry.entry_id) await hass.async_block_till_done() entry = hass.config_entries.async_get_entry(mock_config_entry.entry_id)