Have integration points for AIP-69 in Internal API

apache · Jul 20, 2024 · e8291d3 · e8291d3
1 parent a7527e5
commit e8291d3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/api_internal/endpoints/test_rpc_api_endpoint.py b/tests/api_internal/endpoints/test_rpc_api_endpoint.py
@@ -75,12 +75,12 @@ def setup_attrs(self, minimal_app_for_internal_api: Flask) -> Generator:
         mock_test_method.reset_mock()
         mock_test_method.side_effect = None
         with mock.patch(
-            "airflow.api_internal.endpoints.rpc_api_endpoint._initialize_map"
-        ) as mock_initialize_map:
-            mock_initialize_map.return_value = {
+            "airflow.api_internal.endpoints.rpc_api_endpoint.initialize_method_map"
+        ) as mock_initialize_method_map:
+            mock_initialize_method_map.return_value = {
                 TEST_METHOD_NAME: mock_test_method,
             }
-            yield mock_initialize_map
+            yield mock_initialize_method_map
 
     @pytest.mark.parametrize(
         "input_params, method_result, result_cmp_func, method_params",