def test_backend_fallback_to_env_var(self, mock_get_uri):
mock_get_uri.return_value = None
backends = ensure_secrets_loaded()
backend_classes = [backend.__class__.__name__ for backend in backends]
self.assertIn('SystemsManagerParameterStoreBackend', backend_classes)
uri = Connection.get_connections_from_secrets(conn_id="test_mysql")
# Assert that SystemsManagerParameterStoreBackend.get_conn_uri was called
mock_get_uri.assert_called_once_with(conn_id='test_mysql')
self.assertEqual('mysql://*****:*****@host:5432/airflow', uri[0].get_uri())
def test_get_connections_first_try(self, mock_env_get, mock_meta_get):
mock_env_get.side_effect = [["something"]] # returns nonempty list
Connection.get_connections_from_secrets("fake_conn_id")
mock_env_get.assert_called_once_with(conn_id="fake_conn_id")
mock_meta_get.not_called()
def test_get_connections_second_try(self, mock_env_get, mock_meta_get):
mock_env_get.side_effect = [[]] # return empty list
Connection.get_connections_from_secrets("fake_conn_id")
mock_meta_get.assert_called_once_with(conn_id="fake_conn_id")
mock_env_get.assert_called_once_with(conn_id="fake_conn_id")
