noxfile.py

"""Nox sessions for Answer Architect project."""

import nox

PYTHON_VERSIONS = ["3.11"]
LOCATIONS = ["src", "tests", "tools", "scripts", "infra"]


@nox.session(python=PYTHON_VERSIONS)
def lint(session):
    """Run code linting with ruff."""
    session.install("ruff", "black", "mypy")
    session.run("ruff", "check", *LOCATIONS)
    session.run("black", "--check", "--diff", *LOCATIONS)
    session.run("mypy", *LOCATIONS)


@nox.session(python=PYTHON_VERSIONS)
def format_code(session):
    """Format code with black and ruff."""
    session.install("ruff", "black")
    session.run("black", *LOCATIONS)
    session.run("ruff", "check", "--fix", *LOCATIONS)


@nox.session(python=PYTHON_VERSIONS)
def tests(session):
    """Run unit tests with pytest."""
    session.install("poetry")
    session.run("poetry", "install", "--with", "dev")
    session.run("poetry", "run", "pytest", "-v", "--cov=src", "--cov-report=term-missing")


@nox.session(python=PYTHON_VERSIONS)
def e2e_latency(session):
    """Run end-to-end latency tests."""
    session.install("poetry")
    session.run("poetry", "install", "--with", "dev")
    # Mock E2E test with latency check
    session.run("poetry", "run", "pytest", "tests/e2e/", "-v", "-m", "latency")


@nox.session(python=PYTHON_VERSIONS)
def package(session):
    """Build and package the application."""
    session.install("poetry")
    session.run("poetry", "build")


@nox.session(python=PYTHON_VERSIONS)
def cdk_synth(session):
    """Synthesize CDK templates."""
    session.install("poetry")
    session.run("poetry", "install")
    session.run("poetry", "run", "python", "-m", "scripts.deploy", "--dry-run")


@nox.session(python=PYTHON_VERSIONS)
def teardown(session):
    """Teardown development resources."""
    session.install("poetry")
    session.run("poetry", "install")
    session.run("poetry", "run", "python", "-m", "scripts.deploy", "--teardown")