Black
This commit is contained in:
Paulus Schoutsen
parent
da05dfe708
commit
4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
|
|
@ -14,26 +14,35 @@ import pytz
|
|||
import pytest
|
||||
|
||||
import homeassistant.core as ha
|
||||
from homeassistant.exceptions import (InvalidEntityFormatError,
|
||||
InvalidStateError)
|
||||
from homeassistant.exceptions import InvalidEntityFormatError, InvalidStateError
|
||||
from homeassistant.util.async_ import run_coroutine_threadsafe
|
||||
import homeassistant.util.dt as dt_util
|
||||
from homeassistant.util.unit_system import (METRIC_SYSTEM)
|
||||
from homeassistant.util.unit_system import METRIC_SYSTEM
|
||||
from homeassistant.const import (
|
||||
__version__, EVENT_STATE_CHANGED, ATTR_FRIENDLY_NAME, CONF_UNIT_SYSTEM,
|
||||
ATTR_NOW, EVENT_TIME_CHANGED, EVENT_TIMER_OUT_OF_SYNC, ATTR_SECONDS,
|
||||
EVENT_HOMEASSISTANT_STOP, EVENT_HOMEASSISTANT_CLOSE,
|
||||
EVENT_SERVICE_REGISTERED, EVENT_SERVICE_REMOVED, EVENT_CALL_SERVICE,
|
||||
EVENT_CORE_CONFIG_UPDATE)
|
||||
__version__,
|
||||
EVENT_STATE_CHANGED,
|
||||
ATTR_FRIENDLY_NAME,
|
||||
CONF_UNIT_SYSTEM,
|
||||
ATTR_NOW,
|
||||
EVENT_TIME_CHANGED,
|
||||
EVENT_TIMER_OUT_OF_SYNC,
|
||||
ATTR_SECONDS,
|
||||
EVENT_HOMEASSISTANT_STOP,
|
||||
EVENT_HOMEASSISTANT_CLOSE,
|
||||
EVENT_SERVICE_REGISTERED,
|
||||
EVENT_SERVICE_REMOVED,
|
||||
EVENT_CALL_SERVICE,
|
||||
EVENT_CORE_CONFIG_UPDATE,
|
||||
)
|
||||
|
||||
from tests.common import get_test_home_assistant, async_mock_service
|
||||
|
||||
PST = pytz.timezone('America/Los_Angeles')
|
||||
PST = pytz.timezone("America/Los_Angeles")
|
||||
|
||||
|
||||
def test_split_entity_id():
|
||||
"""Test split_entity_id."""
|
||||
assert ha.split_entity_id('domain.object_id') == ['domain', 'object_id']
|
||||
assert ha.split_entity_id("domain.object_id") == ["domain", "object_id"]
|
||||
|
||||
|
||||
def test_async_add_job_schedule_callback():
|
||||
|
|
@ -78,6 +87,7 @@ def test_async_add_job_schedule_partial_coroutinefunction(loop):
|
|||
|
||||
async def job():
|
||||
pass
|
||||
|
||||
partial = functools.partial(job)
|
||||
|
||||
ha.HomeAssistant.async_add_job(hass, partial)
|
||||
|
|
@ -145,11 +155,9 @@ def test_stage_shutdown():
|
|||
test_close = []
|
||||
test_all = []
|
||||
|
||||
hass.bus.listen(
|
||||
EVENT_HOMEASSISTANT_STOP, lambda event: test_stop.append(event))
|
||||
hass.bus.listen(
|
||||
EVENT_HOMEASSISTANT_CLOSE, lambda event: test_close.append(event))
|
||||
hass.bus.listen('*', lambda event: test_all.append(event))
|
||||
hass.bus.listen(EVENT_HOMEASSISTANT_STOP, lambda event: test_stop.append(event))
|
||||
hass.bus.listen(EVENT_HOMEASSISTANT_CLOSE, lambda event: test_close.append(event))
|
||||
hass.bus.listen("*", lambda event: test_all.append(event))
|
||||
|
||||
hass.stop()
|
||||
|
||||
|
|
@ -178,14 +186,13 @@ class TestHomeAssistant(unittest.TestCase):
|
|||
@asyncio.coroutine
|
||||
def test_coro():
|
||||
"""Test Coro."""
|
||||
call_count.append('call')
|
||||
call_count.append("call")
|
||||
|
||||
for _ in range(3):
|
||||
self.hass.add_job(test_coro())
|
||||
|
||||
run_coroutine_threadsafe(
|
||||
asyncio.wait(self.hass._pending_tasks),
|
||||
loop=self.hass.loop
|
||||
asyncio.wait(self.hass._pending_tasks), loop=self.hass.loop
|
||||
).result()
|
||||
|
||||
assert len(self.hass._pending_tasks) == 3
|
||||
|
|
@ -198,7 +205,7 @@ class TestHomeAssistant(unittest.TestCase):
|
|||
@asyncio.coroutine
|
||||
def test_coro():
|
||||
"""Test Coro."""
|
||||
call_count.append('call')
|
||||
call_count.append("call")
|
||||
|
||||
for _ in range(2):
|
||||
self.hass.add_job(test_coro())
|
||||
|
|
@ -209,8 +216,7 @@ class TestHomeAssistant(unittest.TestCase):
|
|||
yield from asyncio.sleep(0)
|
||||
yield from asyncio.sleep(0)
|
||||
|
||||
run_coroutine_threadsafe(
|
||||
wait_finish_callback(), self.hass.loop).result()
|
||||
run_coroutine_threadsafe(wait_finish_callback(), self.hass.loop).result()
|
||||
|
||||
assert len(self.hass._pending_tasks) == 2
|
||||
self.hass.block_till_done()
|
||||
|
|
@ -222,7 +228,7 @@ class TestHomeAssistant(unittest.TestCase):
|
|||
|
||||
def test_executor():
|
||||
"""Test executor."""
|
||||
call_count.append('call')
|
||||
call_count.append("call")
|
||||
|
||||
@asyncio.coroutine
|
||||
def wait_finish_callback():
|
||||
|
|
@ -233,8 +239,7 @@ class TestHomeAssistant(unittest.TestCase):
|
|||
for _ in range(2):
|
||||
self.hass.add_job(test_executor)
|
||||
|
||||
run_coroutine_threadsafe(
|
||||
wait_finish_callback(), self.hass.loop).result()
|
||||
run_coroutine_threadsafe(wait_finish_callback(), self.hass.loop).result()
|
||||
|
||||
assert len(self.hass._pending_tasks) == 2
|
||||
self.hass.block_till_done()
|
||||
|
|
@ -247,7 +252,7 @@ class TestHomeAssistant(unittest.TestCase):
|
|||
@ha.callback
|
||||
def test_callback():
|
||||
"""Test callback."""
|
||||
call_count.append('call')
|
||||
call_count.append("call")
|
||||
|
||||
@asyncio.coroutine
|
||||
def wait_finish_callback():
|
||||
|
|
@ -258,8 +263,7 @@ class TestHomeAssistant(unittest.TestCase):
|
|||
for _ in range(2):
|
||||
self.hass.add_job(test_callback)
|
||||
|
||||
run_coroutine_threadsafe(
|
||||
wait_finish_callback(), self.hass.loop).result()
|
||||
run_coroutine_threadsafe(wait_finish_callback(), self.hass.loop).result()
|
||||
|
||||
self.hass.block_till_done()
|
||||
|
||||
|
|
@ -269,7 +273,7 @@ class TestHomeAssistant(unittest.TestCase):
|
|||
def test_add_job_with_none(self):
|
||||
"""Try to add a job with None as function."""
|
||||
with pytest.raises(ValueError):
|
||||
self.hass.add_job(None, 'test_arg')
|
||||
self.hass.add_job(None, "test_arg")
|
||||
|
||||
|
||||
class TestEvent(unittest.TestCase):
|
||||
|
|
@ -278,10 +282,10 @@ class TestEvent(unittest.TestCase):
|
|||
def test_eq(self):
|
||||
"""Test events."""
|
||||
now = dt_util.utcnow()
|
||||
data = {'some': 'attr'}
|
||||
data = {"some": "attr"}
|
||||
context = ha.Context()
|
||||
event1, event2 = [
|
||||
ha.Event('some_type', data, time_fired=now, context=context)
|
||||
ha.Event("some_type", data, time_fired=now, context=context)
|
||||
for _ in range(2)
|
||||
]
|
||||
|
||||
|
|
@ -289,30 +293,28 @@ class TestEvent(unittest.TestCase):
|
|||
|
||||
def test_repr(self):
|
||||
"""Test that repr method works."""
|
||||
assert "<Event TestEvent[L]>" == \
|
||||
str(ha.Event("TestEvent"))
|
||||
assert "<Event TestEvent[L]>" == str(ha.Event("TestEvent"))
|
||||
|
||||
assert "<Event TestEvent[R]: beer=nice>" == \
|
||||
str(ha.Event("TestEvent",
|
||||
{"beer": "nice"},
|
||||
ha.EventOrigin.remote))
|
||||
assert "<Event TestEvent[R]: beer=nice>" == str(
|
||||
ha.Event("TestEvent", {"beer": "nice"}, ha.EventOrigin.remote)
|
||||
)
|
||||
|
||||
def test_as_dict(self):
|
||||
"""Test as dictionary."""
|
||||
event_type = 'some_type'
|
||||
event_type = "some_type"
|
||||
now = dt_util.utcnow()
|
||||
data = {'some': 'attr'}
|
||||
data = {"some": "attr"}
|
||||
|
||||
event = ha.Event(event_type, data, ha.EventOrigin.local, now)
|
||||
expected = {
|
||||
'event_type': event_type,
|
||||
'data': data,
|
||||
'origin': 'LOCAL',
|
||||
'time_fired': now,
|
||||
'context': {
|
||||
'id': event.context.id,
|
||||
'parent_id': None,
|
||||
'user_id': event.context.user_id,
|
||||
"event_type": event_type,
|
||||
"data": data,
|
||||
"origin": "LOCAL",
|
||||
"time_fired": now,
|
||||
"context": {
|
||||
"id": event.context.id,
|
||||
"parent_id": None,
|
||||
"user_id": event.context.user_id,
|
||||
},
|
||||
}
|
||||
assert expected == event.as_dict()
|
||||
|
|
@ -337,9 +339,10 @@ class TestEventBus(unittest.TestCase):
|
|||
self.hass.allow_pool = False
|
||||
old_count = len(self.bus.listeners)
|
||||
|
||||
def listener(_): pass
|
||||
def listener(_):
|
||||
pass
|
||||
|
||||
unsub = self.bus.listen('test', listener)
|
||||
unsub = self.bus.listen("test", listener)
|
||||
|
||||
assert old_count + 1 == len(self.bus.listeners)
|
||||
|
||||
|
|
@ -359,16 +362,16 @@ class TestEventBus(unittest.TestCase):
|
|||
"""Mock listener."""
|
||||
calls.append(event)
|
||||
|
||||
unsub = self.bus.listen('test', listener)
|
||||
unsub = self.bus.listen("test", listener)
|
||||
|
||||
self.bus.fire('test')
|
||||
self.bus.fire("test")
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(calls) == 1
|
||||
|
||||
unsub()
|
||||
|
||||
self.bus.fire('event')
|
||||
self.bus.fire("event")
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(calls) == 1
|
||||
|
|
@ -381,11 +384,11 @@ class TestEventBus(unittest.TestCase):
|
|||
def event_handler(event):
|
||||
runs.append(event)
|
||||
|
||||
self.bus.listen_once('test_event', event_handler)
|
||||
self.bus.listen_once("test_event", event_handler)
|
||||
|
||||
self.bus.fire('test_event')
|
||||
self.bus.fire("test_event")
|
||||
# Second time it should not increase runs
|
||||
self.bus.fire('test_event')
|
||||
self.bus.fire("test_event")
|
||||
|
||||
self.hass.block_till_done()
|
||||
assert 1 == len(runs)
|
||||
|
|
@ -398,11 +401,11 @@ class TestEventBus(unittest.TestCase):
|
|||
def event_handler(event):
|
||||
runs.append(event)
|
||||
|
||||
self.bus.listen_once('test_event', event_handler)
|
||||
self.bus.listen_once("test_event", event_handler)
|
||||
|
||||
self.bus.fire('test_event')
|
||||
self.bus.fire("test_event")
|
||||
# Second time it should not increase runs
|
||||
self.bus.fire('test_event')
|
||||
self.bus.fire("test_event")
|
||||
|
||||
self.hass.block_till_done()
|
||||
assert 1 == len(runs)
|
||||
|
|
@ -414,11 +417,11 @@ class TestEventBus(unittest.TestCase):
|
|||
def event_handler(event):
|
||||
runs.append(event)
|
||||
|
||||
self.bus.listen_once('test_event', event_handler)
|
||||
self.bus.listen_once("test_event", event_handler)
|
||||
|
||||
self.bus.fire('test_event')
|
||||
self.bus.fire("test_event")
|
||||
# Second time it should not increase runs
|
||||
self.bus.fire('test_event')
|
||||
self.bus.fire("test_event")
|
||||
|
||||
self.hass.block_till_done()
|
||||
assert 1 == len(runs)
|
||||
|
|
@ -430,8 +433,8 @@ class TestEventBus(unittest.TestCase):
|
|||
def thread_listener(event):
|
||||
thread_calls.append(event)
|
||||
|
||||
self.bus.listen('test_thread', thread_listener)
|
||||
self.bus.fire('test_thread')
|
||||
self.bus.listen("test_thread", thread_listener)
|
||||
self.bus.fire("test_thread")
|
||||
self.hass.block_till_done()
|
||||
assert len(thread_calls) == 1
|
||||
|
||||
|
|
@ -443,8 +446,8 @@ class TestEventBus(unittest.TestCase):
|
|||
def callback_listener(event):
|
||||
callback_calls.append(event)
|
||||
|
||||
self.bus.listen('test_callback', callback_listener)
|
||||
self.bus.fire('test_callback')
|
||||
self.bus.listen("test_callback", callback_listener)
|
||||
self.bus.fire("test_callback")
|
||||
self.hass.block_till_done()
|
||||
assert len(callback_calls) == 1
|
||||
|
||||
|
|
@ -456,8 +459,8 @@ class TestEventBus(unittest.TestCase):
|
|||
def coroutine_listener(event):
|
||||
coroutine_calls.append(event)
|
||||
|
||||
self.bus.listen('test_coroutine', coroutine_listener)
|
||||
self.bus.fire('test_coroutine')
|
||||
self.bus.listen("test_coroutine", coroutine_listener)
|
||||
self.bus.fire("test_coroutine")
|
||||
self.hass.block_till_done()
|
||||
assert len(coroutine_calls) == 1
|
||||
|
||||
|
|
@ -465,73 +468,78 @@ class TestEventBus(unittest.TestCase):
|
|||
def test_state_init():
|
||||
"""Test state.init."""
|
||||
with pytest.raises(InvalidEntityFormatError):
|
||||
ha.State('invalid_entity_format', 'test_state')
|
||||
ha.State("invalid_entity_format", "test_state")
|
||||
|
||||
with pytest.raises(InvalidStateError):
|
||||
ha.State('domain.long_state', 't' * 256)
|
||||
ha.State("domain.long_state", "t" * 256)
|
||||
|
||||
|
||||
def test_state_domain():
|
||||
"""Test domain."""
|
||||
state = ha.State('some_domain.hello', 'world')
|
||||
assert 'some_domain' == state.domain
|
||||
state = ha.State("some_domain.hello", "world")
|
||||
assert "some_domain" == state.domain
|
||||
|
||||
|
||||
def test_state_object_id():
|
||||
"""Test object ID."""
|
||||
state = ha.State('domain.hello', 'world')
|
||||
assert 'hello' == state.object_id
|
||||
state = ha.State("domain.hello", "world")
|
||||
assert "hello" == state.object_id
|
||||
|
||||
|
||||
def test_state_name_if_no_friendly_name_attr():
|
||||
"""Test if there is no friendly name."""
|
||||
state = ha.State('domain.hello_world', 'world')
|
||||
assert 'hello world' == state.name
|
||||
state = ha.State("domain.hello_world", "world")
|
||||
assert "hello world" == state.name
|
||||
|
||||
|
||||
def test_state_name_if_friendly_name_attr():
|
||||
"""Test if there is a friendly name."""
|
||||
name = 'Some Unique Name'
|
||||
state = ha.State('domain.hello_world', 'world',
|
||||
{ATTR_FRIENDLY_NAME: name})
|
||||
name = "Some Unique Name"
|
||||
state = ha.State("domain.hello_world", "world", {ATTR_FRIENDLY_NAME: name})
|
||||
assert name == state.name
|
||||
|
||||
|
||||
def test_state_dict_conversion():
|
||||
"""Test conversion of dict."""
|
||||
state = ha.State('domain.hello', 'world', {'some': 'attr'})
|
||||
state = ha.State("domain.hello", "world", {"some": "attr"})
|
||||
assert state == ha.State.from_dict(state.as_dict())
|
||||
|
||||
|
||||
def test_state_dict_conversion_with_wrong_data():
|
||||
"""Test conversion with wrong data."""
|
||||
assert ha.State.from_dict(None) is None
|
||||
assert ha.State.from_dict({'state': 'yes'}) is None
|
||||
assert ha.State.from_dict({'entity_id': 'yes'}) is None
|
||||
assert ha.State.from_dict({"state": "yes"}) is None
|
||||
assert ha.State.from_dict({"entity_id": "yes"}) is None
|
||||
# Make sure invalid context data doesn't crash
|
||||
wrong_context = ha.State.from_dict({
|
||||
'entity_id': 'light.kitchen',
|
||||
'state': 'on',
|
||||
'context': {
|
||||
'id': '123',
|
||||
'non-existing': 'crash'
|
||||
wrong_context = ha.State.from_dict(
|
||||
{
|
||||
"entity_id": "light.kitchen",
|
||||
"state": "on",
|
||||
"context": {"id": "123", "non-existing": "crash"},
|
||||
}
|
||||
})
|
||||
)
|
||||
assert wrong_context is not None
|
||||
assert wrong_context.context.id == '123'
|
||||
assert wrong_context.context.id == "123"
|
||||
|
||||
|
||||
def test_state_repr():
|
||||
"""Test state.repr."""
|
||||
assert "<state happy.happy=on @ 1984-12-08T12:00:00+00:00>" == \
|
||||
str(ha.State(
|
||||
"happy.happy", "on",
|
||||
last_changed=datetime(1984, 12, 8, 12, 0, 0)))
|
||||
assert "<state happy.happy=on @ 1984-12-08T12:00:00+00:00>" == str(
|
||||
ha.State("happy.happy", "on", last_changed=datetime(1984, 12, 8, 12, 0, 0))
|
||||
)
|
||||
|
||||
assert "<state happy.happy=on; brightness=144 @ " \
|
||||
"1984-12-08T12:00:00+00:00>" == \
|
||||
str(ha.State("happy.happy", "on", {"brightness": 144},
|
||||
datetime(1984, 12, 8, 12, 0, 0)))
|
||||
assert (
|
||||
"<state happy.happy=on; brightness=144 @ "
|
||||
"1984-12-08T12:00:00+00:00>"
|
||||
== str(
|
||||
ha.State(
|
||||
"happy.happy",
|
||||
"on",
|
||||
{"brightness": 144},
|
||||
datetime(1984, 12, 8, 12, 0, 0),
|
||||
)
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
class TestStateMachine(unittest.TestCase):
|
||||
|
|
@ -552,25 +560,25 @@ class TestStateMachine(unittest.TestCase):
|
|||
|
||||
def test_is_state(self):
|
||||
"""Test is_state method."""
|
||||
assert self.states.is_state('light.Bowl', 'on')
|
||||
assert not self.states.is_state('light.Bowl', 'off')
|
||||
assert not self.states.is_state('light.Non_existing', 'on')
|
||||
assert self.states.is_state("light.Bowl", "on")
|
||||
assert not self.states.is_state("light.Bowl", "off")
|
||||
assert not self.states.is_state("light.Non_existing", "on")
|
||||
|
||||
def test_entity_ids(self):
|
||||
"""Test get_entity_ids method."""
|
||||
ent_ids = self.states.entity_ids()
|
||||
assert 2 == len(ent_ids)
|
||||
assert 'light.bowl' in ent_ids
|
||||
assert 'switch.ac' in ent_ids
|
||||
assert "light.bowl" in ent_ids
|
||||
assert "switch.ac" in ent_ids
|
||||
|
||||
ent_ids = self.states.entity_ids('light')
|
||||
ent_ids = self.states.entity_ids("light")
|
||||
assert 1 == len(ent_ids)
|
||||
assert 'light.bowl' in ent_ids
|
||||
assert "light.bowl" in ent_ids
|
||||
|
||||
def test_all(self):
|
||||
"""Test everything."""
|
||||
states = sorted(state.entity_id for state in self.states.all())
|
||||
assert ['light.bowl', 'switch.ac'] == states
|
||||
assert ["light.bowl", "switch.ac"] == states
|
||||
|
||||
def test_remove(self):
|
||||
"""Test remove method."""
|
||||
|
|
@ -582,19 +590,19 @@ class TestStateMachine(unittest.TestCase):
|
|||
|
||||
self.hass.bus.listen(EVENT_STATE_CHANGED, callback)
|
||||
|
||||
assert 'light.bowl' in self.states.entity_ids()
|
||||
assert self.states.remove('light.bowl')
|
||||
assert "light.bowl" in self.states.entity_ids()
|
||||
assert self.states.remove("light.bowl")
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert 'light.bowl' not in self.states.entity_ids()
|
||||
assert "light.bowl" not in self.states.entity_ids()
|
||||
assert 1 == len(events)
|
||||
assert 'light.bowl' == events[0].data.get('entity_id')
|
||||
assert events[0].data.get('old_state') is not None
|
||||
assert 'light.bowl' == events[0].data['old_state'].entity_id
|
||||
assert events[0].data.get('new_state') is None
|
||||
assert "light.bowl" == events[0].data.get("entity_id")
|
||||
assert events[0].data.get("old_state") is not None
|
||||
assert "light.bowl" == events[0].data["old_state"].entity_id
|
||||
assert events[0].data.get("new_state") is None
|
||||
|
||||
# If it does not exist, we should get False
|
||||
assert not self.states.remove('light.Bowl')
|
||||
assert not self.states.remove("light.Bowl")
|
||||
self.hass.block_till_done()
|
||||
assert 1 == len(events)
|
||||
|
||||
|
|
@ -608,23 +616,23 @@ class TestStateMachine(unittest.TestCase):
|
|||
|
||||
self.hass.bus.listen(EVENT_STATE_CHANGED, callback)
|
||||
|
||||
self.states.set('light.BOWL', 'off')
|
||||
self.states.set("light.BOWL", "off")
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert self.states.is_state('light.bowl', 'off')
|
||||
assert self.states.is_state("light.bowl", "off")
|
||||
assert 1 == len(runs)
|
||||
|
||||
def test_last_changed_not_updated_on_same_state(self):
|
||||
"""Test to not update the existing, same state."""
|
||||
state = self.states.get('light.Bowl')
|
||||
state = self.states.get("light.Bowl")
|
||||
|
||||
future = dt_util.utcnow() + timedelta(hours=10)
|
||||
|
||||
with patch('homeassistant.util.dt.utcnow', return_value=future):
|
||||
self.states.set("light.Bowl", "on", {'attr': 'triggers_change'})
|
||||
with patch("homeassistant.util.dt.utcnow", return_value=future):
|
||||
self.states.set("light.Bowl", "on", {"attr": "triggers_change"})
|
||||
self.hass.block_till_done()
|
||||
|
||||
state2 = self.states.get('light.Bowl')
|
||||
state2 = self.states.get("light.Bowl")
|
||||
assert state2 is not None
|
||||
assert state.last_changed == state2.last_changed
|
||||
|
||||
|
|
@ -638,25 +646,26 @@ class TestStateMachine(unittest.TestCase):
|
|||
|
||||
self.hass.bus.listen(EVENT_STATE_CHANGED, callback)
|
||||
|
||||
self.states.set('light.bowl', 'on')
|
||||
self.states.set("light.bowl", "on")
|
||||
self.hass.block_till_done()
|
||||
assert 0 == len(events)
|
||||
|
||||
self.states.set('light.bowl', 'on', None, True)
|
||||
self.states.set("light.bowl", "on", None, True)
|
||||
self.hass.block_till_done()
|
||||
assert 1 == len(events)
|
||||
|
||||
|
||||
def test_service_call_repr():
|
||||
"""Test ServiceCall repr."""
|
||||
call = ha.ServiceCall('homeassistant', 'start')
|
||||
assert str(call) == \
|
||||
"<ServiceCall homeassistant.start (c:{})>".format(call.context.id)
|
||||
call = ha.ServiceCall("homeassistant", "start")
|
||||
assert str(call) == "<ServiceCall homeassistant.start (c:{})>".format(
|
||||
call.context.id
|
||||
)
|
||||
|
||||
call2 = ha.ServiceCall('homeassistant', 'start', {'fast': 'yes'})
|
||||
assert str(call2) == \
|
||||
"<ServiceCall homeassistant.start (c:{}): fast=yes>".format(
|
||||
call2.context.id)
|
||||
call2 = ha.ServiceCall("homeassistant", "start", {"fast": "yes"})
|
||||
assert str(call2) == "<ServiceCall homeassistant.start (c:{}): fast=yes>".format(
|
||||
call2.context.id
|
||||
)
|
||||
|
||||
|
||||
class TestServiceRegistry(unittest.TestCase):
|
||||
|
|
@ -707,22 +716,20 @@ class TestServiceRegistry(unittest.TestCase):
|
|||
"""Service handler."""
|
||||
calls.append(call)
|
||||
|
||||
self.services.register(
|
||||
"test_domain", "register_calls", service_handler)
|
||||
self.services.register("test_domain", "register_calls", service_handler)
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(self.calls_register) == 1
|
||||
assert self.calls_register[-1].data['domain'] == 'test_domain'
|
||||
assert self.calls_register[-1].data['service'] == 'register_calls'
|
||||
assert self.calls_register[-1].data["domain"] == "test_domain"
|
||||
assert self.calls_register[-1].data["service"] == "register_calls"
|
||||
|
||||
assert self.services.call('test_domain', 'REGISTER_CALLS',
|
||||
blocking=True)
|
||||
assert self.services.call("test_domain", "REGISTER_CALLS", blocking=True)
|
||||
assert 1 == len(calls)
|
||||
|
||||
def test_call_non_existing_with_blocking(self):
|
||||
"""Test non-existing with blocking."""
|
||||
with pytest.raises(ha.ServiceNotFound):
|
||||
self.services.call('test_domain', 'i_do_not_exist', blocking=True)
|
||||
self.services.call("test_domain", "i_do_not_exist", blocking=True)
|
||||
|
||||
def test_async_service(self):
|
||||
"""Test registering and calling an async service."""
|
||||
|
|
@ -732,16 +739,14 @@ class TestServiceRegistry(unittest.TestCase):
|
|||
"""Service handler coroutine."""
|
||||
calls.append(call)
|
||||
|
||||
self.services.register(
|
||||
'test_domain', 'register_calls', service_handler)
|
||||
self.services.register("test_domain", "register_calls", service_handler)
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(self.calls_register) == 1
|
||||
assert self.calls_register[-1].data['domain'] == 'test_domain'
|
||||
assert self.calls_register[-1].data['service'] == 'register_calls'
|
||||
assert self.calls_register[-1].data["domain"] == "test_domain"
|
||||
assert self.calls_register[-1].data["service"] == "register_calls"
|
||||
|
||||
assert self.services.call('test_domain', 'REGISTER_CALLS',
|
||||
blocking=True)
|
||||
assert self.services.call("test_domain", "REGISTER_CALLS", blocking=True)
|
||||
self.hass.block_till_done()
|
||||
assert 1 == len(calls)
|
||||
|
||||
|
|
@ -754,16 +759,15 @@ class TestServiceRegistry(unittest.TestCase):
|
|||
calls.append(call)
|
||||
|
||||
self.services.register(
|
||||
'test_domain', 'register_calls',
|
||||
functools.partial(service_handler))
|
||||
"test_domain", "register_calls", functools.partial(service_handler)
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(self.calls_register) == 1
|
||||
assert self.calls_register[-1].data['domain'] == 'test_domain'
|
||||
assert self.calls_register[-1].data['service'] == 'register_calls'
|
||||
assert self.calls_register[-1].data["domain"] == "test_domain"
|
||||
assert self.calls_register[-1].data["service"] == "register_calls"
|
||||
|
||||
assert self.services.call('test_domain', 'REGISTER_CALLS',
|
||||
blocking=True)
|
||||
assert self.services.call("test_domain", "REGISTER_CALLS", blocking=True)
|
||||
self.hass.block_till_done()
|
||||
assert len(calls) == 1
|
||||
|
||||
|
|
@ -776,16 +780,14 @@ class TestServiceRegistry(unittest.TestCase):
|
|||
"""Service handler coroutine."""
|
||||
calls.append(call)
|
||||
|
||||
self.services.register(
|
||||
'test_domain', 'register_calls', service_handler)
|
||||
self.services.register("test_domain", "register_calls", service_handler)
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(self.calls_register) == 1
|
||||
assert self.calls_register[-1].data['domain'] == 'test_domain'
|
||||
assert self.calls_register[-1].data['service'] == 'register_calls'
|
||||
assert self.calls_register[-1].data["domain"] == "test_domain"
|
||||
assert self.calls_register[-1].data["service"] == "register_calls"
|
||||
|
||||
assert self.services.call('test_domain', 'REGISTER_CALLS',
|
||||
blocking=True)
|
||||
assert self.services.call("test_domain", "REGISTER_CALLS", blocking=True)
|
||||
self.hass.block_till_done()
|
||||
assert 1 == len(calls)
|
||||
|
||||
|
|
@ -800,15 +802,15 @@ class TestServiceRegistry(unittest.TestCase):
|
|||
|
||||
self.hass.bus.listen(EVENT_SERVICE_REMOVED, mock_event_remove)
|
||||
|
||||
assert self.services.has_service('test_Domain', 'test_Service')
|
||||
assert self.services.has_service("test_Domain", "test_Service")
|
||||
|
||||
self.services.remove('test_Domain', 'test_Service')
|
||||
self.services.remove("test_Domain", "test_Service")
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert not self.services.has_service('test_Domain', 'test_Service')
|
||||
assert not self.services.has_service("test_Domain", "test_Service")
|
||||
assert len(calls_remove) == 1
|
||||
assert calls_remove[-1].data['domain'] == 'test_domain'
|
||||
assert calls_remove[-1].data['service'] == 'test_service'
|
||||
assert calls_remove[-1].data["domain"] == "test_domain"
|
||||
assert calls_remove[-1].data["service"] == "test_service"
|
||||
|
||||
def test_remove_service_that_not_exists(self):
|
||||
"""Test remove service that not exists."""
|
||||
|
|
@ -821,48 +823,46 @@ class TestServiceRegistry(unittest.TestCase):
|
|||
|
||||
self.hass.bus.listen(EVENT_SERVICE_REMOVED, mock_event_remove)
|
||||
|
||||
assert not self.services.has_service('test_xxx', 'test_yyy')
|
||||
self.services.remove('test_xxx', 'test_yyy')
|
||||
assert not self.services.has_service("test_xxx", "test_yyy")
|
||||
self.services.remove("test_xxx", "test_yyy")
|
||||
self.hass.block_till_done()
|
||||
assert len(calls_remove) == 0
|
||||
|
||||
def test_async_service_raise_exception(self):
|
||||
"""Test registering and calling an async service raise exception."""
|
||||
|
||||
async def service_handler(_):
|
||||
"""Service handler coroutine."""
|
||||
raise ValueError
|
||||
|
||||
self.services.register(
|
||||
'test_domain', 'register_calls', service_handler)
|
||||
self.services.register("test_domain", "register_calls", service_handler)
|
||||
self.hass.block_till_done()
|
||||
|
||||
with pytest.raises(ValueError):
|
||||
assert self.services.call('test_domain', 'REGISTER_CALLS',
|
||||
blocking=True)
|
||||
assert self.services.call("test_domain", "REGISTER_CALLS", blocking=True)
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Non-blocking service call never throw exception
|
||||
self.services.call('test_domain', 'REGISTER_CALLS', blocking=False)
|
||||
self.services.call("test_domain", "REGISTER_CALLS", blocking=False)
|
||||
self.hass.block_till_done()
|
||||
|
||||
def test_callback_service_raise_exception(self):
|
||||
"""Test registering and calling an callback service raise exception."""
|
||||
|
||||
@ha.callback
|
||||
def service_handler(_):
|
||||
"""Service handler coroutine."""
|
||||
raise ValueError
|
||||
|
||||
self.services.register(
|
||||
'test_domain', 'register_calls', service_handler)
|
||||
self.services.register("test_domain", "register_calls", service_handler)
|
||||
self.hass.block_till_done()
|
||||
|
||||
with pytest.raises(ValueError):
|
||||
assert self.services.call('test_domain', 'REGISTER_CALLS',
|
||||
blocking=True)
|
||||
assert self.services.call("test_domain", "REGISTER_CALLS", blocking=True)
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Non-blocking service call never throw exception
|
||||
self.services.call('test_domain', 'REGISTER_CALLS', blocking=False)
|
||||
self.services.call("test_domain", "REGISTER_CALLS", blocking=False)
|
||||
self.hass.block_till_done()
|
||||
|
||||
|
||||
|
|
@ -877,31 +877,29 @@ class TestConfig(unittest.TestCase):
|
|||
|
||||
def test_path_with_file(self):
|
||||
"""Test get_config_path method."""
|
||||
self.config.config_dir = '/tmp/ha-config'
|
||||
assert "/tmp/ha-config/test.conf" == \
|
||||
self.config.path("test.conf")
|
||||
self.config.config_dir = "/tmp/ha-config"
|
||||
assert "/tmp/ha-config/test.conf" == self.config.path("test.conf")
|
||||
|
||||
def test_path_with_dir_and_file(self):
|
||||
"""Test get_config_path method."""
|
||||
self.config.config_dir = '/tmp/ha-config'
|
||||
assert "/tmp/ha-config/dir/test.conf" == \
|
||||
self.config.path("dir", "test.conf")
|
||||
self.config.config_dir = "/tmp/ha-config"
|
||||
assert "/tmp/ha-config/dir/test.conf" == self.config.path("dir", "test.conf")
|
||||
|
||||
def test_as_dict(self):
|
||||
"""Test as dict."""
|
||||
self.config.config_dir = '/tmp/ha-config'
|
||||
self.config.config_dir = "/tmp/ha-config"
|
||||
expected = {
|
||||
'latitude': 0,
|
||||
'longitude': 0,
|
||||
'elevation': 0,
|
||||
"latitude": 0,
|
||||
"longitude": 0,
|
||||
"elevation": 0,
|
||||
CONF_UNIT_SYSTEM: METRIC_SYSTEM.as_dict(),
|
||||
'location_name': "Home",
|
||||
'time_zone': 'UTC',
|
||||
'components': set(),
|
||||
'config_dir': '/tmp/ha-config',
|
||||
'whitelist_external_dirs': set(),
|
||||
'version': __version__,
|
||||
'config_source': "default",
|
||||
"location_name": "Home",
|
||||
"time_zone": "UTC",
|
||||
"components": set(),
|
||||
"config_dir": "/tmp/ha-config",
|
||||
"whitelist_external_dirs": set(),
|
||||
"version": __version__,
|
||||
"config_source": "default",
|
||||
}
|
||||
|
||||
assert expected == self.config.as_dict()
|
||||
|
|
@ -911,23 +909,17 @@ class TestConfig(unittest.TestCase):
|
|||
with TemporaryDirectory() as tmp_dir:
|
||||
# The created dir is in /tmp. This is a symlink on OS X
|
||||
# causing this test to fail unless we resolve path first.
|
||||
self.config.whitelist_external_dirs = set((
|
||||
os.path.realpath(tmp_dir),
|
||||
))
|
||||
self.config.whitelist_external_dirs = set((os.path.realpath(tmp_dir),))
|
||||
|
||||
test_file = os.path.join(tmp_dir, "test.jpg")
|
||||
with open(test_file, "w") as tmp_file:
|
||||
tmp_file.write("test")
|
||||
|
||||
valid = [
|
||||
test_file,
|
||||
tmp_dir,
|
||||
os.path.join(tmp_dir, 'notfound321')
|
||||
]
|
||||
valid = [test_file, tmp_dir, os.path.join(tmp_dir, "notfound321")]
|
||||
for path in valid:
|
||||
assert self.config.is_allowed_path(path)
|
||||
|
||||
self.config.whitelist_external_dirs = set(('/home', '/var'))
|
||||
self.config.whitelist_external_dirs = set(("/home", "/var"))
|
||||
|
||||
unvalid = [
|
||||
"/hass/config/secure",
|
||||
|
|
@ -960,16 +952,16 @@ async def test_event_on_update(hass, hass_storage):
|
|||
|
||||
assert hass.config.latitude == 12
|
||||
assert len(events) == 1
|
||||
assert events[0].data == {'latitude': 12}
|
||||
assert events[0].data == {"latitude": 12}
|
||||
|
||||
|
||||
async def test_bad_timezone_raises_value_error(hass):
|
||||
"""Test bad timezone raises ValueError."""
|
||||
with pytest.raises(ValueError):
|
||||
await hass.config.async_update(time_zone='not_a_timezone')
|
||||
await hass.config.async_update(time_zone="not_a_timezone")
|
||||
|
||||
|
||||
@patch('homeassistant.core.monotonic')
|
||||
@patch("homeassistant.core.monotonic")
|
||||
def test_create_timer(mock_monotonic, loop):
|
||||
"""Test create timer."""
|
||||
hass = MagicMock()
|
||||
|
|
@ -982,9 +974,10 @@ def test_create_timer(mock_monotonic, loop):
|
|||
|
||||
mock_monotonic.side_effect = 10.2, 10.8, 11.3
|
||||
|
||||
with patch.object(ha, 'callback', mock_callback), \
|
||||
patch('homeassistant.core.dt_util.utcnow',
|
||||
return_value=datetime(2018, 12, 31, 3, 4, 5, 333333)):
|
||||
with patch.object(ha, "callback", mock_callback), patch(
|
||||
"homeassistant.core.dt_util.utcnow",
|
||||
return_value=datetime(2018, 12, 31, 3, 4, 5, 333333),
|
||||
):
|
||||
ha._async_create_timer(hass)
|
||||
|
||||
assert len(funcs) == 2
|
||||
|
|
@ -996,8 +989,10 @@ def test_create_timer(mock_monotonic, loop):
|
|||
assert callback is fire_time_event
|
||||
assert abs(target - 10.866667) < 0.001
|
||||
|
||||
with patch('homeassistant.core.dt_util.utcnow',
|
||||
return_value=datetime(2018, 12, 31, 3, 4, 6, 100000)):
|
||||
with patch(
|
||||
"homeassistant.core.dt_util.utcnow",
|
||||
return_value=datetime(2018, 12, 31, 3, 4, 6, 100000),
|
||||
):
|
||||
callback(target)
|
||||
|
||||
assert len(hass.bus.async_listen_once.mock_calls) == 1
|
||||
|
|
@ -1018,7 +1013,7 @@ def test_create_timer(mock_monotonic, loop):
|
|||
assert event_data[ATTR_NOW] == datetime(2018, 12, 31, 3, 4, 6, 100000)
|
||||
|
||||
|
||||
@patch('homeassistant.core.monotonic')
|
||||
@patch("homeassistant.core.monotonic")
|
||||
def test_timer_out_of_sync(mock_monotonic, loop):
|
||||
"""Test create timer."""
|
||||
hass = MagicMock()
|
||||
|
|
@ -1031,15 +1026,18 @@ def test_timer_out_of_sync(mock_monotonic, loop):
|
|||
|
||||
mock_monotonic.side_effect = 10.2, 13.3, 13.4
|
||||
|
||||
with patch.object(ha, 'callback', mock_callback), \
|
||||
patch('homeassistant.core.dt_util.utcnow',
|
||||
return_value=datetime(2018, 12, 31, 3, 4, 5, 333333)):
|
||||
with patch.object(ha, "callback", mock_callback), patch(
|
||||
"homeassistant.core.dt_util.utcnow",
|
||||
return_value=datetime(2018, 12, 31, 3, 4, 5, 333333),
|
||||
):
|
||||
ha._async_create_timer(hass)
|
||||
|
||||
delay, callback, target = hass.loop.call_later.mock_calls[0][1]
|
||||
|
||||
with patch('homeassistant.core.dt_util.utcnow',
|
||||
return_value=datetime(2018, 12, 31, 3, 4, 8, 200000)):
|
||||
with patch(
|
||||
"homeassistant.core.dt_util.utcnow",
|
||||
return_value=datetime(2018, 12, 31, 3, 4, 8, 200000),
|
||||
):
|
||||
callback(target)
|
||||
|
||||
event_type, event_data = hass.bus.async_fire.mock_calls[1][1]
|
||||
|
|
@ -1063,7 +1061,7 @@ def test_hass_start_starts_the_timer(loop):
|
|||
hass = ha.HomeAssistant(loop=loop)
|
||||
|
||||
try:
|
||||
with patch('homeassistant.core._async_create_timer') as mock_timer:
|
||||
with patch("homeassistant.core._async_create_timer") as mock_timer:
|
||||
yield from hass.async_start()
|
||||
|
||||
assert hass.state == ha.CoreState.running
|
||||
|
|
@ -1083,15 +1081,15 @@ def test_start_taking_too_long(loop, caplog):
|
|||
caplog.set_level(logging.WARNING)
|
||||
|
||||
try:
|
||||
with patch('homeassistant.core.timeout',
|
||||
side_effect=asyncio.TimeoutError), \
|
||||
patch('homeassistant.core._async_create_timer') as mock_timer:
|
||||
with patch(
|
||||
"homeassistant.core.timeout", side_effect=asyncio.TimeoutError
|
||||
), patch("homeassistant.core._async_create_timer") as mock_timer:
|
||||
yield from hass.async_start()
|
||||
|
||||
assert hass.state == ha.CoreState.running
|
||||
assert len(mock_timer.mock_calls) == 1
|
||||
assert mock_timer.mock_calls[0][1][0] is hass
|
||||
assert 'Something is blocking Home Assistant' in caplog.text
|
||||
assert "Something is blocking Home Assistant" in caplog.text
|
||||
|
||||
finally:
|
||||
yield from hass.async_stop()
|
||||
|
|
@ -1116,27 +1114,27 @@ def test_track_task_functions(loop):
|
|||
|
||||
async def test_service_executed_with_subservices(hass):
|
||||
"""Test we block correctly till all services done."""
|
||||
calls = async_mock_service(hass, 'test', 'inner')
|
||||
calls = async_mock_service(hass, "test", "inner")
|
||||
context = ha.Context()
|
||||
|
||||
async def handle_outer(call):
|
||||
"""Handle outer service call."""
|
||||
calls.append(call)
|
||||
call1 = hass.services.async_call('test', 'inner', blocking=True,
|
||||
context=call.context)
|
||||
call2 = hass.services.async_call('test', 'inner', blocking=True,
|
||||
context=call.context)
|
||||
call1 = hass.services.async_call(
|
||||
"test", "inner", blocking=True, context=call.context
|
||||
)
|
||||
call2 = hass.services.async_call(
|
||||
"test", "inner", blocking=True, context=call.context
|
||||
)
|
||||
await asyncio.wait([call1, call2])
|
||||
calls.append(call)
|
||||
|
||||
hass.services.async_register('test', 'outer', handle_outer)
|
||||
hass.services.async_register("test", "outer", handle_outer)
|
||||
|
||||
await hass.services.async_call('test', 'outer', blocking=True,
|
||||
context=context)
|
||||
await hass.services.async_call("test", "outer", blocking=True, context=context)
|
||||
|
||||
assert len(calls) == 4
|
||||
assert [call.service for call in calls] == [
|
||||
'outer', 'inner', 'inner', 'outer']
|
||||
assert [call.service for call in calls] == ["outer", "inner", "inner", "outer"]
|
||||
assert all(call.context is context for call in calls)
|
||||
|
||||
|
||||
|
|
@ -1150,20 +1148,20 @@ async def test_service_call_event_contains_original_data(hass):
|
|||
|
||||
hass.bus.async_listen(EVENT_CALL_SERVICE, callback)
|
||||
|
||||
calls = async_mock_service(hass, 'test', 'service', vol.Schema({
|
||||
'number': vol.Coerce(int)
|
||||
}))
|
||||
calls = async_mock_service(
|
||||
hass, "test", "service", vol.Schema({"number": vol.Coerce(int)})
|
||||
)
|
||||
|
||||
context = ha.Context()
|
||||
await hass.services.async_call('test', 'service', {
|
||||
'number': '23'
|
||||
}, blocking=True, context=context)
|
||||
await hass.services.async_call(
|
||||
"test", "service", {"number": "23"}, blocking=True, context=context
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
assert len(events) == 1
|
||||
assert events[0].data['service_data']['number'] == '23'
|
||||
assert events[0].data["service_data"]["number"] == "23"
|
||||
assert events[0].context is context
|
||||
assert len(calls) == 1
|
||||
assert calls[0].data['number'] == 23
|
||||
assert calls[0].data["number"] == 23
|
||||
assert calls[0].context is context
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue