Use helper for testing an event change (#71579)

This commit is contained in:
Jan Bouwhuis 2022-05-09 11:52:08 +02:00 committed by GitHub
parent bb4a5ccc2c
commit 5d9dc8252b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -24,6 +24,7 @@ from homeassistant.setup import async_setup_component
from tests.common import (
MockConfigEntry,
async_capture_events,
async_fire_mqtt_message,
mock_device_registry,
mock_entity_platform,
@ -438,14 +439,7 @@ async def test_rediscover(hass, mqtt_mock, caplog):
async def test_rapid_rediscover(hass, mqtt_mock, caplog):
"""Test immediate rediscover of removed component."""
events = []
@ha.callback
def callback(event):
"""Verify event got called."""
events.append(event)
hass.bus.async_listen(EVENT_STATE_CHANGED, callback)
events = async_capture_events(hass, EVENT_STATE_CHANGED)
async_fire_mqtt_message(
hass,