diff --git a/script/hassfest/ssdp.py b/script/hassfest/ssdp.py index 5ee2076ecf4..a32e07f4aac 100644 --- a/script/hassfest/ssdp.py +++ b/script/hassfest/ssdp.py @@ -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") diff --git a/script/hassfest/zeroconf.py b/script/hassfest/zeroconf.py index 2a1bb936871..7e1b7eae727 100644 --- a/script/hassfest/zeroconf.py +++ b/script/hassfest/zeroconf.py @@ -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(