(airflow) branch main updated: Fix removal of `__str__` method from Datafusion Format enums (#62830)

[go...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

gopidesu pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/main by this push:
     new c4e2d0054bd Fix removal of `__str__` method from Datafusion Format enums (#62830)
c4e2d0054bd is described below

commit c4e2d0054bd0649d9db9af41837f46cb0dcbc31d
Author: Jarek Potiuk <ja...@potiuk.com>
AuthorDate: Wed Mar 4 01:47:59 2026 +0100

    Fix removal of `__str__` method from Datafusion Format enums (#62830)
    
    The enum resolves to full enum name - not lowercase name in
    datafusion 52.0.0 - so the registered mock method had wrong name.
---
 .../common/sql/tests/unit/common/sql/datafusion/test_format_handlers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/providers/common/sql/tests/unit/common/sql/datafusion/test_format_handlers.py b/providers/common/sql/tests/unit/common/sql/datafusion/test_format_handlers.py
index 7c8c89eb759..c6221c62bf4 100644
--- a/providers/common/sql/tests/unit/common/sql/datafusion/test_format_handlers.py
+++ b/providers/common/sql/tests/unit/common/sql/datafusion/test_format_handlers.py
@@ -79,7 +79,7 @@ class TestFormatHandlers:
             conn_id="conn_id",
         )
         handler = handler_class(datasource_config)
-        register_method = getattr(session_context_mock, f"register_{format}")
+        register_method = getattr(session_context_mock, f"register_{format.name.lower()}")
         register_method.side_effect = Exception("Error")
         with pytest.raises(
             FileFormatRegistrationException, match=f"Failed to register {format} data source:"