Fix implicit-return in test schema extractions (#122787)

This commit is contained in:
epenet 2024-07-29 22:16:00 +02:00 committed by GitHub
parent 7b08e625b4
commit fdab23c3f9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 6 additions and 2 deletions

View File

@ -148,6 +148,7 @@ def get_suggested(schema, key):
if k.description is None or "suggested_value" not in k.description:
return None
return k.description["suggested_value"]
return None
@pytest.mark.parametrize(

View File

@ -850,16 +850,17 @@ async def test_invalid_discovery_prefix(
assert mock_reload_after_entry_update.call_count == 0
def get_default(schema: vol.Schema, key: str) -> Any:
def get_default(schema: vol.Schema, key: str) -> Any | None:
"""Get default value for key in voluptuous schema."""
for schema_key in schema:
if schema_key == key:
if schema_key.default == vol.UNDEFINED:
return None
return schema_key.default()
return None
def get_suggested(schema: vol.Schema, key: str) -> Any:
def get_suggested(schema: vol.Schema, key: str) -> Any | None:
"""Get suggested value for key in voluptuous schema."""
for schema_key in schema:
if schema_key == key:
@ -869,6 +870,7 @@ def get_suggested(schema: vol.Schema, key: str) -> Any:
):
return None
return schema_key.description["suggested_value"]
return None
@pytest.mark.usefixtures("mock_reload_after_entry_update")

View File

@ -32,3 +32,4 @@ def get_schema_suggestion(schema, key):
if k.description is None or "suggested_value" not in k.description:
return None
return k.description["suggested_value"]
return None