jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-25 22:27:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update input component tests to async (#18290)

Browse Source
This commit is contained in:
Adam Mills 2018-11-07 03:56:32 -05:00 committed by Paulus Schoutsen
parent ce069be16e
commit 65be458ce0
5 changed files with 473 additions and 549 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

96
tests/components/test_input_boolean.py
View File

@ -1,65 +1,22 @@
"""The tests for the input_boolean component.""" """The tests for the input_boolean component."""
# pylint: disable=protected-access # pylint: disable=protected-access
import asyncio import asyncio
import unittest
import logging import logging
from homeassistant.core import CoreState, State, Context from homeassistant.core import CoreState, State, Context
from homeassistant.setup import setup_component, async_setup_component from homeassistant.setup import async_setup_component
from homeassistant.components.input_boolean import ( from homeassistant.components.input_boolean import (
is_on, CONF_INITIAL, DOMAIN) is_on, CONF_INITIAL, DOMAIN)
from homeassistant.const import ( from homeassistant.const import (
STATE_ON, STATE_OFF, ATTR_ENTITY_ID, ATTR_FRIENDLY_NAME, ATTR_ICON, STATE_ON, STATE_OFF, ATTR_ENTITY_ID, ATTR_FRIENDLY_NAME, ATTR_ICON,
SERVICE_TOGGLE, SERVICE_TURN_OFF, SERVICE_TURN_ON) SERVICE_TOGGLE, SERVICE_TURN_OFF, SERVICE_TURN_ON)
from homeassistant.loader import bind_hass
from tests.common import ( from tests.common import mock_component, mock_restore_cache
get_test_home_assistant, mock_component, mock_restore_cache)
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@bind_hass async def test_config(hass):
def toggle(hass, entity_id):
"""Set input_boolean to False.
This is a legacy helper method. Do not use it for new tests.
"""
hass.services.call(DOMAIN, SERVICE_TOGGLE, {ATTR_ENTITY_ID: entity_id})
@bind_hass
def turn_on(hass, entity_id):
"""Set input_boolean to True.
This is a legacy helper method. Do not use it for new tests.
"""
hass.services.call(DOMAIN, SERVICE_TURN_ON, {ATTR_ENTITY_ID: entity_id})
@bind_hass
def turn_off(hass, entity_id):
"""Set input_boolean to False.
This is a legacy helper method. Do not use it for new tests.
"""
hass.services.call(DOMAIN, SERVICE_TURN_OFF, {ATTR_ENTITY_ID: entity_id})
class TestInputBoolean(unittest.TestCase):
"""Test the input boolean module."""
# pylint: disable=invalid-name
def setUp(self):
"""Set up things to be run when tests are started."""
self.hass = get_test_home_assistant()
# pylint: disable=invalid-name
def tearDown(self):
"""Stop everything that was started."""
self.hass.stop()
def test_config(self):
"""Test config.""" """Test config."""
invalid_configs = [ invalid_configs = [
None, None,
@ -69,42 +26,47 @@ class TestInputBoolean(unittest.TestCase):
] ]
for cfg in invalid_configs: for cfg in invalid_configs:
assert not setup_component(self.hass, DOMAIN, {DOMAIN: cfg}) assert not await async_setup_component(hass, DOMAIN, {DOMAIN: cfg})
def test_methods(self):
async def test_methods(hass):
"""Test is_on, turn_on, turn_off methods.""" """Test is_on, turn_on, turn_off methods."""
assert setup_component(self.hass, DOMAIN, {DOMAIN: { assert await async_setup_component(hass, DOMAIN, {DOMAIN: {
'test_1': None, 'test_1': None,
}}) }})
entity_id = 'input_boolean.test_1' entity_id = 'input_boolean.test_1'
assert not is_on(self.hass, entity_id) assert not is_on(hass, entity_id)
turn_on(self.hass, entity_id) await hass.services.async_call(
DOMAIN, SERVICE_TURN_ON, {ATTR_ENTITY_ID: entity_id})
self.hass.block_till_done() await hass.async_block_till_done()
assert is_on(self.hass, entity_id) assert is_on(hass, entity_id)
turn_off(self.hass, entity_id) await hass.services.async_call(
DOMAIN, SERVICE_TURN_OFF, {ATTR_ENTITY_ID: entity_id})
self.hass.block_till_done() await hass.async_block_till_done()
assert not is_on(self.hass, entity_id) assert not is_on(hass, entity_id)
toggle(self.hass, entity_id) await hass.services.async_call(
DOMAIN, SERVICE_TOGGLE, {ATTR_ENTITY_ID: entity_id})
self.hass.block_till_done() await hass.async_block_till_done()
assert is_on(self.hass, entity_id) assert is_on(hass, entity_id)
def test_config_options(self):
async def test_config_options(hass):
"""Test configuration options.""" """Test configuration options."""
count_start = len(self.hass.states.entity_ids()) count_start = len(hass.states.async_entity_ids())
_LOGGER.debug('ENTITIES @ start: %s', self.hass.states.entity_ids()) _LOGGER.debug('ENTITIES @ start: %s', hass.states.async_entity_ids())
assert setup_component(self.hass, DOMAIN, {DOMAIN: { assert await async_setup_component(hass, DOMAIN, {DOMAIN: {
'test_1': None, 'test_1': None,
'test_2': { 'test_2': {
'name': 'Hello World', 'name': 'Hello World',
@ -113,12 +75,12 @@ class TestInputBoolean(unittest.TestCase):
}, },
}}) }})
_LOGGER.debug('ENTITIES: %s', self.hass.states.entity_ids()) _LOGGER.debug('ENTITIES: %s', hass.states.async_entity_ids())
assert count_start + 2 == len(self.hass.states.entity_ids()) assert count_start + 2 == len(hass.states.async_entity_ids())
state_1 = self.hass.states.get('input_boolean.test_1') state_1 = hass.states.get('input_boolean.test_1')
state_2 = self.hass.states.get('input_boolean.test_2') state_2 = hass.states.get('input_boolean.test_2')
assert state_1 is not None assert state_1 is not None
assert state_2 is not None assert state_2 is not None

22
tests/components/test_input_datetime.py
View File

@ -1,15 +1,14 @@
"""Tests for the Input slider component.""" """Tests for the Input slider component."""
# pylint: disable=protected-access # pylint: disable=protected-access
import asyncio import asyncio
import unittest
import datetime import datetime
from homeassistant.core import CoreState, State, Context from homeassistant.core import CoreState, State, Context
from homeassistant.setup import setup_component, async_setup_component from homeassistant.setup import async_setup_component
from homeassistant.components.input_datetime import ( from homeassistant.components.input_datetime import (
DOMAIN, ATTR_ENTITY_ID, ATTR_DATE, ATTR_TIME, SERVICE_SET_DATETIME) DOMAIN, ATTR_ENTITY_ID, ATTR_DATE, ATTR_TIME, SERVICE_SET_DATETIME)
from tests.common import get_test_home_assistant, mock_restore_cache from tests.common import mock_restore_cache
async def async_set_datetime(hass, entity_id, dt_value): async def async_set_datetime(hass, entity_id, dt_value):
@ -21,20 +20,7 @@ async def async_set_datetime(hass, entity_id, dt_value):
}, blocking=True) }, blocking=True)
class TestInputDatetime(unittest.TestCase): async def test_invalid_configs(hass):
"""Test the input datetime component."""
# pylint: disable=invalid-name
def setUp(self):
"""Set up things to be run when tests are started."""
self.hass = get_test_home_assistant()
# pylint: disable=invalid-name
def tearDown(self):
"""Stop everything that was started."""
self.hass.stop()
def test_invalid_configs(self):
"""Test config.""" """Test config."""
invalid_configs = [ invalid_configs = [
None, None,
@ -46,7 +32,7 @@ class TestInputDatetime(unittest.TestCase):
}}, }},
] ]
for cfg in invalid_configs: for cfg in invalid_configs:
assert not setup_component(self.hass, DOMAIN, {DOMAIN: cfg}) assert not await async_setup_component(hass, DOMAIN, {DOMAIN: cfg})
@asyncio.coroutine @asyncio.coroutine

