Merge pull request #154 from runpod/catch-e2e-results

Update CI-e2e.yml

Author: justinmerrell

.github/workflows/CI-e2e.yml

@@ -63,9 +63,22 @@ jobs:
 
     steps:
       - uses: actions/checkout@v4
+
       - name: Run Tests
+        id: run-tests
         uses: direlines/runpod-test-runner@v1.7
         with:
           image-tag: ${{ vars.DOCKERHUB_REPO }}/${{ vars.DOCKERHUB_IMG }}:${{ needs.e2e-build.outputs.docker_tag }}
           runpod-api-key: ${{ secrets.RUNPOD_API_KEY }}
           request-timeout: 1200
+
+      - name: Verify Tests
+        env:
+          TOTAL_TESTS: ${{ steps.run-tests.outputs.total-tests }}
+          SUCCESSFUL_TESTS: ${{ steps.run-tests.outputs.succeeded }}
+        run: |
+          echo "Total tests: $TOTAL_TESTS"
+          echo "Successful tests: $SUCCESSFUL_TESTS"
+          if [ "$TOTAL_TESTS" != "$SUCCESSFUL_TESTS" ]; then
+              exit 1
+          fi

runpod/serverless/modules/rp_job.py

@@ -151,7 +151,7 @@ async def run_job(handler: Callable, job: Dict[str, Any]) -> Dict[str, Any]:
 
         log.error(f'{job["id"]} | Captured Handler Exception')
         log.error(json.dumps(error_info, indent=4))
-        run_result = {"error": error_info}
+        run_result = {"error": json.dumps(error_info)}
 
     finally:
         log.debug(f'{job["id"]} | run_job return: {run_result}')

tests/test_serverless/test_modules/test_job.py

@@ -201,6 +201,17 @@ async def test_job_with_errors(self):
 
         assert job_result == {"error": "test"}
 
+    async def test_job_with_raised_exception(self):
+        '''
+        Tests the run_job function with a raised exception
+        '''
+        mock_handler = Mock()
+        mock_handler.side_effect = Exception
+
+        job_result = await rp_job.run_job(mock_handler, self.sample_job)
+
+        assert "error" in job_result
+
     async def test_job_with_refresh_worker(self):
         '''
         Tests the run_job function with refresh_worker