From 58e607aa3db03df6f7afc8eb5f0bc9784fe5e151 Mon Sep 17 00:00:00 2001
From: Alberto Miranda <alberto.miranda@seqera.io>
Date: Wed, 26 Feb 2025 18:09:16 +0100
Subject: [PATCH] ci: Improve `fusion-symlink.nf` test

Signed-off-by: Alberto Miranda <alberto.miranda@seqera.io>
---
 tests/checks/fusion-symlink.nf/.checks | 51 +++++++++++++++++++++++++-
 tests/checks/fusion-symlink.nf/.config |  1 +
 2 files changed, 50 insertions(+), 2 deletions(-)

diff --git a/tests/checks/fusion-symlink.nf/.checks b/tests/checks/fusion-symlink.nf/.checks
index a28366d245..c43f1eeadf 100644
--- a/tests/checks/fusion-symlink.nf/.checks
+++ b/tests/checks/fusion-symlink.nf/.checks
@@ -1,5 +1,43 @@
 #!/bin/bash
 
+# Retry a command with exponential backoff. The function returns 0 if the command was successful
+# on its first attempt, 1 if the command failed after all attempts, and 2 if the command was successful
+# after one or more retries.
+function _retry {
+
+  if [[ $# -lt 4 ]]; then
+    echo "Usage: _retry <max_attempts> <initial_delay> <max_delay> <cmd>"
+    return 1
+  fi
+
+  local max_attempts="$1"; shift
+  local initial_delay="$1"; shift
+  local max_delay="$1"; shift
+  local cmd=( "$@" )
+  local attempt_num=1
+  local max_attempts=${max_attempts}
+  local max_delay=${max_delay}
+  local initial_delay=${initial_delay}
+  local exit_code=0
+
+  until "${cmd[@]}"; do
+    exit_code=2
+    if (( attempt_num == max_attempts )); then
+      echo "-- [$attempt_num/$max_attempts] attempt failed! No more attempts left."
+      return 1
+    fi
+    echo "-- [$attempt_num/$max_attempts] attempt failed! Retrying in ${initial_delay}s..."
+    sleep "$initial_delay"
+    (( attempt_num++ ))
+    (( initial_delay *= 2 ))
+    if (( initial_delay > max_delay )); then
+      initial_delay=$max_delay
+    fi
+  done
+  echo "-- [$attempt_num/$max_attempts] attempt succeeded!"
+  return $exit_code
+}
+
 # Skip test if AWS keys are missing
 if [ -z "$AWS_ACCESS_KEY_ID" ]; then 
   echo "Missing AWS credentials -- Skipping test"
@@ -12,7 +50,11 @@ fi
 echo initial run
 $NXF_RUN -c .config
 
-$NXF_CMD fs cp s3://nextflow-ci/work/ci-test/fusion-symlink/data.txt data.txt
+_retry 5 1 16 "$NXF_CMD" fs cp s3://nextflow-ci/work/ci-test/fusion-symlink/data.txt data.txt
+if [ $? -eq 2 ]; then
+  echo "succeeded on retry"
+  false
+fi
 cmp data.txt .expected || false
 
 #
@@ -21,5 +63,10 @@ cmp data.txt .expected || false
 echo resumed run
 $NXF_RUN -c .config -resume
 
-$NXF_CMD fs cp s3://nextflow-ci/work/ci-test/fusion-symlink/data.txt data.txt
+_retry 5 1 16 "$NXF_CMD" fs cp s3://nextflow-ci/work/ci-test/fusion-symlink/data.txt data.txt
+if [ $? -eq 2 ]; then
+  echo "succeeded on retry"
+  false
+
+fi
 cmp data.txt .expected || false
diff --git a/tests/checks/fusion-symlink.nf/.config b/tests/checks/fusion-symlink.nf/.config
index f1a32bb34a..fd9b12f2b0 100644
--- a/tests/checks/fusion-symlink.nf/.config
+++ b/tests/checks/fusion-symlink.nf/.config
@@ -2,3 +2,4 @@ workDir = 's3://nextflow-ci/work'
 fusion.enabled = true
 fusion.exportStorageCredentials = true
 wave.enabled = true
+docker.runOptions = '-e FUSION_TRACING_DESTINATION=objectstore://'