diff --git a/airbyte-ci/connectors/connector_ops/connector_ops/utils.py b/airbyte-ci/connectors/connector_ops/connector_ops/utils.py index d22ab2bba05c..b62bc9eef5b1 100644 --- a/airbyte-ci/connectors/connector_ops/connector_ops/utils.py +++ b/airbyte-ci/connectors/connector_ops/connector_ops/utils.py @@ -51,6 +51,10 @@ def download_catalog(catalog_url): "ql": 400, } +ALLOWED_HOST_THRESHOLD = { + "ql": 300, +} + class ConnectorInvalidNameError(Exception): pass @@ -346,6 +350,15 @@ def requires_high_test_strictness_level(self) -> bool: """ return self.ab_internal_ql >= IMPORTANT_CONNECTOR_THRESHOLDS["ql"] + @property + def requires_allowed_hosts_check(self) -> bool: + """Check if a connector requires allowed hosts. + + Returns: + bool: True if the connector requires allowed hosts, False otherwise. + """ + return self.ab_internal_ql >= ALLOWED_HOST_THRESHOLD["ql"] + @property def allowed_hosts(self) -> Optional[List[str]]: return self.metadata.get("allowedHosts") if self.metadata else None