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

142
tests/components/automation/test_mqtt.py
View File

@@ -6,91 +6,98 @@ from homeassistant.setup import async_setup_component
import homeassistant.components.automation as automation
from tests.common import (
async_fire_mqtt_message,
mock_component, async_mock_service, async_mock_mqtt_component)
mock_component,
async_mock_service,
async_mock_mqtt_component,
)
from tests.components.automation import common
@pytest.fixture
def calls(hass):
"""Track calls to a mock serivce."""
return async_mock_service(hass, 'test', 'automation')
return async_mock_service(hass, "test", "automation")
@pytest.fixture(autouse=True)
def setup_comp(hass):
"""Initialize components."""
mock_component(hass, 'group')
mock_component(hass, "group")
hass.loop.run_until_complete(async_mock_mqtt_component(hass))
async def test_if_fires_on_topic_match(hass, calls):
"""Test if message is fired on topic match."""
assert await async_setup_component(hass, automation.DOMAIN, {
automation.DOMAIN: {
'trigger': {
'platform': 'mqtt',
'topic': 'test-topic'
},
'action': {
'service': 'test.automation',
'data_template': {
'some': '{{ trigger.platform }} - {{ trigger.topic }}'
' - {{ trigger.payload }} - '
'{{ trigger.payload_json.hello }}'
assert await async_setup_component(
hass,
automation.DOMAIN,
{
automation.DOMAIN: {
"trigger": {"platform": "mqtt", "topic": "test-topic"},
"action": {
"service": "test.automation",
"data_template": {
"some": "{{ trigger.platform }} - {{ trigger.topic }}"
" - {{ trigger.payload }} - "
"{{ trigger.payload_json.hello }}"
},
},
}
}
})
},
)
async_fire_mqtt_message(hass, 'test-topic', '{ "hello": "world" }')
async_fire_mqtt_message(hass, "test-topic", '{ "hello": "world" }')
await hass.async_block_till_done()
assert 1 == len(calls)
assert 'mqtt - test-topic - { "hello": "world" } - world' == \
calls[0].data['some']
assert 'mqtt - test-topic - { "hello": "world" } - world' == calls[0].data["some"]
await common.async_turn_off(hass)
await hass.async_block_till_done()
async_fire_mqtt_message(hass, 'test-topic', 'test_payload')
async_fire_mqtt_message(hass, "test-topic", "test_payload")
await hass.async_block_till_done()
assert 1 == len(calls)
async def test_if_fires_on_topic_and_payload_match(hass, calls):
"""Test if message is fired on topic and payload match."""
assert await async_setup_component(hass, automation.DOMAIN, {
automation.DOMAIN: {
'trigger': {
'platform': 'mqtt',
'topic': 'test-topic',
'payload': 'hello'
},
'action': {
'service': 'test.automation'
assert await async_setup_component(
hass,
automation.DOMAIN,
{
automation.DOMAIN: {
"trigger": {
"platform": "mqtt",
"topic": "test-topic",
"payload": "hello",
},
"action": {"service": "test.automation"},
}
}
})
},
)
async_fire_mqtt_message(hass, 'test-topic', 'hello')
async_fire_mqtt_message(hass, "test-topic", "hello")
await hass.async_block_till_done()
assert 1 == len(calls)
async def test_if_not_fires_on_topic_but_no_payload_match(hass, calls):
"""Test if message is not fired on topic but no payload."""
assert await async_setup_component(hass, automation.DOMAIN, {
automation.DOMAIN: {
'trigger': {
'platform': 'mqtt',
'topic': 'test-topic',
'payload': 'hello'
},
'action': {
'service': 'test.automation'
assert await async_setup_component(
hass,
automation.DOMAIN,
{
automation.DOMAIN: {
"trigger": {
"platform": "mqtt",
"topic": "test-topic",
"payload": "hello",
},
"action": {"service": "test.automation"},
}
}
})
},
)
async_fire_mqtt_message(hass, 'test-topic', 'no-hello')
async_fire_mqtt_message(hass, "test-topic", "no-hello")
await hass.async_block_till_done()
assert 0 == len(calls)
@@ -99,38 +106,35 @@ async def test_encoding_default(hass, calls):
"""Test default encoding."""
mock_mqtt = await async_mock_mqtt_component(hass)
assert await async_setup_component(hass, automation.DOMAIN, {
automation.DOMAIN: {
'trigger': {
'platform': 'mqtt',
'topic': 'test-topic'
},
'action': {
'service': 'test.automation'
assert await async_setup_component(
hass,
automation.DOMAIN,
{
automation.DOMAIN: {
"trigger": {"platform": "mqtt", "topic": "test-topic"},
"action": {"service": "test.automation"},
}
}
})
},
)
mock_mqtt.async_subscribe.assert_called_once_with(
'test-topic', mock.ANY, 0, 'utf-8')
"test-topic", mock.ANY, 0, "utf-8"
)
async def test_encoding_custom(hass, calls):
"""Test default encoding."""
mock_mqtt = await async_mock_mqtt_component(hass)
assert await async_setup_component(hass, automation.DOMAIN, {
automation.DOMAIN: {
'trigger': {
'platform': 'mqtt',
'topic': 'test-topic',
'encoding': ''
},
'action': {
'service': 'test.automation'
assert await async_setup_component(
hass,
automation.DOMAIN,
{
automation.DOMAIN: {
"trigger": {"platform": "mqtt", "topic": "test-topic", "encoding": ""},
"action": {"service": "test.automation"},
}
}
})
},
)
mock_mqtt.async_subscribe.assert_called_once_with(
'test-topic', mock.ANY, 0, None)
mock_mqtt.async_subscribe.assert_called_once_with("test-topic", mock.ANY, 0, None)