@@ -230,7 +230,7 @@ def test_variable_env_var_do_not_access_team_specific(self):
230230 assert Variable .get_variable_from_secrets (key = "_team___myvar" ) is None
231231
232232
233- class _LegacyGetConnectionBackend (BaseSecretsBackend ):
233+ class _TeamUnawareGetConnectionBackend (BaseSecretsBackend ):
234234 """Backend overriding ``get_connection`` with the pre-3.2 ``(self, conn_id)`` signature (e.g. Vault)."""
235235
236236 def __init__ (self , conns : dict [str , Connection ]):
@@ -240,7 +240,7 @@ def get_connection(self, conn_id):
240240 return self ._conns .get (conn_id )
241241
242242
243- class _LegacyGetVariableBackend (BaseSecretsBackend ):
243+ class _TeamUnawareGetVariableBackend (BaseSecretsBackend ):
244244 """Backend overriding ``get_variable`` with the pre-3.2 ``(self, key)`` signature."""
245245
246246 def __init__ (self , variables : dict [str , str ]):
@@ -251,7 +251,7 @@ def get_variable(self, key):
251251
252252
253253@skip_if_force_lowest_dependencies_marker
254- class TestLegacyBackendSignatureCompat :
254+ class TestTeamUnawareBackendSignatureCompat :
255255 """Backends whose overrides predate the ``team_name`` keyword must keep working (issue #1333)."""
256256
257257 def setup_method (self ) -> None :
@@ -260,21 +260,21 @@ def setup_method(self) -> None:
260260 @pytest .mark .parametrize ("team_name" , [None , "team_a" ])
261261 @conf_vars ({("core" , "multi_team" ): "True" })
262262 @mock .patch .dict ("sys.modules" , {"airflow.sdk.execution_time.task_runner" : None })
263- def test_get_connection_with_legacy_get_connection_override (self , team_name ):
264- backend = _LegacyGetConnectionBackend (
265- {"legacy_conn " : Connection (conn_id = "legacy_conn " , conn_type = "mysql" , host = "h" )}
263+ def test_get_connection_with_team_unaware_override (self , team_name ):
264+ backend = _TeamUnawareGetConnectionBackend (
265+ {"team_unaware_conn " : Connection (conn_id = "team_unaware_conn " , conn_type = "mysql" , host = "h" )}
266266 )
267267 with mock .patch ("airflow.configuration.ensure_secrets_loaded" , return_value = [backend ]):
268- conn = Connection .get_connection_from_secrets ("legacy_conn " , team_name = team_name )
268+ conn = Connection .get_connection_from_secrets ("team_unaware_conn " , team_name = team_name )
269269
270- assert conn .conn_id == "legacy_conn "
270+ assert conn .conn_id == "team_unaware_conn "
271271 assert conn .conn_type == "mysql"
272272
273273 @pytest .mark .parametrize ("team_name" , [None , "team_a" ])
274- def test_get_variable_with_legacy_get_variable_override (self , team_name ):
275- backend = _LegacyGetVariableBackend ({"legacy_var " : "secret_value" })
274+ def test_get_variable_with_team_unaware_override (self , team_name ):
275+ backend = _TeamUnawareGetVariableBackend ({"team_unaware_var " : "secret_value" })
276276 with mock .patch ("airflow.models.variable.ensure_secrets_loaded" , return_value = [backend ]):
277- value = Variable .get_variable_from_secrets ("legacy_var " , team_name = team_name )
277+ value = Variable .get_variable_from_secrets ("team_unaware_var " , team_name = team_name )
278278
279279 assert value == "secret_value"
280280
0 commit comments