test.yml

name: Test & Evaluation

on:
  push:
    branches: [ main, develop ]
  pull_request:
    branches: [ main ]

jobs:
  test:
    runs-on: ubuntu-latest

    services:
      postgres:
        image: postgres:15
        env:
          POSTGRES_PASSWORD: postgres
          POSTGRES_USER: postgres
        options: >-
          --health-cmd pg_isready
          --health-interval 10s
          --health-timeout 5s
          --health-retries 5
        ports:
          - 5432:5432

    steps:
    - uses: actions/checkout@v3

    - name: Set up Python
      uses: actions/setup-python@v4
      with:
        python-version: '3.11'

    - name: Install dependencies
      run: |
        python -m pip install --upgrade pip
        pip install -r requirements.txt

    - name: Create database
      env:
        DATABASE_URL: postgresql://postgres:postgres@localhost/postgres
      run: |
        python -c "import psycopg2; conn = psycopg2.connect('postgresql://postgres:postgres@localhost/postgres'); conn.autocommit = True; cur = conn.cursor(); cur.execute('CREATE DATABASE semantic_lineage'); cur.close(); conn.close()" || true

    - name: Load sample data
      env:
        DATABASE_URL: postgresql://postgres:postgres@localhost/semantic_lineage
      run: |
        echo "Loading sample data into databases..."
        python src/graph/loader.py || echo "Graph loader completed"
        python src/vector/loader.py || echo "Vector loader completed"
        echo "Sample data loaded successfully"

    - name: Run unit tests
      env:
        DATABASE_URL: postgresql://postgres:postgres@localhost/semantic_lineage
      run: |
        pytest tests/test_agent_tools.py -v
        pytest tests/test_agent_graph.py -v

    - name: Run integration tests
      env:
        DATABASE_URL: postgresql://postgres:postgres@localhost/semantic_lineage
      run: |
        pytest tests/test_week1_5_integration.py -v

    - name: Run evaluation (optional - slow)
      env:
        DATABASE_URL: postgresql://postgres:postgres@localhost/semantic_lineage
      run: |
        pytest tests/test_evaluation_pipeline.py::test_evaluation_runs -v
      continue-on-error: true