mirror of
https://github.com/home-assistant/core.git
synced 2025-07-11 23:37:18 +00:00
Fix mock_integration and mock_platform test helpers (#98109)
This commit is contained in:
parent
023f2f8bb7
commit
0317afeb17
@ -1336,8 +1336,17 @@ def mock_integration(
|
|||||||
integration._import_platform = mock_import_platform
|
integration._import_platform = mock_import_platform
|
||||||
|
|
||||||
_LOGGER.info("Adding mock integration: %s", module.DOMAIN)
|
_LOGGER.info("Adding mock integration: %s", module.DOMAIN)
|
||||||
hass.data.setdefault(loader.DATA_INTEGRATIONS, {})[module.DOMAIN] = integration
|
integration_cache = hass.data.get(loader.DATA_INTEGRATIONS)
|
||||||
hass.data.setdefault(loader.DATA_COMPONENTS, {})[module.DOMAIN] = module
|
if integration_cache is None:
|
||||||
|
integration_cache = hass.data[loader.DATA_INTEGRATIONS] = {}
|
||||||
|
loader._async_mount_config_dir(hass)
|
||||||
|
integration_cache[module.DOMAIN] = integration
|
||||||
|
|
||||||
|
module_cache = hass.data.get(loader.DATA_COMPONENTS)
|
||||||
|
if module_cache is None:
|
||||||
|
module_cache = hass.data[loader.DATA_COMPONENTS] = {}
|
||||||
|
loader._async_mount_config_dir(hass)
|
||||||
|
module_cache[module.DOMAIN] = module
|
||||||
|
|
||||||
return integration
|
return integration
|
||||||
|
|
||||||
@ -1361,9 +1370,16 @@ def mock_platform(
|
|||||||
|
|
||||||
platform_path is in form hue.config_flow.
|
platform_path is in form hue.config_flow.
|
||||||
"""
|
"""
|
||||||
domain, platform_name = platform_path.split(".")
|
domain = platform_path.split(".")[0]
|
||||||
integration_cache = hass.data.setdefault(loader.DATA_INTEGRATIONS, {})
|
integration_cache = hass.data.get(loader.DATA_INTEGRATIONS)
|
||||||
module_cache = hass.data.setdefault(loader.DATA_COMPONENTS, {})
|
if integration_cache is None:
|
||||||
|
integration_cache = hass.data[loader.DATA_INTEGRATIONS] = {}
|
||||||
|
loader._async_mount_config_dir(hass)
|
||||||
|
|
||||||
|
module_cache = hass.data.get(loader.DATA_COMPONENTS)
|
||||||
|
if module_cache is None:
|
||||||
|
module_cache = hass.data[loader.DATA_COMPONENTS] = {}
|
||||||
|
loader._async_mount_config_dir(hass)
|
||||||
|
|
||||||
if domain not in integration_cache:
|
if domain not in integration_cache:
|
||||||
mock_integration(hass, MockModule(domain))
|
mock_integration(hass, MockModule(domain))
|
||||||
|
Loading…
x
Reference in New Issue
Block a user