111
tests/components/test_input_number.py
View File

@ -1,7 +1,6 @@
"""The tests for the Input number component.""" """The tests for the Input number component."""
# pylint: disable=protected-access # pylint: disable=protected-access
import asyncio import asyncio
import unittest
from homeassistant.core import CoreState, State, Context from homeassistant.core import CoreState, State, Context
from homeassistant.components.input_number import ( from homeassistant.components.input_number import (
@ -9,9 +8,9 @@ from homeassistant.components.input_number import (
SERVICE_SET_VALUE) SERVICE_SET_VALUE)
from homeassistant.const import ATTR_ENTITY_ID from homeassistant.const import ATTR_ENTITY_ID
from homeassistant.loader import bind_hass from homeassistant.loader import bind_hass
from homeassistant.setup import setup_component, async_setup_component from homeassistant.setup import async_setup_component
from tests.common import get_test_home_assistant, mock_restore_cache from tests.common import mock_restore_cache
@bind_hass @bind_hass
@ -20,10 +19,11 @@ def set_value(hass, entity_id, value):
This is a legacy helper method. Do not use it for new tests. This is a legacy helper method. Do not use it for new tests.
""" """
hass.services.call(DOMAIN, SERVICE_SET_VALUE, { hass.async_create_task(hass.services.async_call(
DOMAIN, SERVICE_SET_VALUE, {
ATTR_ENTITY_ID: entity_id, ATTR_ENTITY_ID: entity_id,
ATTR_VALUE: value, ATTR_VALUE: value,
}) }))
@bind_hass @bind_hass
@ -32,9 +32,10 @@ def increment(hass, entity_id):
This is a legacy helper method. Do not use it for new tests. This is a legacy helper method. Do not use it for new tests.
""" """
hass.services.call(DOMAIN, SERVICE_INCREMENT, { hass.async_create_task(hass.services.async_call(
DOMAIN, SERVICE_INCREMENT, {
ATTR_ENTITY_ID: entity_id ATTR_ENTITY_ID: entity_id
}) }))
@bind_hass @bind_hass
@ -43,25 +44,13 @@ def decrement(hass, entity_id):
This is a legacy helper method. Do not use it for new tests. This is a legacy helper method. Do not use it for new tests.
""" """
hass.services.call(DOMAIN, SERVICE_DECREMENT, { hass.async_create_task(hass.services.async_call(
DOMAIN, SERVICE_DECREMENT, {
ATTR_ENTITY_ID: entity_id ATTR_ENTITY_ID: entity_id
}) }))
class TestInputNumber(unittest.TestCase): async def test_config(hass):
"""Test the input number component."""
# pylint: disable=invalid-name
def setUp(self):
"""Set up things to be run when tests are started."""
self.hass = get_test_home_assistant()
# pylint: disable=invalid-name
def tearDown(self):
"""Stop everything that was started."""
self.hass.stop()
def test_config(self):
"""Test config.""" """Test config."""
invalid_configs = [ invalid_configs = [
None, None,
@ -73,11 +62,12 @@ class TestInputNumber(unittest.TestCase):
}}, }},
] ]
for cfg in invalid_configs: for cfg in invalid_configs:
assert not setup_component(self.hass, DOMAIN, {DOMAIN: cfg}) assert not await async_setup_component(hass, DOMAIN, {DOMAIN: cfg})
def test_set_value(self):
async def test_set_value(hass):
"""Test set_value method.""" """Test set_value method."""
assert setup_component(self.hass, DOMAIN, {DOMAIN: { assert await async_setup_component(hass, DOMAIN, {DOMAIN: {
'test_1': { 'test_1': {
'initial': 50, 'initial': 50,
'min': 0, 'min': 0,
@ -86,30 +76,31 @@ class TestInputNumber(unittest.TestCase):
}}) }})
entity_id = 'input_number.test_1' entity_id = 'input_number.test_1'
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 50 == float(state.state) assert 50 == float(state.state)
set_value(self.hass, entity_id, '30.4') set_value(hass, entity_id, '30.4')
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 30.4 == float(state.state) assert 30.4 == float(state.state)
set_value(self.hass, entity_id, '70') set_value(hass, entity_id, '70')
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 70 == float(state.state) assert 70 == float(state.state)
set_value(self.hass, entity_id, '110') set_value(hass, entity_id, '110')
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 70 == float(state.state) assert 70 == float(state.state)
def test_increment(self):
async def test_increment(hass):
"""Test increment method.""" """Test increment method."""
assert setup_component(self.hass, DOMAIN, {DOMAIN: { assert await async_setup_component(hass, DOMAIN, {DOMAIN: {
'test_2': { 'test_2': {
'initial': 50, 'initial': 50,
'min': 0, 'min': 0,
@ -118,24 +109,25 @@ class TestInputNumber(unittest.TestCase):
}}) }})
entity_id = 'input_number.test_2' entity_id = 'input_number.test_2'
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 50 == float(state.state) assert 50 == float(state.state)
increment(self.hass, entity_id) increment(hass, entity_id)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 51 == float(state.state) assert 51 == float(state.state)
increment(self.hass, entity_id) increment(hass, entity_id)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 51 == float(state.state) assert 51 == float(state.state)
def test_decrement(self):
async def test_decrement(hass):
"""Test decrement method.""" """Test decrement method."""
assert setup_component(self.hass, DOMAIN, {DOMAIN: { assert await async_setup_component(hass, DOMAIN, {DOMAIN: {
'test_3': { 'test_3': {
'initial': 50, 'initial': 50,
'min': 49, 'min': 49,
@ -144,24 +136,25 @@ class TestInputNumber(unittest.TestCase):
}}) }})
entity_id = 'input_number.test_3' entity_id = 'input_number.test_3'
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 50 == float(state.state) assert 50 == float(state.state)
decrement(self.hass, entity_id) decrement(hass, entity_id)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 49 == float(state.state) assert 49 == float(state.state)
decrement(self.hass, entity_id) decrement(hass, entity_id)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 49 == float(state.state) assert 49 == float(state.state)
def test_mode(self):
async def test_mode(hass):
"""Test mode settings.""" """Test mode settings."""
assert setup_component(self.hass, DOMAIN, {DOMAIN: { assert await async_setup_component(hass, DOMAIN, {DOMAIN: {
'test_default_slider': { 'test_default_slider': {
'min': 0, 'min': 0,
'max': 100, 'max': 100,
@ -178,15 +171,15 @@ class TestInputNumber(unittest.TestCase):
}, },
}}) }})
state = self.hass.states.get('input_number.test_default_slider') state = hass.states.get('input_number.test_default_slider')
assert state assert state
assert 'slider' == state.attributes['mode'] assert 'slider' == state.attributes['mode']
state = self.hass.states.get('input_number.test_explicit_box') state = hass.states.get('input_number.test_explicit_box')
assert state assert state
assert 'box' == state.attributes['mode'] assert 'box' == state.attributes['mode']
state = self.hass.states.get('input_number.test_explicit_slider') state = hass.states.get('input_number.test_explicit_slider')
assert state assert state
assert 'slider' == state.attributes['mode'] assert 'slider' == state.attributes['mode']

132
tests/components/test_input_select.py
View File

@ -1,7 +1,6 @@
"""The tests for the Input select component.""" """The tests for the Input select component."""
# pylint: disable=protected-access # pylint: disable=protected-access
import asyncio import asyncio
import unittest
from homeassistant.loader import bind_hass from homeassistant.loader import bind_hass
from homeassistant.components.input_select import ( from homeassistant.components.input_select import (
@ -10,9 +9,9 @@ from homeassistant.components.input_select import (
from homeassistant.const import ( from homeassistant.const import (
ATTR_ENTITY_ID, ATTR_FRIENDLY_NAME, ATTR_ICON) ATTR_ENTITY_ID, ATTR_FRIENDLY_NAME, ATTR_ICON)
from homeassistant.core import State, Context from homeassistant.core import State, Context
from homeassistant.setup import setup_component, async_setup_component from homeassistant.setup import async_setup_component
from tests.common import get_test_home_assistant, mock_restore_cache from tests.common import mock_restore_cache
@bind_hass @bind_hass
@ -21,10 +20,11 @@ def select_option(hass, entity_id, option):
This is a legacy helper method. Do not use it for new tests. This is a legacy helper method. Do not use it for new tests.
""" """
hass.services.call(DOMAIN, SERVICE_SELECT_OPTION, { hass.async_create_task(hass.services.async_call(
DOMAIN, SERVICE_SELECT_OPTION, {
ATTR_ENTITY_ID: entity_id, ATTR_ENTITY_ID: entity_id,
ATTR_OPTION: option, ATTR_OPTION: option,
}) }))
@bind_hass @bind_hass
@ -33,9 +33,10 @@ def select_next(hass, entity_id):
This is a legacy helper method. Do not use it for new tests. This is a legacy helper method. Do not use it for new tests.
""" """
hass.services.call(DOMAIN, SERVICE_SELECT_NEXT, { hass.async_create_task(hass.services.async_call(
DOMAIN, SERVICE_SELECT_NEXT, {
ATTR_ENTITY_ID: entity_id, ATTR_ENTITY_ID: entity_id,
}) }))
@bind_hass @bind_hass
@ -44,25 +45,13 @@ def select_previous(hass, entity_id):
This is a legacy helper method. Do not use it for new tests. This is a legacy helper method. Do not use it for new tests.
""" """
hass.services.call(DOMAIN, SERVICE_SELECT_PREVIOUS, { hass.async_create_task(hass.services.async_call(
DOMAIN, SERVICE_SELECT_PREVIOUS, {
ATTR_ENTITY_ID: entity_id, ATTR_ENTITY_ID: entity_id,
}) }))
class TestInputSelect(unittest.TestCase): async def test_config(hass):
"""Test the input select component."""
# pylint: disable=invalid-name
def setUp(self):
"""Set up things to be run when tests are started."""
self.hass = get_test_home_assistant()
# pylint: disable=invalid-name
def tearDown(self):
"""Stop everything that was started."""
self.hass.stop()
def test_config(self):
"""Test config.""" """Test config."""
invalid_configs = [ invalid_configs = [
None, None,
@ -76,11 +65,12 @@ class TestInputSelect(unittest.TestCase):
] ]
for cfg in invalid_configs: for cfg in invalid_configs:
assert not setup_component(self.hass, DOMAIN, {DOMAIN: cfg}) assert not await async_setup_component(hass, DOMAIN, {DOMAIN: cfg})
def test_select_option(self):
async def test_select_option(hass):
"""Test select_option methods.""" """Test select_option methods."""
assert setup_component(self.hass, DOMAIN, {DOMAIN: { assert await async_setup_component(hass, DOMAIN, {DOMAIN: {
'test_1': { 'test_1': {
'options': [ 'options': [
'some option', 'some option',
@ -90,24 +80,25 @@ class TestInputSelect(unittest.TestCase):
}}) }})
entity_id = 'input_select.test_1' entity_id = 'input_select.test_1'
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'some option' == state.state assert 'some option' == state.state
select_option(self.hass, entity_id, 'another option') select_option(hass, entity_id, 'another option')
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'another option' == state.state assert 'another option' == state.state
select_option(self.hass, entity_id, 'non existing option') select_option(hass, entity_id, 'non existing option')
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'another option' == state.state assert 'another option' == state.state
def test_select_next(self):
async def test_select_next(hass):
"""Test select_next methods.""" """Test select_next methods."""
assert setup_component(self.hass, DOMAIN, {DOMAIN: { assert await async_setup_component(hass, DOMAIN, {DOMAIN: {
'test_1': { 'test_1': {
'options': [ 'options': [
'first option', 'first option',
@ -119,24 +110,25 @@ class TestInputSelect(unittest.TestCase):
}}) }})
entity_id = 'input_select.test_1' entity_id = 'input_select.test_1'
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'middle option' == state.state assert 'middle option' == state.state
select_next(self.hass, entity_id) select_next(hass, entity_id)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'last option' == state.state assert 'last option' == state.state
select_next(self.hass, entity_id) select_next(hass, entity_id)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'first option' == state.state assert 'first option' == state.state
def test_select_previous(self):
async def test_select_previous(hass):
"""Test select_previous methods.""" """Test select_previous methods."""
assert setup_component(self.hass, DOMAIN, {DOMAIN: { assert await async_setup_component(hass, DOMAIN, {DOMAIN: {
'test_1': { 'test_1': {
'options': [ 'options': [
'first option', 'first option',
@ -148,24 +140,25 @@ class TestInputSelect(unittest.TestCase):
}}) }})
entity_id = 'input_select.test_1' entity_id = 'input_select.test_1'
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'middle option' == state.state assert 'middle option' == state.state
select_previous(self.hass, entity_id) select_previous(hass, entity_id)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'first option' == state.state assert 'first option' == state.state
select_previous(self.hass, entity_id) select_previous(hass, entity_id)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'last option' == state.state assert 'last option' == state.state
def test_config_options(self):
async def test_config_options(hass):
"""Test configuration options.""" """Test configuration options."""
count_start = len(self.hass.states.entity_ids()) count_start = len(hass.states.async_entity_ids())
test_2_options = [ test_2_options = [
'Good Option', 'Good Option',
@ -173,7 +166,7 @@ class TestInputSelect(unittest.TestCase):
'Best Option', 'Best Option',
] ]
assert setup_component(self.hass, DOMAIN, { assert await async_setup_component(hass, DOMAIN, {
DOMAIN: { DOMAIN: {
'test_1': { 'test_1': {
'options': [ 'options': [
@ -190,10 +183,10 @@ class TestInputSelect(unittest.TestCase):
} }
}) })
assert count_start + 2 == len(self.hass.states.entity_ids()) assert count_start + 2 == len(hass.states.async_entity_ids())
state_1 = self.hass.states.get('input_select.test_1') state_1 = hass.states.get('input_select.test_1')
state_2 = self.hass.states.get('input_select.test_2') state_2 = hass.states.get('input_select.test_2')
assert state_1 is not None assert state_1 is not None
assert state_2 is not None assert state_2 is not None
@ -210,9 +203,10 @@ class TestInputSelect(unittest.TestCase):
state_2.attributes.get(ATTR_FRIENDLY_NAME) state_2.attributes.get(ATTR_FRIENDLY_NAME)
assert 'mdi:work' == state_2.attributes.get(ATTR_ICON) assert 'mdi:work' == state_2.attributes.get(ATTR_ICON)
def test_set_options_service(self):
async def test_set_options_service(hass):
"""Test set_options service.""" """Test set_options service."""
assert setup_component(self.hass, DOMAIN, {DOMAIN: { assert await async_setup_component(hass, DOMAIN, {DOMAIN: {
'test_1': { 'test_1': {
'options': [ 'options': [
'first option', 'first option',
@ -224,24 +218,24 @@ class TestInputSelect(unittest.TestCase):
}}) }})
entity_id = 'input_select.test_1' entity_id = 'input_select.test_1'
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'middle option' == state.state assert 'middle option' == state.state
data = {ATTR_OPTIONS: ["test1", "test2"], "entity_id": entity_id} data = {ATTR_OPTIONS: ["test1", "test2"], "entity_id": entity_id}
self.hass.services.call(DOMAIN, SERVICE_SET_OPTIONS, data) await hass.services.async_call(DOMAIN, SERVICE_SET_OPTIONS, data)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'test1' == state.state assert 'test1' == state.state
select_option(self.hass, entity_id, 'first option') select_option(hass, entity_id, 'first option')
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'test1' == state.state assert 'test1' == state.state
select_option(self.hass, entity_id, 'test2') select_option(hass, entity_id, 'test2')
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'test2' == state.state assert 'test2' == state.state

57
tests/components/test_input_text.py
View File

@ -1,16 +1,15 @@
"""The tests for the Input text component.""" """The tests for the Input text component."""
# pylint: disable=protected-access # pylint: disable=protected-access
import asyncio import asyncio
import unittest
from homeassistant.components.input_text import ( from homeassistant.components.input_text import (
ATTR_VALUE, DOMAIN, SERVICE_SET_VALUE) ATTR_VALUE, DOMAIN, SERVICE_SET_VALUE)
from homeassistant.const import ATTR_ENTITY_ID from homeassistant.const import ATTR_ENTITY_ID
from homeassistant.core import CoreState, State, Context from homeassistant.core import CoreState, State, Context
from homeassistant.loader import bind_hass from homeassistant.loader import bind_hass
from homeassistant.setup import setup_component, async_setup_component from homeassistant.setup import async_setup_component
from tests.common import get_test_home_assistant, mock_restore_cache from tests.common import mock_restore_cache
@bind_hass @bind_hass
@ -19,26 +18,14 @@ def set_value(hass, entity_id, value):
This is a legacy helper method. Do not use it for new tests. This is a legacy helper method. Do not use it for new tests.
""" """
hass.services.call(DOMAIN, SERVICE_SET_VALUE, { hass.async_create_task(hass.services.async_call(
DOMAIN, SERVICE_SET_VALUE, {
ATTR_ENTITY_ID: entity_id, ATTR_ENTITY_ID: entity_id,
ATTR_VALUE: value, ATTR_VALUE: value,
}) }))
class TestInputText(unittest.TestCase): async def test_config(hass):
"""Test the input slider component."""
# pylint: disable=invalid-name
def setUp(self):
"""Set up things to be run when tests are started."""
self.hass = get_test_home_assistant()
# pylint: disable=invalid-name
def tearDown(self):
"""Stop everything that was started."""
self.hass.stop()
def test_config(self):
"""Test config.""" """Test config."""
invalid_configs = [ invalid_configs = [
None, None,
@ -50,11 +37,12 @@ class TestInputText(unittest.TestCase):
}}, }},
] ]
for cfg in invalid_configs: for cfg in invalid_configs:
assert not setup_component(self.hass, DOMAIN, {DOMAIN: cfg}) assert not await async_setup_component(hass, DOMAIN, {DOMAIN: cfg})
def test_set_value(self):
async def test_set_value(hass):
"""Test set_value method.""" """Test set_value method."""
assert setup_component(self.hass, DOMAIN, {DOMAIN: { assert await async_setup_component(hass, DOMAIN, {DOMAIN: {
'test_1': { 'test_1': {
'initial': 'test', 'initial': 'test',
'min': 3, 'min': 3,
@ -63,24 +51,25 @@ class TestInputText(unittest.TestCase):
}}) }})
entity_id = 'input_text.test_1' entity_id = 'input_text.test_1'
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'test' == str(state.state) assert 'test' == str(state.state)
set_value(self.hass, entity_id, 'testing') set_value(hass, entity_id, 'testing')
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'testing' == str(state.state) assert 'testing' == str(state.state)
set_value(self.hass, entity_id, 'testing too long') set_value(hass, entity_id, 'testing too long')
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(entity_id) state = hass.states.get(entity_id)
assert 'testing' == str(state.state) assert 'testing' == str(state.state)
def test_mode(self):
async def test_mode(hass):
"""Test mode settings.""" """Test mode settings."""
assert setup_component(self.hass, DOMAIN, {DOMAIN: { assert await async_setup_component(hass, DOMAIN, {DOMAIN: {
'test_default_text': { 'test_default_text': {
'initial': 'test', 'initial': 'test',
'min': 3, 'min': 3,
@ -100,15 +89,15 @@ class TestInputText(unittest.TestCase):
}, },
}}) }})
state = self.hass.states.get('input_text.test_default_text') state = hass.states.get('input_text.test_default_text')
assert state assert state
assert 'text' == state.attributes['mode'] assert 'text' == state.attributes['mode']
state = self.hass.states.get('input_text.test_explicit_text') state = hass.states.get('input_text.test_explicit_text')
assert state assert state
assert 'text' == state.attributes['mode'] assert 'text' == state.attributes['mode']
state = self.hass.states.get('input_text.test_explicit_password') state = hass.states.get('input_text.test_explicit_password')
assert state assert state
assert 'password' == state.attributes['mode'] assert 'password' == state.attributes['mode']