diff --git a/tests/test_setup.py b/tests/test_setup.py index 4ff0f465e21..1e19f1a7b76 100644 --- a/tests/test_setup.py +++ b/tests/test_setup.py @@ -12,11 +12,7 @@ from homeassistant import config_entries, loader, setup from homeassistant.const import EVENT_COMPONENT_LOADED, EVENT_HOMEASSISTANT_START from homeassistant.core import CoreState, HomeAssistant, callback from homeassistant.exceptions import HomeAssistantError -from homeassistant.helpers import discovery, translation -from homeassistant.helpers.config_validation import ( - PLATFORM_SCHEMA, - PLATFORM_SCHEMA_BASE, -) +from homeassistant.helpers import config_validation as cv, discovery, translation from homeassistant.helpers.dispatcher import ( async_dispatcher_connect, async_dispatcher_send, @@ -88,8 +84,8 @@ async def test_validate_platform_config( hass: HomeAssistant, caplog: pytest.LogCaptureFixture ) -> None: """Test validating platform configuration.""" - platform_schema = PLATFORM_SCHEMA.extend({"hello": str}) - platform_schema_base = PLATFORM_SCHEMA_BASE.extend({}) + platform_schema = cv.PLATFORM_SCHEMA.extend({"hello": str}) + platform_schema_base = cv.PLATFORM_SCHEMA_BASE.extend({}) mock_integration( hass, MockModule("platform_conf", platform_schema_base=platform_schema_base), @@ -149,8 +145,8 @@ async def test_validate_platform_config_2( hass: HomeAssistant, caplog: pytest.LogCaptureFixture ) -> None: """Test component PLATFORM_SCHEMA_BASE prio over PLATFORM_SCHEMA.""" - platform_schema = PLATFORM_SCHEMA.extend({"hello": str}) - platform_schema_base = PLATFORM_SCHEMA_BASE.extend({"hello": "world"}) + platform_schema = cv.PLATFORM_SCHEMA.extend({"hello": str}) + platform_schema_base = cv.PLATFORM_SCHEMA_BASE.extend({"hello": "world"}) mock_integration( hass, MockModule( @@ -183,8 +179,8 @@ async def test_validate_platform_config_3( hass: HomeAssistant, caplog: pytest.LogCaptureFixture ) -> None: """Test fallback to component PLATFORM_SCHEMA.""" - component_schema = PLATFORM_SCHEMA_BASE.extend({"hello": str}) - platform_schema = PLATFORM_SCHEMA.extend({"cheers": str, "hello": "world"}) + component_schema = cv.PLATFORM_SCHEMA_BASE.extend({"hello": str}) + platform_schema = cv.PLATFORM_SCHEMA.extend({"cheers": str, "hello": "world"}) mock_integration( hass, MockModule("platform_conf", platform_schema=component_schema) ) @@ -210,8 +206,8 @@ async def test_validate_platform_config_3( async def test_validate_platform_config_4(hass: HomeAssistant) -> None: """Test entity_namespace in PLATFORM_SCHEMA.""" - component_schema = PLATFORM_SCHEMA_BASE - platform_schema = PLATFORM_SCHEMA + component_schema = cv.PLATFORM_SCHEMA_BASE + platform_schema = cv.PLATFORM_SCHEMA mock_integration( hass, MockModule("platform_conf", platform_schema_base=component_schema), @@ -386,7 +382,9 @@ async def test_component_setup_with_validation_and_dependency( async def test_platform_specific_config_validation(hass: HomeAssistant) -> None: """Test platform that specifies config.""" - platform_schema = PLATFORM_SCHEMA.extend({"valid": True}, extra=vol.PREVENT_EXTRA) + platform_schema = cv.PLATFORM_SCHEMA.extend( + {"valid": True}, extra=vol.PREVENT_EXTRA + ) mock_setup = Mock(spec_set=True) @@ -533,7 +531,7 @@ async def test_component_warn_slow_setup(hass: HomeAssistant) -> None: async def test_platform_no_warn_slow(hass: HomeAssistant) -> None: """Do not warn for long entity setup time.""" mock_integration( - hass, MockModule("test_component1", platform_schema=PLATFORM_SCHEMA) + hass, MockModule("test_component1", platform_schema=cv.PLATFORM_SCHEMA) ) with patch.object(hass.loop, "call_later") as mock_call: result = await setup.async_setup_component(hass, "test_component1", {})