test: fix test

jina-ai · Dec 16, 2022 · 774ba86 · 774ba86
1 parent 1517368
commit 774ba86
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/k8s/test_k8s.py b/tests/k8s/test_k8s.py
@@ -1282,7 +1282,10 @@ async def test_flow_multiple_protocols_gateway(
                 import requests
 
                 resp = requests.get(f'http://localhost:{http_port}').json()
-                assert resp['protocol'] == 'http'
+                assert resp in [
+                    {'protocol': 'http'},
+                    {},
+                ]  # first one for the custom gateway implementation, the second is for the built-in multiprotocol gateway
 
         # test portforwarding the gateway pod and service using grpc
         forward_args = [