53 lines
2.1 KiB
Python
53 lines
2.1 KiB
Python
|
"""
|
||
|
|
Test script to verify the Alembic migration for the avdb_image table.
|
||
|
|
"""
|
||
|
|
import sqlalchemy
|
||
|
|
from sqlalchemy import inspect, text
|
||
|
|
|
||
|
|
# Database URL from alembic.ini
|
||
|
|
DATABASE_URL = "postgresql+psycopg2://postgres:pgRocks@localhost:5432/testdb"
|
||
|
|
|
||
|
|
def test_avdb_image_table_exists():
|
||
|
|
"""Test if the avdb_image table exists in the database."""
|
||
|
|
engine = sqlalchemy.create_engine(DATABASE_URL)
|
||
|
|
inspector = inspect(engine)
|
||
|
|
|
||
|
|
# Print database and table configuration
|
||
|
|
print("Database URL:", DATABASE_URL)
|
||
|
|
print("\nChecking for avdb_image table...")
|
||
|
|
|
||
|
|
# Check if the table exists
|
||
|
|
assert inspector.has_table("avdb_image"), "avdb_image table does not exist!"
|
||
|
|
print("✓ avdb_image table exists.")
|
||
|
|
|
||
|
|
# Verify the table structure
|
||
|
|
columns = inspector.get_columns("avdb_image")
|
||
|
|
column_names = [column["name"] for column in columns]
|
||
|
|
|
||
|
|
print("\nColumns in avdb_image table:")
|
||
|
|
for column in columns:
|
||
|
|
print(f" - {column['name']}: {column['type']}, Nullable: {column['nullable']}")
|
||
|
|
|
||
|
|
expected_columns = ["img_id", "img_path", "img_name", "width", "height", "created_at"]
|
||
|
|
for col in expected_columns:
|
||
|
|
assert col in column_names, f"Column '{col}' not found in avdb_image table!"
|
||
|
|
print("\n✓ All expected columns are present in the avdb_image table.")
|
||
|
|
|
||
|
|
# Verify primary key
|
||
|
|
primary_key = inspector.get_pk_constraint("avdb_image")
|
||
|
|
assert primary_key is not None, "Primary key not found in avdb_image table!"
|
||
|
|
assert primary_key["constrained_columns"] == ["img_id"], "Primary key is not set on img_id!"
|
||
|
|
print("✓ Primary key is correctly set on img_id.")
|
||
|
|
|
||
|
|
# Verify nullable constraints
|
||
|
|
nullable_columns = [col["name"] for col in columns if col["nullable"]]
|
||
|
|
non_nullable_columns = ["img_path", "img_name", "width", "height"]
|
||
|
|
for col in non_nullable_columns:
|
||
|
|
assert col not in nullable_columns, f"Column '{col}' should not be nullable!"
|
||
|
|
print("✓ Non-nullable constraints are correctly set.")
|
||
|
|
|
||
|
|
print("\nAll tests passed! The avdb_image table was created successfully.")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
test_avdb_image_table_exists()
|