Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 12:59:34 +00:00
Code Activity

Use a future for mock coro (#34989)

Browse Source
This commit is contained in:
Paulus Schoutsen
2020-04-30 16:31:00 -07:00
committed by GitHub
parent ba7391528f
commit 76f392476b
26 changed files with 154 additions and 262 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/components/melissa/test_init.py
View File

@@ -1,23 +1,22 @@
"""The test for the Melissa Climate component."""
from homeassistant.components import melissa
from tests.common import MockDependency, mock_coro_func
from tests.async_mock import AsyncMock, patch
VALID_CONFIG = {"melissa": {"username": "********", "password": "********"}}
async def test_setup(hass):
"""Test setting up the Melissa component."""
with MockDependency("melissa") as mocked_melissa:
melissa.melissa = mocked_melissa
mocked_melissa.AsyncMelissa().async_connect = mock_coro_func()
with patch("melissa.AsyncMelissa") as mocked_melissa, patch.object(
melissa, "async_load_platform"
):
mocked_melissa.return_value.async_connect = AsyncMock()
await melissa.async_setup(hass, VALID_CONFIG)
mocked_melissa.AsyncMelissa.assert_called_with(
username="********", password="********"
)
mocked_melissa.assert_called_with(username="********", password="********")
assert melissa.DATA_MELISSA in hass.data
assert isinstance(
hass.data[melissa.DATA_MELISSA], type(mocked_melissa.AsyncMelissa())
hass.data[melissa.DATA_MELISSA], type(mocked_melissa.return_value),
)