src/scripts/test_db.py

"""
test_db.py

This script verifies the database connection for the Expressive TTS Arena project.
It attempts to connect to the PostgreSQL database using SQLAlchemy and executes a simple query.

Functionality:
- Loads the database connection from `database.py`.
- Attempts to establish a connection to the database.
- Executes a test query (`SELECT 1`) to confirm connectivity.
- Prints a success message if the connection is valid.
- Prints an error message if the connection fails.

Usage:
    python src/test_db.py

Expected Output:
    Database connection successful!  (if the database is reachable)
    Database connection failed: <error message> (if there are connection issues)

Troubleshooting:
- Ensure the `.env` file contains a valid `DATABASE_URL`.
- Check that the database server is running and accessible.
- Verify PostgreSQL credentials and network settings.

"""

# Third-Party Library Imports
from sqlalchemy import text
from sqlalchemy.exc import OperationalError

# Local Application Imports
from src.config import logger
from src.database import engine

try:
    with engine.connect() as conn:
        result = conn.execute(text("SELECT 1"))
        logger.info("Database connection successful!")
except OperationalError as e:
    logger.error(f"Database connection failed: {e}")