[ci] Simplify tmpnet monitoring action

Incorporate the following into the action to simplify usage: - Configure network shutdown delay to ensure a final metrics scrape - Save tmpnet dir as a github artifact
ava-labs · Feb 21, 2025 · 71d46bf · 71d46bf
1 parent 8f34860
commit 71d46bf
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 18 deletions.
diff --git a/.github/actions/run-monitored-tmpnet-cmd/action.yml b/.github/actions/run-monitored-tmpnet-cmd/action.yml
@@ -7,6 +7,8 @@ inputs:
     required: true
   filter_by_owner:
     default: ''
+  artifact_name:
+    default: 'tmpnet-data'
   prometheus_username:
     required: true
   prometheus_password:
@@ -65,9 +67,15 @@ runs:
       shell: bash
       run: ${{ inputs.run }}
       env:
+        TMPNET_DELAY_NETWORK_SHUTDOWN: true # Ensure shutdown waits for a final metrics scrape
         GH_REPO: ${{ inputs.repository_owner }}/${{ inputs.repository_name }}
         GH_WORKFLOW: ${{ inputs.workflow }}
         GH_RUN_ID: ${{ inputs.run_id }}
         GH_RUN_NUMBER: ${{ inputs.run_number }}
         GH_RUN_ATTEMPT: ${{ inputs.run_attempt }}
         GH_JOB_ID: ${{ inputs.job }}
+    - name: Upload tmpnet network dir
+      uses: ./.github/actions/upload-tmpnet-artifact
+      if: always()
+      with:
+        name: ${{ inputs.artifact_name }}
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -62,17 +62,13 @@ jobs:
       - name: Run e2e tests
         uses: ./.github/actions/run-monitored-tmpnet-cmd
         with:
-          run: E2E_SERIAL=1 ./scripts/tests.e2e.sh --delay-network-shutdown
+          run: E2E_SERIAL=1 ./scripts/tests.e2e.sh
+          artifact_name: e2e-tmpnet-data
           filter_by_owner: avalanchego-e2e
           prometheus_username: ${{ secrets.PROMETHEUS_ID || '' }}
           prometheus_password: ${{ secrets.PROMETHEUS_PASSWORD || '' }}
           loki_username: ${{ secrets.LOKI_ID || '' }}
           loki_password: ${{ secrets.LOKI_PASSWORD || '' }}
-      - name: Upload tmpnet network dir
-        uses: ./.github/actions/upload-tmpnet-artifact
-        if: always()
-        with:
-          name: e2e-tmpnet-data
   e2e_existing_network:
     runs-on: ubuntu-latest
     steps:
@@ -88,16 +84,12 @@ jobs:
       - name: Run e2e tests with existing network
         uses: ./.github/actions/run-monitored-tmpnet-cmd
         with:
-          run: E2E_SERIAL=1 ./scripts/tests.e2e.existing.sh --delay-network-shutdown
+          run: E2E_SERIAL=1 ./scripts/tests.e2e.existing.sh
+          artifact_name: e2e-existing-network-tmpnet-data
           prometheus_username: ${{ secrets.PROMETHEUS_ID || '' }}
           prometheus_password: ${{ secrets.PROMETHEUS_PASSWORD || '' }}
           loki_username: ${{ secrets.LOKI_ID || '' }}
           loki_password: ${{ secrets.LOKI_PASSWORD || '' }}
-      - name: Upload tmpnet network dir
-        uses: ./.github/actions/upload-tmpnet-artifact
-        if: always()
-        with:
-          name: e2e-existing-network-tmpnet-data
   Upgrade:
     runs-on: ubuntu-latest
     steps:
@@ -114,15 +106,11 @@ jobs:
         uses: ./.github/actions/run-monitored-tmpnet-cmd
         with:
           run: ./scripts/tests.upgrade.sh
+          artifact_name: upgrade-tmpnet-data
           prometheus_username: ${{ secrets.PROMETHEUS_ID || '' }}
           prometheus_password: ${{ secrets.PROMETHEUS_PASSWORD || '' }}
           loki_username: ${{ secrets.LOKI_ID || '' }}
           loki_password: ${{ secrets.LOKI_PASSWORD || '' }}
-      - name: Upload tmpnet network dir
-        uses: ./.github/actions/upload-tmpnet-artifact
-        if: always()
-        with:
-          name: upgrade-tmpnet-data
   Lint:
     runs-on: ubuntu-latest
     steps:

diff --git a/tests/fixture/e2e/flags.go b/tests/fixture/e2e/flags.go
@@ -10,6 +10,8 @@ import (
 	"path/filepath"
 	"time"
 
+	"github.com/spf13/cast"
+
 	"github.com/ava-labs/avalanchego/tests/fixture/tmpnet"
 )
 
@@ -143,7 +145,7 @@ func RegisterFlags() *FlagVars {
 	flag.BoolVar(
 		&vars.delayNetworkShutdown,
 		"delay-network-shutdown",
-		false,
+		cast.ToBool(GetEnvWithDefault(tmpnet.DelayNetworkShutdownEnvName, "false")),
 		"[optional] whether to delay network shutdown to allow a final metrics scrape.",
 	)
 	flag.BoolVar(

diff --git a/tests/fixture/tmpnet/node_process.go b/tests/fixture/tmpnet/node_process.go
@@ -37,6 +37,11 @@ const (
 var (
 	AvalancheGoPluginDirEnvName = config.EnvVarName(config.EnvPrefix, config.PluginDirKey)
 
+	// Only necessary for process-based nodes since nodes running in
+	// kube will always be assumed to be monitored and delay their
+	// shutdown accordingly.
+	DelayNetworkShutdownEnvName = "TMPNET_DELAY_NETWORK_SHUTDOWN"
+
 	errNodeAlreadyRunning = errors.New("failed to start node: node is already running")
 	errNotRunning         = errors.New("node is not running")
 )