evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Prosegur late review comments (#88859)

Browse source 
* address late comments on #76428

* adress review

* extra tweaks
This commit is contained in:
Diogo Gomes 2023-02-28 07:16:22 +00:00 • committed by GitHub
parent 7b3cab1bfe
commit 07c25b3dd8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 14 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/components/prosegur/conftest.py
View file

@ -1,5 +1,5 @@
"""Define test fixtures for Prosegur."""
from unittest.mock import AsyncMock, patch
from unittest.mock import AsyncMock, MagicMock, patch
from pyprosegur.installation import Camera
import pytest
@ -30,9 +30,12 @@ def mock_config_entry() -> MockConfigEntry:
@pytest.fixture
def mock_install() -> AsyncMock:
"""Return the mocked alarm install."""
install = AsyncMock()
install = MagicMock()
install.contract = CONTRACT
install.cameras = [Camera("1", "test_cam")]
install.arm = AsyncMock()
install.disarm = AsyncMock()
install.arm_partially = AsyncMock()
install.get_image = AsyncMock(return_value=b"ABC")
install.request_image = AsyncMock()
@ -51,7 +54,7 @@ async def init_integration(
with patch(
"pyprosegur.installation.Installation.retrieve", return_value=mock_install
), patch("pyprosegur.auth.Auth.login", return_value=AsyncMock()):
), patch("pyprosegur.auth.Auth.login"):
await hass.config_entries.async_setup(mock_config_entry.entry_id)
await hass.async_block_till_done()