ci: improve astradb setup (#329)

Author: nicoloboschi

.github/actions/setup-astra-db/action.yml

@@ -34,21 +34,33 @@ runs:
       env:
         TERM: linux
       run: |
-        set -x
+        set -e
         (curl -Ls "https://dtsx.io/get-astra-cli" | bash) || true
-        /home/runner/.astra/cli/astra db create -v "${{ inputs.db-name }}" \
+        (/home/runner/.astra/cli/astra db create -v "${{ inputs.db-name }}" \
           --env "${{ inputs.env }}" \
           -k default_keyspace \
           --token "${{ inputs.astra-token }}" \
           --vector \
           --region "${{ inputs.region }}" \
           --cloud "${{ inputs.cloud }}" \
-          --timeout 600
+          --timeout 1200) || "Failed to create database, check if creation is still in progress"
 
-        describe_out=$(/home/runner/.astra/cli/astra db describe "${{ inputs.db-name }}" \
-         --token "${{ inputs.astra-token }}" --env "${{ inputs.env }}" -o json)
+        while true; do
+          describe_out=$(/home/runner/.astra/cli/astra db describe "${{ inputs.db-name }}" \
+              --token "${{ inputs.astra-token }}" --env "${{ inputs.env }}" -o json)
+          status=$(echo "$describe_out" | jq -r '.data.status')
+          database_id=$(echo "$describe_out" | jq -r '.data.id')
+          if [ "$status" == "ACTIVE" ]; then
+            break
+          fi
+          if [ "$status" == "INITIALIZING" ]; then
+            echo "Database is not active, status: $status"
+            sleep 10
+          fi
+          echo "Database creation failed, status: $status"
+          exit 1
+        done
 
-        database_id=$(echo "$describe_out" | jq -r '.data.id')
         echo "Found database id: $database_id"
         domain="astra"
         if [ "${{ inputs.env }}" == "DEV" ]; then

.github/workflows/_run_e2e_tests.yml

@@ -70,6 +70,7 @@ jobs:
       - name: Setup AstraDB
         uses: ./.github/actions/setup-astra-db
         id: astra-db
+        timeout-minutes: 10
         if: ${{ inputs.astradb }}
         with:
           astra-token: ${{ secrets[inputs.astradb-token-secret-name] }}