jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-14 21:40:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add slots to core EventBus (#95856)

Browse Source
This commit is contained in:
J. Nick Koston
2023-07-05 11:47:24 -05:00
committed by GitHub
parent 20dc9203dd
commit dc5ee71d7a
8 changed files with 217 additions and 411 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
tests/components/ffmpeg/test_init.py
View File

@@ -8,7 +8,11 @@ from homeassistant.components.ffmpeg import (
SERVICE_START,
SERVICE_STOP,
)
from homeassistant.const import ATTR_ENTITY_ID
from homeassistant.const import (
ATTR_ENTITY_ID,
EVENT_HOMEASSISTANT_START,
EVENT_HOMEASSISTANT_STOP,
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.setup import async_setup_component, setup_component
@@ -54,7 +58,7 @@ class MockFFmpegDev(ffmpeg.FFmpegBase):
self.hass = hass
self.entity_id = entity_id
self.ffmpeg = MagicMock
self.ffmpeg = MagicMock()
self.called_stop = False
self.called_start = False
self.called_restart = False
@@ -104,12 +108,18 @@ async def test_setup_component_test_register(hass: HomeAssistant) -> None:
with assert_setup_component(1):
await async_setup_component(hass, ffmpeg.DOMAIN, {ffmpeg.DOMAIN: {}})
hass.bus.async_listen_once = MagicMock()
ffmpeg_dev = MockFFmpegDev(hass)
ffmpeg_dev._async_stop_ffmpeg = AsyncMock()
ffmpeg_dev._async_start_ffmpeg = AsyncMock()
await ffmpeg_dev.async_added_to_hass()
assert hass.bus.async_listen_once.called
assert hass.bus.async_listen_once.call_count == 2
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
await hass.async_block_till_done()
assert len(ffmpeg_dev._async_start_ffmpeg.mock_calls) == 2
hass.bus.async_fire(EVENT_HOMEASSISTANT_STOP)
await hass.async_block_till_done()
assert len(ffmpeg_dev._async_stop_ffmpeg.mock_calls) == 2
async def test_setup_component_test_register_no_startup(hass: HomeAssistant) -> None:
@@ -117,12 +127,18 @@ async def test_setup_component_test_register_no_startup(hass: HomeAssistant) ->
with assert_setup_component(1):
await async_setup_component(hass, ffmpeg.DOMAIN, {ffmpeg.DOMAIN: {}})
hass.bus.async_listen_once = MagicMock()
ffmpeg_dev = MockFFmpegDev(hass, False)
ffmpeg_dev._async_stop_ffmpeg = AsyncMock()
ffmpeg_dev._async_start_ffmpeg = AsyncMock()
await ffmpeg_dev.async_added_to_hass()
assert hass.bus.async_listen_once.called
assert hass.bus.async_listen_once.call_count == 1
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
await hass.async_block_till_done()
assert len(ffmpeg_dev._async_start_ffmpeg.mock_calls) == 1
hass.bus.async_fire(EVENT_HOMEASSISTANT_STOP)
await hass.async_block_till_done()
assert len(ffmpeg_dev._async_stop_ffmpeg.mock_calls) == 2
async def test_setup_component_test_service_start(hass: HomeAssistant) -> None: