diff --git a/tests/components/vesync/test_init.py b/tests/components/vesync/test_init.py index 1c53c051a076..485c42830ec1 100644 --- a/tests/components/vesync/test_init.py +++ b/tests/components/vesync/test_init.py @@ -10,10 +10,7 @@ from pyvesync.utils.errors import ( VeSyncServerError, ) -from homeassistant.components.vesync import ( - async_remove_config_entry_device, - async_setup_entry, -) +from homeassistant.components.vesync import async_remove_config_entry_device from homeassistant.components.vesync.const import DOMAIN from homeassistant.config_entries import ConfigEntry, ConfigEntryState from homeassistant.const import Platform @@ -54,8 +51,7 @@ async def test_async_setup_entry__no_devices( ) -> None: """Test setup connects to vesync and creates empty config when no devices.""" with patch.object(hass.config_entries, "async_forward_entry_setups") as setups_mock: - # pylint: disable-next=home-assistant-tests-direct-async-setup-entry - assert await async_setup_entry(hass, config_entry) + assert await hass.config_entries.async_setup(config_entry.entry_id) # Assert platforms loaded await hass.async_block_till_done() assert setups_mock.call_count == 1 @@ -82,8 +78,7 @@ async def test_async_setup_entry__loads_fans( manager._dev_list["fans"].append(fan) with patch.object(hass.config_entries, "async_forward_entry_setups") as setups_mock: - # pylint: disable-next=home-assistant-tests-direct-async-setup-entry - assert await async_setup_entry(hass, config_entry) + assert await hass.config_entries.async_setup(config_entry.entry_id) # Assert platforms loaded await hass.async_block_till_done() assert setups_mock.call_count == 1