Drop use of mock_mqtt_component (#37013)
Co-authored-by: Paulus Schoutsen <paulus@home-assistant.io>
This commit is contained in:
Erik Montnemery
GitHub
parent
cf6480cda0
commit
4e77969f5e
13 changed files with 2168 additions and 2375 deletions
|
|
@ -23,7 +23,7 @@ from homeassistant.auth import (
|
|||
providers as auth_providers,
|
||||
)
|
||||
from homeassistant.auth.permissions import system_policies
|
||||
from homeassistant.components import mqtt, recorder
|
||||
from homeassistant.components import recorder
|
||||
from homeassistant.components.device_automation import ( # noqa: F401
|
||||
_async_get_device_automation_capabilities as async_get_device_automation_capabilities,
|
||||
_async_get_device_automations as async_get_device_automations,
|
||||
|
|
@ -53,13 +53,13 @@ from homeassistant.helpers import (
|
|||
storage,
|
||||
)
|
||||
from homeassistant.helpers.json import JSONEncoder
|
||||
from homeassistant.setup import async_setup_component, setup_component
|
||||
from homeassistant.setup import setup_component
|
||||
from homeassistant.util.async_ import run_callback_threadsafe
|
||||
import homeassistant.util.dt as date_util
|
||||
from homeassistant.util.unit_system import METRIC_SYSTEM
|
||||
import homeassistant.util.yaml.loader as yaml_loader
|
||||
|
||||
from tests.async_mock import AsyncMock, MagicMock, Mock, patch
|
||||
from tests.async_mock import AsyncMock, Mock, patch
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
INSTANCES = []
|
||||
|
|
@ -324,39 +324,6 @@ def mock_state_change_event(hass, new_state, old_state=None):
|
|||
hass.bus.fire(EVENT_STATE_CHANGED, event_data, context=new_state.context)
|
||||
|
||||
|
||||
async def async_mock_mqtt_component(hass, config=None):
|
||||
"""Mock the MQTT component."""
|
||||
if config is None:
|
||||
config = {mqtt.CONF_BROKER: "mock-broker"}
|
||||
|
||||
@ha.callback
|
||||
def _async_fire_mqtt_message(topic, payload, qos, retain):
|
||||
async_fire_mqtt_message(hass, topic, payload, qos, retain)
|
||||
|
||||
with patch("paho.mqtt.client.Client") as mock_client:
|
||||
mock_client = mock_client.return_value
|
||||
mock_client.connect.return_value = 0
|
||||
mock_client.subscribe.return_value = (0, 0)
|
||||
mock_client.unsubscribe.return_value = (0, 0)
|
||||
mock_client.publish.side_effect = _async_fire_mqtt_message
|
||||
|
||||
result = await async_setup_component(hass, mqtt.DOMAIN, {mqtt.DOMAIN: config})
|
||||
assert result
|
||||
await hass.async_block_till_done()
|
||||
|
||||
mqtt_component_mock = MagicMock(
|
||||
spec_set=hass.data["mqtt"], wraps=hass.data["mqtt"]
|
||||
)
|
||||
hass.data["mqtt"].connected = mqtt_component_mock.connected
|
||||
mqtt_component_mock._mqttc = mock_client
|
||||
|
||||
hass.data["mqtt"] = mqtt_component_mock
|
||||
return hass.data["mqtt"]
|
||||
|
||||
|
||||
mock_mqtt_component = threadsafe_coroutine_factory(async_mock_mqtt_component)
|
||||
|
||||
|
||||
@ha.callback
|
||||
def mock_component(hass, component):
|
||||
"""Mock a component is setup."""
|
||||
|
|
|
|||
|
|
@ -15,7 +15,6 @@ from homeassistant.const import (
|
|||
SERVICE_ALARM_DISARM,
|
||||
SERVICE_ALARM_TRIGGER,
|
||||
)
|
||||
from homeassistant.loader import bind_hass
|
||||
|
||||
|
||||
async def async_alarm_disarm(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
||||
|
|
@ -29,18 +28,6 @@ async def async_alarm_disarm(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
|||
await hass.services.async_call(DOMAIN, SERVICE_ALARM_DISARM, data, blocking=True)
|
||||
|
||||
|
||||
@bind_hass
|
||||
def alarm_disarm(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
||||
"""Send the alarm the command for disarm."""
|
||||
data = {}
|
||||
if code:
|
||||
data[ATTR_CODE] = code
|
||||
if entity_id:
|
||||
data[ATTR_ENTITY_ID] = entity_id
|
||||
|
||||
hass.services.call(DOMAIN, SERVICE_ALARM_DISARM, data)
|
||||
|
||||
|
||||
async def async_alarm_arm_home(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
||||
"""Send the alarm the command for disarm."""
|
||||
data = {}
|
||||
|
|
@ -52,18 +39,6 @@ async def async_alarm_arm_home(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
|||
await hass.services.async_call(DOMAIN, SERVICE_ALARM_ARM_HOME, data, blocking=True)
|
||||
|
||||
|
||||
@bind_hass
|
||||
def alarm_arm_home(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
||||
"""Send the alarm the command for arm home."""
|
||||
data = {}
|
||||
if code:
|
||||
data[ATTR_CODE] = code
|
||||
if entity_id:
|
||||
data[ATTR_ENTITY_ID] = entity_id
|
||||
|
||||
hass.services.call(DOMAIN, SERVICE_ALARM_ARM_HOME, data)
|
||||
|
||||
|
||||
async def async_alarm_arm_away(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
||||
"""Send the alarm the command for disarm."""
|
||||
data = {}
|
||||
|
|
@ -75,18 +50,6 @@ async def async_alarm_arm_away(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
|||
await hass.services.async_call(DOMAIN, SERVICE_ALARM_ARM_AWAY, data, blocking=True)
|
||||
|
||||
|
||||
@bind_hass
|
||||
def alarm_arm_away(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
||||
"""Send the alarm the command for arm away."""
|
||||
data = {}
|
||||
if code:
|
||||
data[ATTR_CODE] = code
|
||||
if entity_id:
|
||||
data[ATTR_ENTITY_ID] = entity_id
|
||||
|
||||
hass.services.call(DOMAIN, SERVICE_ALARM_ARM_AWAY, data)
|
||||
|
||||
|
||||
async def async_alarm_arm_night(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
||||
"""Send the alarm the command for disarm."""
|
||||
data = {}
|
||||
|
|
@ -98,18 +61,6 @@ async def async_alarm_arm_night(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
|||
await hass.services.async_call(DOMAIN, SERVICE_ALARM_ARM_NIGHT, data, blocking=True)
|
||||
|
||||
|
||||
@bind_hass
|
||||
def alarm_arm_night(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
||||
"""Send the alarm the command for arm night."""
|
||||
data = {}
|
||||
if code:
|
||||
data[ATTR_CODE] = code
|
||||
if entity_id:
|
||||
data[ATTR_ENTITY_ID] = entity_id
|
||||
|
||||
hass.services.call(DOMAIN, SERVICE_ALARM_ARM_NIGHT, data)
|
||||
|
||||
|
||||
async def async_alarm_trigger(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
||||
"""Send the alarm the command for disarm."""
|
||||
data = {}
|
||||
|
|
@ -121,18 +72,6 @@ async def async_alarm_trigger(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
|||
await hass.services.async_call(DOMAIN, SERVICE_ALARM_TRIGGER, data, blocking=True)
|
||||
|
||||
|
||||
@bind_hass
|
||||
def alarm_trigger(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
||||
"""Send the alarm the command for trigger."""
|
||||
data = {}
|
||||
if code:
|
||||
data[ATTR_CODE] = code
|
||||
if entity_id:
|
||||
data[ATTR_ENTITY_ID] = entity_id
|
||||
|
||||
hass.services.call(DOMAIN, SERVICE_ALARM_TRIGGER, data)
|
||||
|
||||
|
||||
async def async_alarm_arm_custom_bypass(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
||||
"""Send the alarm the command for disarm."""
|
||||
data = {}
|
||||
|
|
@ -144,15 +83,3 @@ async def async_alarm_arm_custom_bypass(hass, code=None, entity_id=ENTITY_MATCH_
|
|||
await hass.services.async_call(
|
||||
DOMAIN, SERVICE_ALARM_ARM_CUSTOM_BYPASS, data, blocking=True
|
||||
)
|
||||
|
||||
|
||||
@bind_hass
|
||||
def alarm_arm_custom_bypass(hass, code=None, entity_id=ENTITY_MATCH_ALL):
|
||||
"""Send the alarm the command for arm custom bypass."""
|
||||
data = {}
|
||||
if code:
|
||||
data[ATTR_CODE] = code
|
||||
if entity_id:
|
||||
data[ATTR_ENTITY_ID] = entity_id
|
||||
|
||||
hass.services.call(DOMAIN, SERVICE_ALARM_ARM_CUSTOM_BYPASS, data)
|
||||
|
|
|
|||
File diff suppressed because it is too large
Load diff
|
|
@ -17,6 +17,7 @@ from homeassistant.const import (
|
|||
STATE_ALARM_TRIGGERED,
|
||||
STATE_UNKNOWN,
|
||||
)
|
||||
from homeassistant.setup import async_setup_component
|
||||
|
||||
from .test_common import (
|
||||
help_test_availability_when_connection_lost,
|
||||
|
|
@ -41,11 +42,7 @@ from .test_common import (
|
|||
help_test_update_with_json_attrs_not_dict,
|
||||
)
|
||||
|
||||
from tests.common import (
|
||||
assert_setup_component,
|
||||
async_fire_mqtt_message,
|
||||
async_setup_component,
|
||||
)
|
||||
from tests.common import assert_setup_component, async_fire_mqtt_message
|
||||
from tests.components.alarm_control_panel import common
|
||||
|
||||
CODE_NUMBER = "1234"
|
||||
|
|
|
|||
|
|
@ -23,6 +23,7 @@ from homeassistant.components.climate.const import (
|
|||
SUPPORT_TARGET_TEMPERATURE_RANGE,
|
||||
)
|
||||
from homeassistant.const import STATE_OFF
|
||||
from homeassistant.setup import async_setup_component
|
||||
|
||||
from .test_common import (
|
||||
help_test_availability_when_connection_lost,
|
||||
|
|
@ -48,7 +49,7 @@ from .test_common import (
|
|||
)
|
||||
|
||||
from tests.async_mock import call
|
||||
from tests.common import async_fire_mqtt_message, async_setup_component
|
||||
from tests.common import async_fire_mqtt_message
|
||||
from tests.components.climate import common
|
||||
|
||||
ENTITY_CLIMATE = "climate.test"
|
||||
|
|
|
|||
|
|
@ -10,9 +10,10 @@ from homeassistant.components.mqtt.const import MQTT_DISCONNECTED
|
|||
from homeassistant.components.mqtt.discovery import async_start
|
||||
from homeassistant.const import ATTR_ASSUMED_STATE, STATE_UNAVAILABLE
|
||||
from homeassistant.helpers.dispatcher import async_dispatcher_send
|
||||
from homeassistant.setup import async_setup_component
|
||||
|
||||
from tests.async_mock import ANY
|
||||
from tests.common import async_fire_mqtt_message, async_setup_component, mock_registry
|
||||
from tests.common import async_fire_mqtt_message, mock_registry
|
||||
|
||||
DEFAULT_CONFIG_DEVICE_INFO_ID = {
|
||||
"identifiers": ["helloworld"],
|
||||
|
|
|
|||
|
|
@ -7,15 +7,13 @@ import homeassistant.components.mqtt_eventstream as eventstream
|
|||
from homeassistant.const import EVENT_STATE_CHANGED
|
||||
from homeassistant.core import State, callback
|
||||
from homeassistant.helpers.json import JSONEncoder
|
||||
from homeassistant.setup import setup_component
|
||||
from homeassistant.setup import async_setup_component
|
||||
import homeassistant.util.dt as dt_util
|
||||
|
||||
from tests.async_mock import ANY, patch
|
||||
from tests.common import (
|
||||
fire_mqtt_message,
|
||||
fire_time_changed,
|
||||
get_test_home_assistant,
|
||||
mock_mqtt_component,
|
||||
async_fire_mqtt_message,
|
||||
async_fire_time_changed,
|
||||
mock_state_change_event,
|
||||
)
|
||||
|
||||
|
|
@ -25,177 +23,170 @@ def mock_storage(hass_storage):
|
|||
"""Autouse hass_storage for the TestCase tests."""
|
||||
|
||||
|
||||
class TestMqttEventStream:
|
||||
"""Test the MQTT eventstream module."""
|
||||
async def add_eventstream(hass, sub_topic=None, pub_topic=None, ignore_event=None):
|
||||
"""Add a mqtt_eventstream component."""
|
||||
config = {}
|
||||
if sub_topic:
|
||||
config["subscribe_topic"] = sub_topic
|
||||
if pub_topic:
|
||||
config["publish_topic"] = pub_topic
|
||||
if ignore_event:
|
||||
config["ignore_event"] = ignore_event
|
||||
return await async_setup_component(
|
||||
hass, eventstream.DOMAIN, {eventstream.DOMAIN: config}
|
||||
)
|
||||
|
||||
def setup_method(self):
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.hass = get_test_home_assistant()
|
||||
self.mock_mqtt = mock_mqtt_component(self.hass)
|
||||
|
||||
def teardown_method(self):
|
||||
"""Stop everything that was started."""
|
||||
self.hass.stop()
|
||||
async def test_setup_succeeds(hass, mqtt_mock):
|
||||
"""Test the success of the setup."""
|
||||
assert await add_eventstream(hass)
|
||||
|
||||
def add_eventstream(self, sub_topic=None, pub_topic=None, ignore_event=None):
|
||||
"""Add a mqtt_eventstream component."""
|
||||
config = {}
|
||||
if sub_topic:
|
||||
config["subscribe_topic"] = sub_topic
|
||||
if pub_topic:
|
||||
config["publish_topic"] = pub_topic
|
||||
if ignore_event:
|
||||
config["ignore_event"] = ignore_event
|
||||
return setup_component(
|
||||
self.hass, eventstream.DOMAIN, {eventstream.DOMAIN: config}
|
||||
)
|
||||
|
||||
def test_setup_succeeds(self):
|
||||
"""Test the success of the setup."""
|
||||
assert self.add_eventstream()
|
||||
async def test_setup_with_pub(hass, mqtt_mock):
|
||||
"""Test the setup with subscription."""
|
||||
# Should start off with no listeners for all events
|
||||
assert hass.bus.async_listeners().get("*") is None
|
||||
|
||||
def test_setup_with_pub(self):
|
||||
"""Test the setup with subscription."""
|
||||
# Should start off with no listeners for all events
|
||||
assert self.hass.bus.listeners.get("*") is None
|
||||
assert await add_eventstream(hass, pub_topic="bar")
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert self.add_eventstream(pub_topic="bar")
|
||||
self.hass.block_till_done()
|
||||
# Verify that the event handler has been added as a listener
|
||||
assert hass.bus.async_listeners().get("*") == 1
|
||||
|
||||
# Verify that the event handler has been added as a listener
|
||||
assert self.hass.bus.listeners.get("*") == 1
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_subscribe")
|
||||
def test_subscribe(self, mock_sub):
|
||||
"""Test the subscription."""
|
||||
sub_topic = "foo"
|
||||
assert self.add_eventstream(sub_topic=sub_topic)
|
||||
self.hass.block_till_done()
|
||||
async def test_subscribe(hass, mqtt_mock):
|
||||
"""Test the subscription."""
|
||||
sub_topic = "foo"
|
||||
assert await add_eventstream(hass, sub_topic=sub_topic)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Verify that the this entity was subscribed to the topic
|
||||
mock_sub.assert_called_with(self.hass, sub_topic, ANY)
|
||||
# Verify that the this entity was subscribed to the topic
|
||||
mqtt_mock.async_subscribe.assert_called_with(sub_topic, ANY, 0, ANY)
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_publish")
|
||||
@patch("homeassistant.core.dt_util.utcnow")
|
||||
def test_state_changed_event_sends_message(self, mock_utcnow, mock_pub):
|
||||
"""Test the sending of a new message if event changed."""
|
||||
now = dt_util.as_utc(dt_util.now())
|
||||
e_id = "fake.entity"
|
||||
pub_topic = "bar"
|
||||
mock_utcnow.return_value = now
|
||||
|
||||
async def test_state_changed_event_sends_message(hass, mqtt_mock):
|
||||
"""Test the sending of a new message if event changed."""
|
||||
now = dt_util.as_utc(dt_util.now())
|
||||
e_id = "fake.entity"
|
||||
pub_topic = "bar"
|
||||
with patch(
|
||||
("homeassistant.core.dt_util.utcnow"), return_value=now,
|
||||
):
|
||||
# Add the eventstream component for publishing events
|
||||
assert self.add_eventstream(pub_topic=pub_topic)
|
||||
self.hass.block_till_done()
|
||||
assert await add_eventstream(hass, pub_topic=pub_topic)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_eventstream state change on initialization, etc.
|
||||
mock_pub.reset_mock()
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(self.hass, State(e_id, "on"))
|
||||
self.hass.block_till_done()
|
||||
mock_state_change_event(hass, State(e_id, "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# The order of the JSON is indeterminate,
|
||||
# so first just check that publish was called
|
||||
mock_pub.assert_called_with(self.hass, pub_topic, ANY)
|
||||
assert mock_pub.called
|
||||
# The order of the JSON is indeterminate,
|
||||
# so first just check that publish was called
|
||||
mqtt_mock.async_publish.assert_called_with(pub_topic, ANY, 0, False)
|
||||
assert mqtt_mock.async_publish.called
|
||||
|
||||
# Get the actual call to publish and make sure it was the one
|
||||
# we were looking for
|
||||
msg = mock_pub.call_args[0][2]
|
||||
event = {}
|
||||
event["event_type"] = EVENT_STATE_CHANGED
|
||||
new_state = {
|
||||
"last_updated": now.isoformat(),
|
||||
"state": "on",
|
||||
"entity_id": e_id,
|
||||
"attributes": {},
|
||||
"last_changed": now.isoformat(),
|
||||
}
|
||||
event["event_data"] = {"new_state": new_state, "entity_id": e_id}
|
||||
# Get the actual call to publish and make sure it was the one
|
||||
# we were looking for
|
||||
msg = mqtt_mock.async_publish.call_args[0][1]
|
||||
event = {}
|
||||
event["event_type"] = EVENT_STATE_CHANGED
|
||||
new_state = {
|
||||
"last_updated": now.isoformat(),
|
||||
"state": "on",
|
||||
"entity_id": e_id,
|
||||
"attributes": {},
|
||||
"last_changed": now.isoformat(),
|
||||
}
|
||||
event["event_data"] = {"new_state": new_state, "entity_id": e_id}
|
||||
|
||||
# Verify that the message received was that expected
|
||||
result = json.loads(msg)
|
||||
result["event_data"]["new_state"].pop("context")
|
||||
assert result == event
|
||||
# Verify that the message received was that expected
|
||||
result = json.loads(msg)
|
||||
result["event_data"]["new_state"].pop("context")
|
||||
assert result == event
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_publish")
|
||||
def test_time_event_does_not_send_message(self, mock_pub):
|
||||
"""Test the sending of a new message if time event."""
|
||||
assert self.add_eventstream(pub_topic="bar")
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_eventstream state change on initialization, etc.
|
||||
mock_pub.reset_mock()
|
||||
async def test_time_event_does_not_send_message(hass, mqtt_mock):
|
||||
"""Test the sending of a new message if time event."""
|
||||
assert await add_eventstream(hass, pub_topic="bar")
|
||||
await hass.async_block_till_done()
|
||||
|
||||
fire_time_changed(self.hass, dt_util.utcnow())
|
||||
assert not mock_pub.called
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_eventstream state change on initialization, etc.
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
|
||||
def test_receiving_remote_event_fires_hass_event(self):
|
||||
"""Test the receiving of the remotely fired event."""
|
||||
sub_topic = "foo"
|
||||
assert self.add_eventstream(sub_topic=sub_topic)
|
||||
self.hass.block_till_done()
|
||||
async_fire_time_changed(hass, dt_util.utcnow())
|
||||
assert not mqtt_mock.async_publish.called
|
||||
|
||||
calls = []
|
||||
|
||||
@callback
|
||||
def listener(_):
|
||||
calls.append(1)
|
||||
async def test_receiving_remote_event_fires_hass_event(hass, mqtt_mock):
|
||||
"""Test the receiving of the remotely fired event."""
|
||||
sub_topic = "foo"
|
||||
assert await add_eventstream(hass, sub_topic=sub_topic)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
self.hass.bus.listen_once("test_event", listener)
|
||||
self.hass.block_till_done()
|
||||
calls = []
|
||||
|
||||
payload = json.dumps(
|
||||
{"event_type": "test_event", "event_data": {}}, cls=JSONEncoder
|
||||
)
|
||||
fire_mqtt_message(self.hass, sub_topic, payload)
|
||||
self.hass.block_till_done()
|
||||
@callback
|
||||
def listener(_):
|
||||
calls.append(1)
|
||||
|
||||
assert 1 == len(calls)
|
||||
hass.bus.async_listen_once("test_event", listener)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_publish")
|
||||
def test_ignored_event_doesnt_send_over_stream(self, mock_pub):
|
||||
"""Test the ignoring of sending events if defined."""
|
||||
assert self.add_eventstream(pub_topic="bar", ignore_event=["state_changed"])
|
||||
self.hass.block_till_done()
|
||||
payload = json.dumps(
|
||||
{"event_type": "test_event", "event_data": {}}, cls=JSONEncoder
|
||||
)
|
||||
async_fire_mqtt_message(hass, sub_topic, payload)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
e_id = "entity.test_id"
|
||||
event = {}
|
||||
event["event_type"] = EVENT_STATE_CHANGED
|
||||
new_state = {"state": "on", "entity_id": e_id, "attributes": {}}
|
||||
event["event_data"] = {"new_state": new_state, "entity_id": e_id}
|
||||
assert 1 == len(calls)
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_eventstream state change on initialization, etc.
|
||||
mock_pub.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(self.hass, State(e_id, "on"))
|
||||
self.hass.block_till_done()
|
||||
async def test_ignored_event_doesnt_send_over_stream(hass, mqtt_mock):
|
||||
"""Test the ignoring of sending events if defined."""
|
||||
assert await add_eventstream(hass, pub_topic="bar", ignore_event=["state_changed"])
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert not mock_pub.called
|
||||
e_id = "entity.test_id"
|
||||
event = {}
|
||||
event["event_type"] = EVENT_STATE_CHANGED
|
||||
new_state = {"state": "on", "entity_id": e_id, "attributes": {}}
|
||||
event["event_data"] = {"new_state": new_state, "entity_id": e_id}
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_publish")
|
||||
def test_wrong_ignored_event_sends_over_stream(self, mock_pub):
|
||||
"""Test the ignoring of sending events if defined."""
|
||||
assert self.add_eventstream(pub_topic="bar", ignore_event=["statee_changed"])
|
||||
self.hass.block_till_done()
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_eventstream state change on initialization, etc.
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
|
||||
e_id = "entity.test_id"
|
||||
event = {}
|
||||
event["event_type"] = EVENT_STATE_CHANGED
|
||||
new_state = {"state": "on", "entity_id": e_id, "attributes": {}}
|
||||
event["event_data"] = {"new_state": new_state, "entity_id": e_id}
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(hass, State(e_id, "on"))
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_eventstream state change on initialization, etc.
|
||||
mock_pub.reset_mock()
|
||||
assert not mqtt_mock.async_publish.called
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(self.hass, State(e_id, "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert mock_pub.called
|
||||
async def test_wrong_ignored_event_sends_over_stream(hass, mqtt_mock):
|
||||
"""Test the ignoring of sending events if defined."""
|
||||
assert await add_eventstream(hass, pub_topic="bar", ignore_event=["statee_changed"])
|
||||
await hass.async_block_till_done()
|
||||
|
||||
e_id = "entity.test_id"
|
||||
event = {}
|
||||
event["event_type"] = EVENT_STATE_CHANGED
|
||||
new_state = {"state": "on", "entity_id": e_id, "attributes": {}}
|
||||
event["event_data"] = {"new_state": new_state, "entity_id": e_id}
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_eventstream state change on initialization, etc.
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(hass, State(e_id, "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert mqtt_mock.async_publish.called
|
||||
|
|
|
|||
|
|
@ -3,14 +3,10 @@ import pytest
|
|||
|
||||
import homeassistant.components.mqtt_statestream as statestream
|
||||
from homeassistant.core import State
|
||||
from homeassistant.setup import setup_component
|
||||
from homeassistant.setup import async_setup_component
|
||||
|
||||
from tests.async_mock import ANY, call, patch
|
||||
from tests.common import (
|
||||
get_test_home_assistant,
|
||||
mock_mqtt_component,
|
||||
mock_state_change_event,
|
||||
)
|
||||
from tests.async_mock import ANY, call
|
||||
from tests.common import mock_state_change_event
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
|
|
@ -18,360 +14,343 @@ def mock_storage(hass_storage):
|
|||
"""Autouse hass_storage for the TestCase tests."""
|
||||
|
||||
|
||||
class TestMqttStateStream:
|
||||
"""Test the MQTT statestream module."""
|
||||
|
||||
def setup_method(self):
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.hass = get_test_home_assistant()
|
||||
self.mock_mqtt = mock_mqtt_component(self.hass)
|
||||
|
||||
def teardown_method(self):
|
||||
"""Stop everything that was started."""
|
||||
self.hass.stop()
|
||||
|
||||
def add_statestream(
|
||||
self,
|
||||
base_topic=None,
|
||||
publish_attributes=None,
|
||||
publish_timestamps=None,
|
||||
publish_include=None,
|
||||
publish_exclude=None,
|
||||
):
|
||||
"""Add a mqtt_statestream component."""
|
||||
config = {}
|
||||
if base_topic:
|
||||
config["base_topic"] = base_topic
|
||||
if publish_attributes:
|
||||
config["publish_attributes"] = publish_attributes
|
||||
if publish_timestamps:
|
||||
config["publish_timestamps"] = publish_timestamps
|
||||
if publish_include:
|
||||
config["include"] = publish_include
|
||||
if publish_exclude:
|
||||
config["exclude"] = publish_exclude
|
||||
return setup_component(
|
||||
self.hass, statestream.DOMAIN, {statestream.DOMAIN: config}
|
||||
)
|
||||
|
||||
def test_fails_with_no_base(self):
|
||||
"""Setup should fail if no base_topic is set."""
|
||||
assert self.add_statestream() is False
|
||||
|
||||
def test_setup_succeeds_without_attributes(self):
|
||||
"""Test the success of the setup with a valid base_topic."""
|
||||
assert self.add_statestream(base_topic="pub")
|
||||
|
||||
def test_setup_succeeds_with_attributes(self):
|
||||
"""Test setup with a valid base_topic and publish_attributes."""
|
||||
assert self.add_statestream(base_topic="pub", publish_attributes=True)
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_publish")
|
||||
@patch("homeassistant.core.dt_util.utcnow")
|
||||
def test_state_changed_event_sends_message(self, mock_utcnow, mock_pub):
|
||||
"""Test the sending of a new message if event changed."""
|
||||
e_id = "fake.entity"
|
||||
base_topic = "pub"
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
assert self.add_statestream(base_topic=base_topic)
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mock_pub.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(self.hass, State(e_id, "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mock_pub.assert_called_with(self.hass, "pub/fake/entity/state", "on", 1, True)
|
||||
assert mock_pub.called
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_publish")
|
||||
@patch("homeassistant.core.dt_util.utcnow")
|
||||
def test_state_changed_event_sends_message_and_timestamp(
|
||||
self, mock_utcnow, mock_pub
|
||||
):
|
||||
"""Test the sending of a message and timestamps if event changed."""
|
||||
e_id = "another.entity"
|
||||
base_topic = "pub"
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
assert self.add_statestream(
|
||||
base_topic=base_topic, publish_attributes=None, publish_timestamps=True
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mock_pub.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(self.hass, State(e_id, "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
calls = [
|
||||
call.async_publish(self.hass, "pub/another/entity/state", "on", 1, True),
|
||||
call.async_publish(
|
||||
self.hass, "pub/another/entity/last_changed", ANY, 1, True
|
||||
),
|
||||
call.async_publish(
|
||||
self.hass, "pub/another/entity/last_updated", ANY, 1, True
|
||||
),
|
||||
]
|
||||
|
||||
mock_pub.assert_has_calls(calls, any_order=True)
|
||||
assert mock_pub.called
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_publish")
|
||||
@patch("homeassistant.core.dt_util.utcnow")
|
||||
def test_state_changed_attr_sends_message(self, mock_utcnow, mock_pub):
|
||||
"""Test the sending of a new message if attribute changed."""
|
||||
e_id = "fake.entity"
|
||||
base_topic = "pub"
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
assert self.add_statestream(base_topic=base_topic, publish_attributes=True)
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mock_pub.reset_mock()
|
||||
|
||||
test_attributes = {"testing": "YES", "list": ["a", "b", "c"], "bool": False}
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(
|
||||
self.hass, State(e_id, "off", attributes=test_attributes)
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
calls = [
|
||||
call.async_publish(self.hass, "pub/fake/entity/state", "off", 1, True),
|
||||
call.async_publish(self.hass, "pub/fake/entity/testing", '"YES"', 1, True),
|
||||
call.async_publish(
|
||||
self.hass, "pub/fake/entity/list", '["a", "b", "c"]', 1, True
|
||||
),
|
||||
call.async_publish(self.hass, "pub/fake/entity/bool", "false", 1, True),
|
||||
]
|
||||
|
||||
mock_pub.assert_has_calls(calls, any_order=True)
|
||||
assert mock_pub.called
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_publish")
|
||||
@patch("homeassistant.core.dt_util.utcnow")
|
||||
def test_state_changed_event_include_domain(self, mock_utcnow, mock_pub):
|
||||
"""Test that filtering on included domain works as expected."""
|
||||
base_topic = "pub"
|
||||
|
||||
incl = {"domains": ["fake"]}
|
||||
excl = {}
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
# Set the filter to allow fake.* items
|
||||
assert self.add_statestream(
|
||||
base_topic=base_topic, publish_include=incl, publish_exclude=excl
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mock_pub.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(self.hass, State("fake.entity", "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mock_pub.assert_called_with(self.hass, "pub/fake/entity/state", "on", 1, True)
|
||||
assert mock_pub.called
|
||||
|
||||
mock_pub.reset_mock()
|
||||
# Set a state of an entity that shouldn't be included
|
||||
mock_state_change_event(self.hass, State("fake2.entity", "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert not mock_pub.called
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_publish")
|
||||
@patch("homeassistant.core.dt_util.utcnow")
|
||||
def test_state_changed_event_include_entity(self, mock_utcnow, mock_pub):
|
||||
"""Test that filtering on included entity works as expected."""
|
||||
base_topic = "pub"
|
||||
|
||||
incl = {"entities": ["fake.entity"]}
|
||||
excl = {}
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
# Set the filter to allow fake.* items
|
||||
assert self.add_statestream(
|
||||
base_topic=base_topic, publish_include=incl, publish_exclude=excl
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mock_pub.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(self.hass, State("fake.entity", "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mock_pub.assert_called_with(self.hass, "pub/fake/entity/state", "on", 1, True)
|
||||
assert mock_pub.called
|
||||
|
||||
mock_pub.reset_mock()
|
||||
# Set a state of an entity that shouldn't be included
|
||||
mock_state_change_event(self.hass, State("fake.entity2", "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert not mock_pub.called
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_publish")
|
||||
@patch("homeassistant.core.dt_util.utcnow")
|
||||
def test_state_changed_event_exclude_domain(self, mock_utcnow, mock_pub):
|
||||
"""Test that filtering on excluded domain works as expected."""
|
||||
base_topic = "pub"
|
||||
|
||||
incl = {}
|
||||
excl = {"domains": ["fake2"]}
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
# Set the filter to allow fake.* items
|
||||
assert self.add_statestream(
|
||||
base_topic=base_topic, publish_include=incl, publish_exclude=excl
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mock_pub.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(self.hass, State("fake.entity", "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mock_pub.assert_called_with(self.hass, "pub/fake/entity/state", "on", 1, True)
|
||||
assert mock_pub.called
|
||||
|
||||
mock_pub.reset_mock()
|
||||
# Set a state of an entity that shouldn't be included
|
||||
mock_state_change_event(self.hass, State("fake2.entity", "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert not mock_pub.called
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_publish")
|
||||
@patch("homeassistant.core.dt_util.utcnow")
|
||||
def test_state_changed_event_exclude_entity(self, mock_utcnow, mock_pub):
|
||||
"""Test that filtering on excluded entity works as expected."""
|
||||
base_topic = "pub"
|
||||
|
||||
incl = {}
|
||||
excl = {"entities": ["fake.entity2"]}
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
# Set the filter to allow fake.* items
|
||||
assert self.add_statestream(
|
||||
base_topic=base_topic, publish_include=incl, publish_exclude=excl
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mock_pub.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(self.hass, State("fake.entity", "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mock_pub.assert_called_with(self.hass, "pub/fake/entity/state", "on", 1, True)
|
||||
assert mock_pub.called
|
||||
|
||||
mock_pub.reset_mock()
|
||||
# Set a state of an entity that shouldn't be included
|
||||
mock_state_change_event(self.hass, State("fake.entity2", "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert not mock_pub.called
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_publish")
|
||||
@patch("homeassistant.core.dt_util.utcnow")
|
||||
def test_state_changed_event_exclude_domain_include_entity(
|
||||
self, mock_utcnow, mock_pub
|
||||
):
|
||||
"""Test filtering with excluded domain and included entity."""
|
||||
base_topic = "pub"
|
||||
|
||||
incl = {"entities": ["fake.entity"]}
|
||||
excl = {"domains": ["fake"]}
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
# Set the filter to allow fake.* items
|
||||
assert self.add_statestream(
|
||||
base_topic=base_topic, publish_include=incl, publish_exclude=excl
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mock_pub.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(self.hass, State("fake.entity", "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mock_pub.assert_called_with(self.hass, "pub/fake/entity/state", "on", 1, True)
|
||||
assert mock_pub.called
|
||||
|
||||
mock_pub.reset_mock()
|
||||
# Set a state of an entity that shouldn't be included
|
||||
mock_state_change_event(self.hass, State("fake.entity2", "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert not mock_pub.called
|
||||
|
||||
@patch("homeassistant.components.mqtt.async_publish")
|
||||
@patch("homeassistant.core.dt_util.utcnow")
|
||||
def test_state_changed_event_include_domain_exclude_entity(
|
||||
self, mock_utcnow, mock_pub
|
||||
):
|
||||
"""Test filtering with included domain and excluded entity."""
|
||||
base_topic = "pub"
|
||||
|
||||
incl = {"domains": ["fake"]}
|
||||
excl = {"entities": ["fake.entity2"]}
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
# Set the filter to allow fake.* items
|
||||
assert self.add_statestream(
|
||||
base_topic=base_topic, publish_include=incl, publish_exclude=excl
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mock_pub.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(self.hass, State("fake.entity", "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mock_pub.assert_called_with(self.hass, "pub/fake/entity/state", "on", 1, True)
|
||||
assert mock_pub.called
|
||||
|
||||
mock_pub.reset_mock()
|
||||
# Set a state of an entity that shouldn't be included
|
||||
mock_state_change_event(self.hass, State("fake.entity2", "on"))
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert not mock_pub.called
|
||||
async def add_statestream(
|
||||
hass,
|
||||
base_topic=None,
|
||||
publish_attributes=None,
|
||||
publish_timestamps=None,
|
||||
publish_include=None,
|
||||
publish_exclude=None,
|
||||
):
|
||||
"""Add a mqtt_statestream component."""
|
||||
config = {}
|
||||
if base_topic:
|
||||
config["base_topic"] = base_topic
|
||||
if publish_attributes:
|
||||
config["publish_attributes"] = publish_attributes
|
||||
if publish_timestamps:
|
||||
config["publish_timestamps"] = publish_timestamps
|
||||
if publish_include:
|
||||
config["include"] = publish_include
|
||||
if publish_exclude:
|
||||
config["exclude"] = publish_exclude
|
||||
return await async_setup_component(
|
||||
hass, statestream.DOMAIN, {statestream.DOMAIN: config}
|
||||
)
|
||||
|
||||
|
||||
async def test_fails_with_no_base(hass, mqtt_mock):
|
||||
"""Setup should fail if no base_topic is set."""
|
||||
assert await add_statestream(hass) is False
|
||||
|
||||
|
||||
async def test_setup_succeeds_without_attributes(hass, mqtt_mock):
|
||||
"""Test the success of the setup with a valid base_topic."""
|
||||
assert await add_statestream(hass, base_topic="pub")
|
||||
|
||||
|
||||
async def test_setup_succeeds_with_attributes(hass, mqtt_mock):
|
||||
"""Test setup with a valid base_topic and publish_attributes."""
|
||||
assert await add_statestream(hass, base_topic="pub", publish_attributes=True)
|
||||
|
||||
|
||||
async def test_state_changed_event_sends_message(hass, mqtt_mock):
|
||||
"""Test the sending of a new message if event changed."""
|
||||
e_id = "fake.entity"
|
||||
base_topic = "pub"
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
assert await add_statestream(hass, base_topic=base_topic)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(hass, State(e_id, "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mqtt_mock.async_publish.assert_called_with("pub/fake/entity/state", "on", 1, True)
|
||||
assert mqtt_mock.async_publish.called
|
||||
|
||||
|
||||
async def test_state_changed_event_sends_message_and_timestamp(hass, mqtt_mock):
|
||||
"""Test the sending of a message and timestamps if event changed."""
|
||||
e_id = "another.entity"
|
||||
base_topic = "pub"
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
assert await add_statestream(
|
||||
hass, base_topic=base_topic, publish_attributes=None, publish_timestamps=True
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(hass, State(e_id, "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
calls = [
|
||||
call.async_publish("pub/another/entity/state", "on", 1, True),
|
||||
call.async_publish("pub/another/entity/last_changed", ANY, 1, True),
|
||||
call.async_publish("pub/another/entity/last_updated", ANY, 1, True),
|
||||
]
|
||||
|
||||
mqtt_mock.async_publish.assert_has_calls(calls, any_order=True)
|
||||
assert mqtt_mock.async_publish.called
|
||||
|
||||
|
||||
async def test_state_changed_attr_sends_message(hass, mqtt_mock):
|
||||
"""Test the sending of a new message if attribute changed."""
|
||||
e_id = "fake.entity"
|
||||
base_topic = "pub"
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
assert await add_statestream(hass, base_topic=base_topic, publish_attributes=True)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
|
||||
test_attributes = {"testing": "YES", "list": ["a", "b", "c"], "bool": False}
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(hass, State(e_id, "off", attributes=test_attributes))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
calls = [
|
||||
call.async_publish("pub/fake/entity/state", "off", 1, True),
|
||||
call.async_publish("pub/fake/entity/testing", '"YES"', 1, True),
|
||||
call.async_publish("pub/fake/entity/list", '["a", "b", "c"]', 1, True),
|
||||
call.async_publish("pub/fake/entity/bool", "false", 1, True),
|
||||
]
|
||||
|
||||
mqtt_mock.async_publish.assert_has_calls(calls, any_order=True)
|
||||
assert mqtt_mock.async_publish.called
|
||||
|
||||
|
||||
async def test_state_changed_event_include_domain(hass, mqtt_mock):
|
||||
"""Test that filtering on included domain works as expected."""
|
||||
base_topic = "pub"
|
||||
|
||||
incl = {"domains": ["fake"]}
|
||||
excl = {}
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
# Set the filter to allow fake.* items
|
||||
assert await add_statestream(
|
||||
hass, base_topic=base_topic, publish_include=incl, publish_exclude=excl
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(hass, State("fake.entity", "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mqtt_mock.async_publish.assert_called_with("pub/fake/entity/state", "on", 1, True)
|
||||
assert mqtt_mock.async_publish.called
|
||||
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
# Set a state of an entity that shouldn't be included
|
||||
mock_state_change_event(hass, State("fake2.entity", "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert not mqtt_mock.async_publish.called
|
||||
|
||||
|
||||
async def test_state_changed_event_include_entity(hass, mqtt_mock):
|
||||
"""Test that filtering on included entity works as expected."""
|
||||
base_topic = "pub"
|
||||
|
||||
incl = {"entities": ["fake.entity"]}
|
||||
excl = {}
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
# Set the filter to allow fake.* items
|
||||
assert await add_statestream(
|
||||
hass, base_topic=base_topic, publish_include=incl, publish_exclude=excl
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(hass, State("fake.entity", "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mqtt_mock.async_publish.assert_called_with("pub/fake/entity/state", "on", 1, True)
|
||||
assert mqtt_mock.async_publish.called
|
||||
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
# Set a state of an entity that shouldn't be included
|
||||
mock_state_change_event(hass, State("fake.entity2", "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert not mqtt_mock.async_publish.called
|
||||
|
||||
|
||||
async def test_state_changed_event_exclude_domain(hass, mqtt_mock):
|
||||
"""Test that filtering on excluded domain works as expected."""
|
||||
base_topic = "pub"
|
||||
|
||||
incl = {}
|
||||
excl = {"domains": ["fake2"]}
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
# Set the filter to allow fake.* items
|
||||
assert await add_statestream(
|
||||
hass, base_topic=base_topic, publish_include=incl, publish_exclude=excl
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(hass, State("fake.entity", "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mqtt_mock.async_publish.assert_called_with("pub/fake/entity/state", "on", 1, True)
|
||||
assert mqtt_mock.async_publish.called
|
||||
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
# Set a state of an entity that shouldn't be included
|
||||
mock_state_change_event(hass, State("fake2.entity", "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert not mqtt_mock.async_publish.called
|
||||
|
||||
|
||||
async def test_state_changed_event_exclude_entity(hass, mqtt_mock):
|
||||
"""Test that filtering on excluded entity works as expected."""
|
||||
base_topic = "pub"
|
||||
|
||||
incl = {}
|
||||
excl = {"entities": ["fake.entity2"]}
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
# Set the filter to allow fake.* items
|
||||
assert await add_statestream(
|
||||
hass, base_topic=base_topic, publish_include=incl, publish_exclude=excl
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(hass, State("fake.entity", "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mqtt_mock.async_publish.assert_called_with("pub/fake/entity/state", "on", 1, True)
|
||||
assert mqtt_mock.async_publish.called
|
||||
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
# Set a state of an entity that shouldn't be included
|
||||
mock_state_change_event(hass, State("fake.entity2", "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert not mqtt_mock.async_publish.called
|
||||
|
||||
|
||||
async def test_state_changed_event_exclude_domain_include_entity(hass, mqtt_mock):
|
||||
"""Test filtering with excluded domain and included entity."""
|
||||
base_topic = "pub"
|
||||
|
||||
incl = {"entities": ["fake.entity"]}
|
||||
excl = {"domains": ["fake"]}
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
# Set the filter to allow fake.* items
|
||||
assert await add_statestream(
|
||||
hass, base_topic=base_topic, publish_include=incl, publish_exclude=excl
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(hass, State("fake.entity", "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mqtt_mock.async_publish.assert_called_with("pub/fake/entity/state", "on", 1, True)
|
||||
assert mqtt_mock.async_publish.called
|
||||
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
# Set a state of an entity that shouldn't be included
|
||||
mock_state_change_event(hass, State("fake.entity2", "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert not mqtt_mock.async_publish.called
|
||||
|
||||
|
||||
async def test_state_changed_event_include_domain_exclude_entity(hass, mqtt_mock):
|
||||
"""Test filtering with included domain and excluded entity."""
|
||||
base_topic = "pub"
|
||||
|
||||
incl = {"domains": ["fake"]}
|
||||
excl = {"entities": ["fake.entity2"]}
|
||||
|
||||
# Add the statestream component for publishing state updates
|
||||
# Set the filter to allow fake.* items
|
||||
assert await add_statestream(
|
||||
hass, base_topic=base_topic, publish_include=incl, publish_exclude=excl
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Reset the mock because it will have already gotten calls for the
|
||||
# mqtt_statestream state change on initialization, etc.
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
|
||||
# Set a state of an entity
|
||||
mock_state_change_event(hass, State("fake.entity", "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Make sure 'on' was published to pub/fake/entity/state
|
||||
mqtt_mock.async_publish.assert_called_with("pub/fake/entity/state", "on", 1, True)
|
||||
assert mqtt_mock.async_publish.called
|
||||
|
||||
mqtt_mock.async_publish.reset_mock()
|
||||
# Set a state of an entity that shouldn't be included
|
||||
mock_state_change_event(hass, State("fake.entity2", "on"))
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert not mqtt_mock.async_publish.called
|
||||
|
|
|
|||
|
|
@ -5,9 +5,10 @@ import pytest
|
|||
|
||||
import homeassistant.components.nextbus.sensor as nextbus
|
||||
import homeassistant.components.sensor as sensor
|
||||
from homeassistant.setup import async_setup_component
|
||||
|
||||
from tests.async_mock import patch
|
||||
from tests.common import assert_setup_component, async_setup_component
|
||||
from tests.common import assert_setup_component
|
||||
|
||||
VALID_AGENCY = "sf-muni"
|
||||
VALID_ROUTE = "F"
|
||||
|
|
|
|||
|
|
@ -2,12 +2,13 @@
|
|||
|
||||
from homeassistant.components import pi_hole
|
||||
from homeassistant.components.pi_hole.const import MIN_TIME_BETWEEN_UPDATES
|
||||
from homeassistant.setup import async_setup_component
|
||||
from homeassistant.util import dt as dt_util
|
||||
|
||||
from . import _create_mocked_hole, _patch_config_flow_hole
|
||||
|
||||
from tests.async_mock import patch
|
||||
from tests.common import async_fire_time_changed, async_setup_component
|
||||
from tests.common import async_fire_time_changed
|
||||
|
||||
|
||||
def _patch_init_hole(mocked_hole):
|
||||
|
|
|
|||
|
|
@ -7,11 +7,12 @@ from pytz import timezone
|
|||
from homeassistant.components.pvpc_hourly_pricing import ATTR_TARIFF, DOMAIN
|
||||
from homeassistant.const import CONF_NAME
|
||||
from homeassistant.core import ATTR_NOW, EVENT_TIME_CHANGED
|
||||
from homeassistant.setup import async_setup_component
|
||||
|
||||
from .conftest import check_valid_state
|
||||
|
||||
from tests.async_mock import patch
|
||||
from tests.common import async_setup_component, date_util
|
||||
from tests.common import date_util
|
||||
from tests.test_util.aiohttp import AiohttpClientMocker
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -25,7 +25,8 @@ from .common import (
|
|||
new_profile_config,
|
||||
)
|
||||
|
||||
from tests.common import MagicMock, MockConfigEntry, patch
|
||||
from tests.async_mock import MagicMock, patch
|
||||
from tests.common import MockConfigEntry
|
||||
|
||||
|
||||
def config_schema_validate(withings_config) -> dict:
|
||||
|
|
|
|||
|
|
@ -6,10 +6,11 @@ import pytest
|
|||
|
||||
import homeassistant.components.sensor as sensor
|
||||
from homeassistant.const import CONF_NAME
|
||||
from homeassistant.setup import async_setup_component
|
||||
import homeassistant.util.dt as dt_util
|
||||
|
||||
from tests.async_mock import patch
|
||||
from tests.common import assert_setup_component, async_setup_component, load_fixture
|
||||
from tests.common import assert_setup_component, load_fixture
|
||||
|
||||
REPLY = json.loads(load_fixture("yandex_transport_reply.json"))
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue