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

Black

Browse Source
This commit is contained in:
Paulus Schoutsen
2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
tests/components/remote/test_init.py
View File

@@ -4,16 +4,21 @@
import unittest
from homeassistant.const import (
ATTR_ENTITY_ID, STATE_ON, STATE_OFF, CONF_PLATFORM,
SERVICE_TURN_ON, SERVICE_TURN_OFF)
ATTR_ENTITY_ID,
STATE_ON,
STATE_OFF,
CONF_PLATFORM,
SERVICE_TURN_ON,
SERVICE_TURN_OFF,
)
import homeassistant.components.remote as remote
from tests.common import mock_service, get_test_home_assistant
from tests.components.remote import common
TEST_PLATFORM = {remote.DOMAIN: {CONF_PLATFORM: 'test'}}
SERVICE_SEND_COMMAND = 'send_command'
SERVICE_LEARN_COMMAND = 'learn_command'
TEST_PLATFORM = {remote.DOMAIN: {CONF_PLATFORM: "test"}}
SERVICE_SEND_COMMAND = "send_command"
SERVICE_LEARN_COMMAND = "learn_command"
class TestRemote(unittest.TestCase):
@@ -31,11 +36,11 @@ class TestRemote(unittest.TestCase):
def test_is_on(self):
"""Test is_on."""
self.hass.states.set('remote.test', STATE_ON)
assert remote.is_on(self.hass, 'remote.test')
self.hass.states.set("remote.test", STATE_ON)
assert remote.is_on(self.hass, "remote.test")
self.hass.states.set('remote.test', STATE_OFF)
assert not remote.is_on(self.hass, 'remote.test')
self.hass.states.set("remote.test", STATE_OFF)
assert not remote.is_on(self.hass, "remote.test")
self.hass.states.set(remote.ENTITY_ID_ALL_REMOTES, STATE_ON)
assert remote.is_on(self.hass)
@@ -45,12 +50,9 @@ class TestRemote(unittest.TestCase):
def test_turn_on(self):
"""Test turn_on."""
turn_on_calls = mock_service(
self.hass, remote.DOMAIN, SERVICE_TURN_ON)
turn_on_calls = mock_service(self.hass, remote.DOMAIN, SERVICE_TURN_ON)
common.turn_on(
self.hass,
entity_id='entity_id_val')
common.turn_on(self.hass, entity_id="entity_id_val")
self.hass.block_till_done()
@@ -61,11 +63,9 @@ class TestRemote(unittest.TestCase):
def test_turn_off(self):
"""Test turn_off."""
turn_off_calls = mock_service(
self.hass, remote.DOMAIN, SERVICE_TURN_OFF)
turn_off_calls = mock_service(self.hass, remote.DOMAIN, SERVICE_TURN_OFF)
common.turn_off(
self.hass, entity_id='entity_id_val')
common.turn_off(self.hass, entity_id="entity_id_val")
self.hass.block_till_done()
@@ -74,17 +74,22 @@ class TestRemote(unittest.TestCase):
assert call.domain == remote.DOMAIN
assert call.service == SERVICE_TURN_OFF
assert call.data[ATTR_ENTITY_ID] == 'entity_id_val'
assert call.data[ATTR_ENTITY_ID] == "entity_id_val"
def test_send_command(self):
"""Test send_command."""
send_command_calls = mock_service(
self.hass, remote.DOMAIN, SERVICE_SEND_COMMAND)
self.hass, remote.DOMAIN, SERVICE_SEND_COMMAND
)
common.send_command(
self.hass, entity_id='entity_id_val',
device='test_device', command=['test_command'],
num_repeats='4', delay_secs='0.6')
self.hass,
entity_id="entity_id_val",
device="test_device",
command=["test_command"],
num_repeats="4",
delay_secs="0.6",
)
self.hass.block_till_done()
@@ -93,17 +98,22 @@ class TestRemote(unittest.TestCase):
assert call.domain == remote.DOMAIN
assert call.service == SERVICE_SEND_COMMAND
assert call.data[ATTR_ENTITY_ID] == 'entity_id_val'
assert call.data[ATTR_ENTITY_ID] == "entity_id_val"
def test_learn_command(self):
"""Test learn_command."""
learn_command_calls = mock_service(
self.hass, remote.DOMAIN, SERVICE_LEARN_COMMAND)
self.hass, remote.DOMAIN, SERVICE_LEARN_COMMAND
)
common.learn_command(
self.hass, entity_id='entity_id_val',
device='test_device', command=['test_command'],
alternative=True, timeout=20)
self.hass,
entity_id="entity_id_val",
device="test_device",
command=["test_command"],
alternative=True,
timeout=20,
)
self.hass.block_till_done()
@@ -112,4 +122,4 @@ class TestRemote(unittest.TestCase):
assert call.domain == remote.DOMAIN
assert call.service == SERVICE_LEARN_COMMAND
assert call.data[ATTR_ENTITY_ID] == 'entity_id_val'
assert call.data[ATTR_ENTITY_ID] == "entity_id_val"