jeena/home-assistant
1
0
Fork 0
Code Issues Pull requests Projects Packages Wiki Activity Actions

Restore for automation entities (#6254)

Browse source 
* Restore for automation entities

* coroutine

* no clue what i'm doing now

* Still passes nicely in py 3.4
This commit is contained in:
Johann Kellerman 2017-03-05 01:19:01 +02:00 • committed by Paulus Schoutsen
parent 8232f1ef65
commit 1522e67351
3 changed files with 80 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/common.py
View file

@ -131,6 +131,7 @@ def async_test_home_assistant(loop):
@ha.callback
def clear_instance(event):
"""Clear global instance."""
global INST_COUNT
INST_COUNT -= 1
@ -140,20 +141,18 @@ def async_test_home_assistant(loop):
def mock_service(hass, domain, service):
"""Setup a fake service.
Return a list that logs all calls to fake service.
"""
"""Setup a fake service & return a list that logs calls to this service."""
calls = []
# pylint: disable=redefined-outer-name
@ha.callback
def mock_service(call):
@asyncio.coroutine
def mock_service_log(call): # pylint: disable=unnecessary-lambda
""""Mocked service call."""
calls.append(call)
# pylint: disable=unnecessary-lambda
hass.services.register(domain, service, mock_service)
if hass.loop.__dict__.get("_thread_ident", 0) == threading.get_ident():
hass.services.async_register(domain, service, mock_service_log)
else:
hass.services.register(domain, service, mock_service_log)
return calls