Skip to content

Commit

Permalink
Fix missing mocking in blink tests
Browse files Browse the repository at this point in the history
extracted from #114539
  • Loading branch information
bdraco committed Mar 31, 2024
1 parent 52741d7 commit e29da50
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 2 deletions.
1 change: 1 addition & 0 deletions tests/components/blink/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
12 changes: 10 additions & 2 deletions tests/components/blink/test_init.py
Original file line number Diff line number Diff line change
@@ -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 (
Expand Down Expand Up @@ -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


Expand Down

0 comments on commit e29da50

Please sign in to comment.