tests.yml

---

name: Tests

# yamllint disable-line rule:truthy
on:
  push:
    branches:
      - main
      - dev
  pull_request:
    branches:
      - main
      - dev

jobs:


  pytest:
    runs-on: ubuntu-22.04
    if: |
      github.event_name == 'push' ||
      github.event.pull_request.draft == false
    strategy:
      fail-fast: false
      matrix:
        tokamak:
          - C-MOD
          - DIII-D
          - HBT-EP
          - MAST
    steps:

      - name: Checkout
        uses: actions/checkout@v6

      - name: Add MDSplus sources
        if: ${{ matrix.tokamak == 'C-MOD'
             || matrix.tokamak == 'DIII-D'
             || matrix.tokamak == 'HBT-EP' }}
        env:
          GPG: http://www.mdsplus.org/dist/mdsplus.gpg.key
          KEY: /usr/share/keyrings/mdsplus.asc
          REPO: http://www.mdsplus.org/dist/Ubuntu22/repo
          APT: /etc/apt/sources.list.d/mdsplus.list
        run: |
          curl -fsSL "$GPG" \
          | sudo tee "$KEY"
          echo "deb [arch=amd64 signed-by=$KEY] $REPO MDSplus stable" \
          | sudo tee "$APT"

      - name: Update sources
        if: ${{ matrix.tokamak == 'C-MOD'
             || matrix.tokamak == 'DIII-D'
             || matrix.tokamak == 'HBT-EP' }}
        run: sudo apt-get update

      - name: Install prerequisites
        if: ${{ matrix.tokamak == 'C-MOD'
             || matrix.tokamak == 'DIII-D'
             || matrix.tokamak == 'HBT-EP' }}
        run: |
          sudo apt-get install msodbcsql17 # pre-installed
          sudo apt-get install tdsodbc unixodbc # freetds
          sudo apt-get install mdsplus-python # mdsplus

      - name: Configure MDSplus
        if: ${{ matrix.tokamak == 'C-MOD'
             || matrix.tokamak == 'DIII-D'
             || matrix.tokamak == 'HBT-EP' }}
        run: |
          {
             echo PYTHONPATH=/usr/local/mdsplus/python
             echo LD_LIBRARY_PATH=/usr/local/mdsplus/lib
          } \
          | tee -a "$GITHUB_ENV"

      - name: Setup SSH key
        if: ${{ matrix.tokamak == 'C-MOD'
             || matrix.tokamak == 'DIII-D'
             || matrix.tokamak == 'HBT-EP' }}
        run: |
          mkdir -p ~/.ssh
          echo "${{ secrets.SSH_TUNNEL }}" \
          | tee ~/.ssh/id_rsa \
          | sha256sum
          chmod 600 ~/.ssh/id_rsa

      - name: Setup C-MOD
        if: ${{ matrix.tokamak == 'C-MOD' }}
        run: |
          ssh "${{ secrets.CMOD_USER }}@${{ secrets.CMOD_HOST }}" \
            -p "${{ secrets.CMOD_PORT }}" \
            -L 8000:mfedata01:8000 \
            -L 1433:alcdb2:1433 \
            -o StrictHostKeyChecking=no \
            -N -v 2> ssh.log &
          echo "$!" \
          | tee ssh.pid
          echo "127.0.0.1 mfedata01 alcdb2" \
          | sudo tee -a /etc/hosts
          echo "${{ secrets.CMOD_LOGIN }}" \
          | tee ~/logbook.sybase_login \
          | sha256sum
          echo DISPY_TOKAMAK=cmod \
          | tee -a "$GITHUB_ENV"

      - name: Setup DIII-D
        if: ${{ matrix.tokamak == 'DIII-D' }}
        env:
          TDS: /usr/lib/x86_64-linux-gnu/odbc/libtdsodbc.so
        run: |
          ssh "${{ secrets.DIIID_USER }}@${{ secrets.DIIID_HOST }}" \
            -p "${{ secrets.DIIID_PORT }}" \
            -L 8000:atlas:8000 \
            -L 8001:d3drdb:8001 \
            -o StrictHostKeyChecking=no \
            -N -v 2> ssh.log &
          echo "$!" \
          | tee ssh.pid
          echo "127.0.0.1 atlas d3drdb" \
          | sudo tee -a /etc/hosts
          echo "${{ secrets.DIIID_LOGIN }}" \
          | tee ~/D3DRDB.sybase_login \
          | sha256sum
          echo -e "[FreeTDS]\nDescription = FreeTDS\nDriver = $TDS" \
          | sudo tee -a /etc/odbcinst.ini
          echo DISPY_TOKAMAK=d3d \
          | tee -a "$GITHUB_ENV"

      - name: Setup HBT-EP
        if: ${{ matrix.tokamak == 'HBT-EP' }}
        run: |
          ssh "${{ secrets.HBTEP_USER }}@${{ secrets.HBTEP_HOST }}" \
            -p "${{ secrets.HBTEP_PORT }}" \
            -L 8003:maxwell.ap.columbia.edu:8003 \
            -o StrictHostKeyChecking=no \
            -N -v 2> ssh.log &
          echo "$!" \
          | tee ssh.pid
          echo "127.0.0.1 maxwell.ap.columbia.edu" \
          | sudo tee -a /etc/hosts
          echo DISPY_TOKAMAK=hbtep \
          | tee -a "$GITHUB_ENV"

      - name: Install poetry
        run: pipx install poetry

      - name: Setup Python
        uses: actions/setup-python@v6
        with:
          python-version: '3.12'
          cache: 'poetry'

      - name: Install requirements
        run: make install-all

      - name: Check tunnel
        if: ${{ matrix.tokamak == 'C-MOD'
             || matrix.tokamak == 'DIII-D'
             || matrix.tokamak == 'HBT-EP' }}
        run: |
          xargs -a ssh.pid ps -p || true
          if ! grep 'Authenticated.*publickey' ssh.log
          then
             curl -s ipinfo.io && echo
             cat -n ssh.log
             false
          fi

      - name: Test SQL
        if: ${{ matrix.tokamak == 'C-MOD'
             || matrix.tokamak == 'DIII-D' }}
        run: poetry run python examples/sql.py

      - name: Test MDSplus
        if: ${{ matrix.tokamak == 'C-MOD'
             || matrix.tokamak == 'DIII-D'
             || matrix.tokamak == 'HBT-EP' }}
        run: poetry run python examples/mdsplus.py

      - name: Test EFIT
        if: ${{ matrix.tokamak == 'C-MOD'
             || matrix.tokamak == 'DIII-D'
             || matrix.tokamak == 'MAST' }}
        run: poetry run python examples/efit.py

      - name: Run all tests
        run: make test

      - name: Close tunnel
        if: ${{ matrix.tokamak == 'C-MOD'
             || matrix.tokamak == 'DIII-D'
             || matrix.tokamak == 'HBT-EP' }}
        run: xargs -a ssh.pid kill -TERM