From ea7f2af966a3a46c1670f14d8cc5315537f3aca0 Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Sun, 31 Mar 2024 23:23:38 -1000 Subject: [PATCH] Fix missing mocking in blink tests (#114540) extracted from #114539 --- tests/components/blink/conftest.py | 1 + tests/components/blink/test_init.py | 12 ++++++++++-- 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/tests/components/blink/conftest.py b/tests/components/blink/conftest.py index b18fdf7615e67a..c6e3ee0960d4e7 100644 --- a/tests/components/blink/conftest.py +++ b/tests/components/blink/conftest.py @@ -60,6 +60,7 @@ def blink_api_fixture(camera) -> MagicMock: mock_blink_api.refresh = AsyncMock(return_value=True) mock_blink_api.sync = MagicMock(return_value=True) mock_blink_api.cameras = {camera.name: camera} + mock_blink_api.request_homescreen = AsyncMock(return_value=True) with patch("homeassistant.components.blink.Blink") as class_mock: class_mock.return_value = mock_blink_api diff --git a/tests/components/blink/test_init.py b/tests/components/blink/test_init.py index 1f3a4c956c4079..46806ef33498ad 100644 --- a/tests/components/blink/test_init.py +++ b/tests/components/blink/test_init.py @@ -1,8 +1,9 @@ """Test the Blink init.""" -from unittest.mock import AsyncMock, MagicMock +from unittest.mock import AsyncMock, MagicMock, patch from aiohttp import ClientError +from blinkpy.auth import LoginError import pytest from homeassistant.components.blink.const import ( @@ -53,9 +54,16 @@ async def test_setup_not_ready_authkey_required( """Test setup failed because 2FA is needed to connect to the Blink system.""" mock_blink_auth_api.check_key_required = MagicMock(return_value=True) + mock_blink_auth_api.send_auth_key = AsyncMock(return_value=False) mock_config_entry.add_to_hass(hass) - assert not await hass.config_entries.async_setup(mock_config_entry.entry_id) + with patch( + "homeassistant.components.blink.config_flow.Auth.startup", + side_effect=LoginError, + ): + assert not await hass.config_entries.async_setup(mock_config_entry.entry_id) + await hass.async_block_till_done() + assert mock_config_entry.state is ConfigEntryState.SETUP_ERROR