49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
"""The tests for SleepIQ sensor platform."""
|
|
from unittest.mock import MagicMock
|
|
|
|
import homeassistant.components.sleepiq.sensor as sleepiq
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.components.sleepiq.test_init import mock_responses
|
|
|
|
CONFIG = {"username": "foo", "password": "bar"}
|
|
|
|
|
|
async def test_setup(hass, requests_mock):
|
|
"""Test for successfully setting up the SleepIQ platform."""
|
|
mock_responses(requests_mock)
|
|
|
|
assert await async_setup_component(hass, "sleepiq", {"sleepiq": CONFIG})
|
|
|
|
device_mock = MagicMock()
|
|
sleepiq.setup_platform(hass, CONFIG, device_mock, MagicMock())
|
|
devices = device_mock.call_args[0][0]
|
|
assert len(devices) == 2
|
|
|
|
left_side = devices[1]
|
|
left_side.hass = hass
|
|
assert left_side.name == "SleepNumber ILE Test1 SleepNumber"
|
|
assert left_side.state == 40
|
|
|
|
right_side = devices[0]
|
|
right_side.hass = hass
|
|
assert right_side.name == "SleepNumber ILE Test2 SleepNumber"
|
|
assert right_side.state == 80
|
|
|
|
|
|
async def test_setup_single(hass, requests_mock):
|
|
"""Test for successfully setting up the SleepIQ platform."""
|
|
mock_responses(requests_mock, single=True)
|
|
|
|
assert await async_setup_component(hass, "sleepiq", {"sleepiq": CONFIG})
|
|
|
|
device_mock = MagicMock()
|
|
sleepiq.setup_platform(hass, CONFIG, device_mock, MagicMock())
|
|
devices = device_mock.call_args[0][0]
|
|
assert len(devices) == 1
|
|
|
|
right_side = devices[0]
|
|
right_side.hass = hass
|
|
assert right_side.name == "SleepNumber ILE Test1 SleepNumber"
|
|
assert right_side.state == 40
|