refactor: call extend rest interface in worker (#6108)

jina/serve/runtimes/worker/request_handling.py

@@ -178,7 +178,6 @@ def call_handle(request):
             ]
 
             return self.process_single_data(request, None, is_generator=is_generator)
-
         app = get_fastapi_app(
             request_models_map=request_models_map, caller=call_handle, **kwargs
         )
@@ -187,7 +186,9 @@ def call_handle(request):
         async def _shutdown():
             await self.close()
 
-        return app
+        from jina.helper import extend_rest_interface
+
+        return extend_rest_interface(app)
 
     def _http_fastapi_sagemaker_app(self, **kwargs):
         from jina.serve.runtimes.worker.http_sagemaker_app import get_fastapi_app

tests/conftest.py

@@ -60,7 +60,7 @@ def test_log_level(monkeypatch):
 
 
 @pytest.fixture(autouse=True)
-def test_grpc_fork_support_false(monkeypatch):
+def test_grpc_fork_support_true(monkeypatch):
     monkeypatch.setenv('GRPC_ENABLE_FORK_SUPPORT', 'true')
 
 
@@ -94,4 +94,4 @@ def set_test_pip_version() -> None:
     os.environ['JINA_GATEWAY_IMAGE'] = 'jinaai/jina:test-pip'
     yield
     if 'JINA_GATEWAY_IMAGE' in os.environ: # maybe another fixture has already removed
-        del os.environ['JINA_GATEWAY_IMAGE']
+        del os.environ['JINA_GATEWAY_IMAGE']