Fix hassfest allowing omitting discovery methods when using OAuth2Flow (#30732)

This commit is contained in:
Franck Nijhof 2020-01-13 14:36:47 +01:00 committed by Paulus Schoutsen
parent 1f9d6ba541
commit 040b283a14
2 changed files with 4 additions and 0 deletions

View File

@ -43,6 +43,7 @@ def generate_and_validate(integrations: Dict[str, Integration]):
content = fp.read()
if (
" async_step_ssdp" not in content
and "AbstractOAuth2FlowHandler" not in content
and "register_discovery_flow" not in content
):
integration.add_error("ssdp", "Config flow has no async_step_ssdp")

View File

@ -41,10 +41,12 @@ def generate_and_validate(integrations: Dict[str, Integration]):
with open(str(integration.path / "config_flow.py")) as fp:
content = fp.read()
uses_discovery_flow = "register_discovery_flow" in content
uses_oauth2_flow = "AbstractOAuth2FlowHandler" in content
if (
service_types
and not uses_discovery_flow
and not uses_oauth2_flow
and " async_step_zeroconf" not in content
):
integration.add_error(
@ -55,6 +57,7 @@ def generate_and_validate(integrations: Dict[str, Integration]):
if (
homekit_models
and not uses_discovery_flow
and not uses_oauth2_flow
and " async_step_homekit" not in content
):
integration.add_error(