From 6ddf0a96634a5420cca137da74404769aa38d198 Mon Sep 17 00:00:00 2001
From: Marko Atanasievski <matanasievski@polygon.technology>
Date: Fri, 21 Jun 2024 15:56:34 +0200
Subject: [PATCH] fix: environment and script

---
 .github/workflows/jerigon.yml | 6 +++---
 zero_bin/tools/prove_rpc.sh   | 8 ++++++--
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/jerigon.yml b/.github/workflows/jerigon.yml
index 743257782..e72d2548a 100644
--- a/.github/workflows/jerigon.yml
+++ b/.github/workflows/jerigon.yml
@@ -18,8 +18,8 @@ env:
 jobs:
   test_jerigon_input_proving:
     name: Test proof generation with jerigon input
-    runs-on: ubuntu-latest
-    timeout-minutes: 60
+    runs-on: zero-ci
+    timeout-minutes: 40
     # if: "! contains(toJSON(github.event.commits.*.message), '[skip-ci]')"
     steps:
       - name: Checkout sources
@@ -65,7 +65,7 @@ jobs:
       - name: Run prove blocks in test_only mode
         run: |
           cd zero_bin/tools
-          ./prove_rpc.sh 0x2 0x7 http://localhost:8546 jerigon true 0 0 test_only
+          OUTPUT_TO_TERMINAL=true ./prove_rpc.sh 0x2 0x7 http://localhost:8546 jerigon true 0 0 test_only
           echo "Proving blocks in test_only mode done"
 
 
diff --git a/zero_bin/tools/prove_rpc.sh b/zero_bin/tools/prove_rpc.sh
index fe8dfc429..1d627fd10 100755
--- a/zero_bin/tools/prove_rpc.sh
+++ b/zero_bin/tools/prove_rpc.sh
@@ -94,6 +94,9 @@ if [[ $8 == "test_only" ]]; then
     command='cargo r --release --features test_only --bin leader -- --runtime in-memory --load-strategy on-demand "$NODE_RPC_TYPE" --rpc-url "$NODE_RPC_URL" --block-interval $BLOCK_INTERVAL --proof-output-dir $PROOF_OUTPUT_DIR $PREV_PROOF_EXTRA_ARG --backoff "$BACKOFF" --max-retries "$RETRIES" '
     if [ "$OUTPUT_TO_TERMINAL" = true ]; then
         eval $command
+        RESULT=$?
+        echo -e "Proof witness generation finished with result: $RESULT"
+        exit $RESULT
     else
         eval $command > $OUT_LOG_PATH 2>&1
         if grep -q 'All proof witnesses have been generated successfully.' $OUT_LOG_PATH; then
@@ -114,6 +117,9 @@ else
     command='cargo r --release --bin leader -- --runtime in-memory --load-strategy on-demand "$NODE_RPC_TYPE" --rpc-url "$3" --block-interval $BLOCK_INTERVAL --proof-output-dir $PROOF_OUTPUT_DIR $PREV_PROOF_EXTRA_ARG --backoff "$BACKOFF" --max-retries "$RETRIES"'
     if [ "$OUTPUT_TO_TERMINAL" = true ]; then
         eval $command
+        RESULT=$?
+        echo -e "Proof generation finished with result: $RESULT"
+        exit $RESULT
     else
         eval $command > $OUT_LOG_PATH 2>&1
         retVal=$?
@@ -132,8 +138,6 @@ else
 fi
 
 
-
-
 # If we're running the verification, we'll do it here.
 if [ "$RUN_VERIFICATION" = true ]; then
   echo "Running the verification"