docker-ci.yaml

name: Docker CI

on:
  push:
    branches: [ "master" ]
  pull_request:
    branches: [ "master" ]
  merge_group:


jobs:
  test-build-and-container:
    runs-on: ubuntu-latest
    strategy:
      fail-fast: false
      matrix:
        python-version: ["3.8", "3.9", "3.10"]
    steps:
      - uses: actions/checkout@v3
        with:
          submodules: recursive

      - name: Test build_image.sh script with custom tagging and gpu flag
        working-directory: docker
        run: ./test_build_image_tagging.sh ${{ matrix.python-version }}

      - name: Determine the branch name
        id: branch-name
        run: |
          if [[ "${GITHUB_REF}" == refs/heads/* ]]; then
            GITHUB_BRANCH=${GITHUB_REF#refs/heads/}
          elif [[ "${GITHUB_REF}" == refs/pull/*/merge ]]; then
            GITHUB_BRANCH=${GITHUB_HEAD_REF}
          fi
          echo "GITHUB_BRANCH=${GITHUB_BRANCH}" >> $GITHUB_OUTPUT

      - name: Build Image for container test
        id: image_build
        working-directory: docker
        run: |
          IMAGE_TAG=test-image-${{ matrix.python-version }}
          REPO_URL="${{ github.event.pull_request.head.repo.clone_url }}"
          if [[ -z "${REPO_URL}" ]]; then
            REPO_URL="https://github.com/pytorch/serve.git"
          fi
          ./build_image.sh -py "${{ matrix.python-version }}" -t "${IMAGE_TAG}" -b "${{ steps.branch-name.outputs.GITHUB_BRANCH }}" -repo ${REPO_URL} -s
          echo "IMAGE_TAG=${IMAGE_TAG}" >> $GITHUB_OUTPUT

      - name: Container Healthcheck
        working-directory: docker
        run: ./test_container_health.sh ${{ steps.image_build.outputs.IMAGE_TAG }}

      - name: Check Python version in container
        working-directory: docker
        run: ./test_container_python_version.sh ${{ steps.image_build.outputs.IMAGE_TAG }} ${{ matrix.python-version }}

      - name: Test model running in container with sample image data
        working-directory: docker
        run: |
          ./test_container_model_prediction.sh ${{ steps.image_build.outputs.IMAGE_TAG }}