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

Add MockPlatform type hints in tests (#120012)

Browse Source 
* Add MockPlatform type hints in tests

* Remove useless code

* Improve

* Revert "Improve"

This reverts commit 9ad04f9255.
This commit is contained in:
epenet
2024-06-21 11:04:15 +02:00
committed by GitHub
parent 53022df8a4
commit dc6c1f4e87
7 changed files with 43 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
tests/helpers/test_entity_component.py
View File

@@ -52,7 +52,7 @@ async def test_setup_loads_platforms(hass: HomeAssistant) -> None:
mock_integration(hass, MockModule("test_component", setup=component_setup))
# mock the dependencies
mock_integration(hass, MockModule("mod2", dependencies=["test_component"]))
mock_platform(hass, "mod2.test_domain", MockPlatform(platform_setup))
mock_platform(hass, "mod2.test_domain", MockPlatform(setup_platform=platform_setup))
component = EntityComponent(_LOGGER, DOMAIN, hass)
@@ -71,8 +71,12 @@ async def test_setup_recovers_when_setup_raises(hass: HomeAssistant) -> None:
platform1_setup = Mock(side_effect=Exception("Broken"))
platform2_setup = Mock(return_value=None)
mock_platform(hass, "mod1.test_domain", MockPlatform(platform1_setup))
mock_platform(hass, "mod2.test_domain", MockPlatform(platform2_setup))
mock_platform(
hass, "mod1.test_domain", MockPlatform(setup_platform=platform1_setup)
)
mock_platform(
hass, "mod2.test_domain", MockPlatform(setup_platform=platform2_setup)
)
component = EntityComponent(_LOGGER, DOMAIN, hass)
@@ -128,7 +132,9 @@ async def test_set_scan_interval_via_config(hass: HomeAssistant) -> None:
"""Test the platform setup."""
add_entities([MockEntity(should_poll=True)])
mock_platform(hass, "platform.test_domain", MockPlatform(platform_setup))
mock_platform(
hass, "platform.test_domain", MockPlatform(setup_platform=platform_setup)
)
component = EntityComponent(_LOGGER, DOMAIN, hass)
@@ -154,7 +160,7 @@ async def test_set_entity_namespace_via_config(hass: HomeAssistant) -> None:
"""Test the platform setup."""
add_entities([MockEntity(name="beer"), MockEntity(name=None)])
platform = MockPlatform(platform_setup)
platform = MockPlatform(setup_platform=platform_setup)
mock_platform(hass, "platform.test_domain", platform)
@@ -204,7 +210,9 @@ async def test_platform_not_ready(hass: HomeAssistant) -> None:
"""Test that we retry when platform not ready."""
platform1_setup = Mock(side_effect=[PlatformNotReady, PlatformNotReady, None])
mock_integration(hass, MockModule("mod1"))
mock_platform(hass, "mod1.test_domain", MockPlatform(platform1_setup))
mock_platform(
hass, "mod1.test_domain", MockPlatform(setup_platform=platform1_setup)
)
component = EntityComponent(_LOGGER, DOMAIN, hass)
@@ -678,7 +686,9 @@ async def test_platforms_shutdown_on_stop(hass: HomeAssistant) -> None:
"""Test that we shutdown platforms on stop."""
platform1_setup = Mock(side_effect=[PlatformNotReady, PlatformNotReady, None])
mock_integration(hass, MockModule("mod1"))
mock_platform(hass, "mod1.test_domain", MockPlatform(platform1_setup))
mock_platform(
hass, "mod1.test_domain", MockPlatform(setup_platform=platform1_setup)
)
component = EntityComponent(_LOGGER, DOMAIN, hass)