Convert discovery helper to use dispatcher (#47008)

This commit is contained in:
Paulus Schoutsen
2021-02-24 13:37:31 -08:00
committed by GitHub
parent 5ab11df551
commit 557ec374f1
7 changed files with 79 additions and 149 deletions

View File

@ -441,10 +441,14 @@ class TestSetup:
"""Test all init work done till start."""
call_order = []
def component1_setup(hass, config):
async def component1_setup(hass, config):
"""Set up mock component."""
discovery.discover(hass, "test_component2", {}, "test_component2", {})
discovery.discover(hass, "test_component3", {}, "test_component3", {})
await discovery.async_discover(
hass, "test_component2", {}, "test_component2", {}
)
await discovery.async_discover(
hass, "test_component3", {}, "test_component3", {}
)
return True
def component_track_setup(hass, config):
@ -453,7 +457,7 @@ class TestSetup:
return True
mock_integration(
self.hass, MockModule("test_component1", setup=component1_setup)
self.hass, MockModule("test_component1", async_setup=component1_setup)
)
mock_integration(