fix unit tests for 3.14

Author: hallvictoria

workers/tests/unittest_proxy/test_dispatcher.py

@@ -20,6 +20,9 @@
 )
 
 
+_real_import = builtins.__import__
+
+
 class TestDispatcher(unittest.TestCase):
 
     @patch("proxy_worker.dispatcher.queue.Queue")
@@ -304,7 +307,6 @@ def get_threadpool_executor():
        return_value=False)
 @patch("proxy_worker.dispatcher.logger")
 @patch("proxy_worker.dispatcher.os.path.exists", side_effect=lambda p: True)
-@patch("builtins.__import__")
 @patch("proxy_worker.dispatcher.protos.StreamingMessage",
        return_value="mocked_init_response")
 @patch("proxy_worker.dispatcher.check_python_eol")
@@ -316,7 +318,7 @@ async def test_worker_init_starts_threadpool(mock_eol, mock_streaming,
     def fake_import(name, *a, **k):
         if name == "azure_functions_runtime":
             return runtime_module
-        return builtins.__import__(name, *a, **k)
+        return _real_import(name, *a, **k)
 
     mock_import.side_effect = fake_import
     dispatcher = Dispatcher(asyncio.get_event_loop(), "localhost", 7071,
@@ -332,7 +334,6 @@ def fake_import(name, *a, **k):
 @patch("proxy_worker.dispatcher.DependencyManager.prioritize_customer_dependencies")
 @patch("proxy_worker.dispatcher.logger")
 @patch("proxy_worker.dispatcher.os.path.exists", side_effect=lambda p: True)
-@patch("builtins.__import__")
 @patch("proxy_worker.dispatcher.protos.StreamingMessage",
        return_value="mocked_reload_response")
 @patch("proxy_worker.dispatcher.check_python_eol")
@@ -344,7 +345,7 @@ async def test_env_reload_starts_threadpool(mock_eol, mock_streaming,
     def fake_import(name, *a, **k):
         if name == "azure_functions_runtime":
             return runtime_module
-        return builtins.__import__(name, *a, **k)
+        return _real_import(name, *a, **k)
 
     mock_import.side_effect = fake_import
     dispatcher = Dispatcher(asyncio.get_event_loop(), "localhost", 7071,
@@ -366,7 +367,6 @@ def fake_import(name, *a, **k):
        return_value=False)
 @patch("proxy_worker.dispatcher.logger")
 @patch("proxy_worker.dispatcher.os.path.exists", side_effect=lambda p: True)
-@patch("builtins.__import__")
 @patch("proxy_worker.dispatcher.protos.StreamingMessage",
        return_value="mocked_init_response")
 @patch("proxy_worker.dispatcher.check_python_eol")
@@ -379,7 +379,7 @@ async def test_worker_init_missing_threadpool_apis(mock_eol,
     def fake_import(name, *a, **k):
         if name == "azure_functions_runtime":
             return runtime_module
-        return builtins.__import__(name, *a, **k)
+        return _real_import(name, *a, **k)
 
     mock_import.side_effect = fake_import
     dispatcher = Dispatcher(asyncio.get_event_loop(), "localhost", 7071,