feat: e2e test streaming

justinmerrell · justinmerrell · commit 495b6328a9db · 2023-08-14T15:49:18.000-04:00
diff --git a/.github/tests.json b/.github/tests.json
@@ -21,5 +21,41 @@
             "mock_return": "Delay test successful returned after waiting 5 minutes.",
             "mock_delay": 120
         }
+    },
+    {
+        "hardwareConfig": {
+            "endpointConfig": {
+                "gpuIds": "AMPERE_16",
+                "name": "runpod-python E2E Test - Generator Handler"
+            },
+            "templateConfig": {
+                "dockerArgs": "python3 -u /handler.py --generator --return_aggregate_stream"
+            }
+        },
+        "input": {
+            "mock_return": [
+                "value1",
+                "value2",
+                "value3"
+            ]
+        }
+    },
+    {
+        "hardwareConfig": {
+            "endpointConfig": {
+                "gpuIds": "AMPERE_16",
+                "name": "runpod-python E2E Test - Async Generator Handler"
+            },
+            "templateConfig": {
+                "dockerArgs": "python3 -u /handler.py --async_generator --return_aggregate_stream"
+            }
+        },
+        "input": {
+            "mock_return": [
+                "value1",
+                "value2",
+                "value3"
+            ]
+        }
     }
 ]
diff --git a/.github/workflows/CI-e2e.yml b/.github/workflows/CI-e2e.yml
@@ -25,7 +25,6 @@ jobs:
       - name: Clone and patch mock-worker
         run: |
           git clone https://github.com/runpod-workers/mock-worker
-          echo "Installing runpod-python from ${{ github.event.pull_request.head.sha }}"
           echo "git+https://github.com/runpod/runpod-python.git@${{ github.event.pull_request.head.sha }}" > mock-worker/builder/requirements.txt
 
       - name: Set up QEMU
