Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

Split mock_service (#8198)

Browse Source
This commit is contained in:
Paulus Schoutsen
2017-06-25 10:53:15 -07:00
committed by GitHub
parent 1cfed4f015
commit 7bf6ceafec
4 changed files with 21 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/common.py
View File

@@ -177,7 +177,8 @@ def get_test_instance_port():
return _TEST_INSTANCE_PORT
def mock_service(hass, domain, service):
@ha.callback
def async_mock_service(hass, domain, service):
"""Set up a fake service & return a calls log list to this service."""
calls = []
@@ -186,14 +187,14 @@ def mock_service(hass, domain, service):
"""Mock service call."""
calls.append(call)
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)
hass.services.async_register(domain, service, mock_service_log)
return calls
mock_service = threadsafe_callback_factory(async_mock_service)
@ha.callback
def async_fire_mqtt_message(hass, topic, payload, qos=0):
"""Fire the MQTT message."""