fix: pass extra arguments to gateway runtime with container pod

jina/orchestrate/pods/container.py

@@ -2,10 +2,10 @@
 import asyncio
 import copy
 import multiprocessing
-import threading
 import os
 import re
 import signal
+import threading
 import time
 from typing import TYPE_CHECKING, Dict, Optional, Union
 
@@ -21,6 +21,7 @@
     get_docker_network,
     get_gpu_device_requests,
 )
+from jina.parsers import set_gateway_parser
 from jina.serve.runtimes.asyncio import AsyncNewLoopRuntime
 from jina.serve.runtimes.gateway import GatewayRuntime
 
@@ -74,9 +75,15 @@ def _docker_run(
 
     args.native = True
 
+    parser = (
+        set_gateway_parser()
+        if args.pod_role == PodRoleType.GATEWAY
+        else set_pod_parser()
+    )
+
     non_defaults = ArgNamespace.get_non_defaults_args(
         args,
-        set_pod_parser(),
+        parser,
         taboo={
             'uses',
             'entrypoint',

tests/unit/orchestrate/pods/container/custom-gateway/dummy_gateway.py

@@ -42,6 +42,13 @@ def _get_response():
                 'arg3': self.arg3,
             }
 
+        @app.get(path='/runtime_info')
+        def _get_info():
+            return {
+                'ports': self.ports,
+                'protocols': self.protocols,
+            }
+
         @app.get(
             path='/stream',
             response_model=ProcessedResponseModel,

tests/unit/orchestrate/pods/container/test_container_pod.py

@@ -2,9 +2,11 @@
 import time
 
 import pytest
+import requests
 
 from jina import Flow
 from jina.constants import __cache_path__
+from jina.enums import GatewayProtocolType
 from jina.excepts import RuntimeFailToStart
 from jina.helper import random_port
 from jina.orchestrate.pods.container import ContainerPod
@@ -273,6 +275,11 @@ def test_container_pod_custom_gateway(dummy_custom_gateway_docker_image_built):
             port, {'arg1': 'hello', 'arg2': 'world', 'arg3': 'default-arg3'}
         )
 
+        # validate protocol inside the gateway
+        resp = requests.get(f'http://127.0.0.1:{port}/runtime_info').json()
+        assert resp['protocols'] == [GatewayProtocolType.HTTP]
+        assert resp['ports'] == [int(port)]
+
         time.sleep(
             2
         )  # to avoid desync between the start and close process which could lead to container never get terminated