Improve test infrastructure #3108
Workflow file for this run
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: E2E Subtensor Tests | |
| concurrency: | |
| group: e2e-subtensor-${{ github.ref }} | |
| cancel-in-progress: true | |
| on: | |
| push: | |
| branches: | |
| - '**' | |
| pull_request: | |
| branches: | |
| - '**' | |
| types: [ opened, synchronize, reopened, ready_for_review ] | |
| workflow_dispatch: | |
| inputs: | |
| verbose: | |
| description: "Output more information when triggered manually" | |
| required: false | |
| default: "" | |
| env: | |
| CARGO_TERM_COLOR: always | |
| VERBOSE: ${{ github.event.inputs.verbose }} | |
| # job to run tests in parallel | |
| jobs: | |
| # Looking for e2e tests | |
| find-tests: | |
| runs-on: ubuntu-latest | |
| if: ${{ github.event_name != 'pull_request' || github.event.pull_request.draft == false }} | |
| outputs: | |
| test-files: ${{ steps.get-tests.outputs.test-files }} | |
| steps: | |
| - name: Check-out repository under $GITHUB_WORKSPACE | |
| uses: actions/checkout@v4 | |
| - name: Find test files | |
| id: get-tests | |
| run: | | |
| test_files=$(find tests/e2e_tests -name "test*.py" | jq -R -s -c 'split("\n") | map(select(. != ""))') | |
| # keep it here for future debug | |
| # test_files=$(find tests/e2e_tests -type f -name "test*.py" | grep -E 'test_(hotkeys|staking)\.py$' | jq -R -s -c 'split("\n") | map(select(. != ""))') | |
| echo "Found test files: $test_files" | |
| echo "test-files=$test_files" >> "$GITHUB_OUTPUT" | |
| shell: bash | |
| # Pull docker image | |
| pull-docker-image: | |
| runs-on: ubuntu-latest | |
| outputs: | |
| image-name: ${{ steps.set-output.outputs.image-name }} | |
| steps: | |
| - name: Set Docker image tag based on branch | |
| id: set-output | |
| run: | | |
| ref="${{ github.ref_name }}" | |
| if [[ "$ref" == "master" ]]; then | |
| image="ghcr.io/opentensor/subtensor-localnet:main" | |
| else | |
| image="ghcr.io/opentensor/subtensor-localnet:devnet-ready" | |
| fi | |
| echo "Using image: $image" | |
| echo "image-name=$image" >> "$GITHUB_OUTPUT" | |
| - name: Log in to GitHub Container Registry | |
| run: echo "${{ secrets.GITHUB_TOKEN }}" | docker login ghcr.io -u $GITHUB_ACTOR --password-stdin | |
| - name: Pull Docker Image | |
| run: docker pull ${{ steps.set-output.outputs.image-name }} | |
| - name: Save Docker Image to Cache | |
| run: docker save -o subtensor-localnet.tar ${{ steps.set-output.outputs.image-name }} | |
| - name: Upload Docker Image as Artifact | |
| uses: actions/upload-artifact@v4 | |
| with: | |
| name: subtensor-localnet | |
| path: subtensor-localnet.tar | |
| # Job to run tests in parallel | |
| run-fast-blocks-e2e-test: | |
| name: "FB: ${{ matrix.test-file }} / Python ${{ matrix.python-version }}" | |
| needs: | |
| - find-tests | |
| - pull-docker-image | |
| runs-on: ubuntu-latest | |
| timeout-minutes: 45 | |
| strategy: | |
| fail-fast: false # Allow other matrix jobs to run even if this job fails | |
| max-parallel: 32 # Set the maximum number of parallel jobs (same as we have cores in ubuntu-latest runner) | |
| matrix: | |
| os: | |
| - ubuntu-latest | |
| test-file: ${{ fromJson(needs.find-tests.outputs.test-files) }} | |
| python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"] | |
| steps: | |
| - name: Check-out repository | |
| uses: actions/checkout@v4 | |
| - name: Set up Python ${{ matrix.python-version }} | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| - name: Install uv | |
| uses: astral-sh/setup-uv@v4 | |
| - name: install dependencies | |
| run: uv sync --extra dev --dev | |
| - name: Download Cached Docker Image | |
| uses: actions/download-artifact@v4 | |
| with: | |
| name: subtensor-localnet | |
| - name: Load Docker Image | |
| run: docker load -i subtensor-localnet.tar | |
| - name: Run tests with retry | |
| env: | |
| LOCALNET_IMAGE_NAME: ${{ needs.pull-docker-image.outputs.image-name }} | |
| run: | | |
| set +e | |
| for i in 1 2 3; do | |
| echo "🔁 Attempt $i: Running tests" | |
| uv run pytest ${{ matrix.test-file }} -s | |
| status=$? | |
| if [ $status -eq 0 ]; then | |
| echo "✅ Tests passed on attempt $i" | |
| break | |
| else | |
| echo "❌ Tests failed on attempt $i" | |
| if [ $i -eq 3 ]; then | |
| echo "Tests failed after 3 attempts" | |
| exit 1 | |
| fi | |
| echo "Retrying..." | |
| sleep 5 | |
| fi | |
| done | |