40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
"""Hooks for initial schema revision a1b2c3d4e5f6."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from nanoid import generate as generate_nanoid
|
|
|
|
from tests.alembic.registry import register_after_upgrade, register_before_upgrade
|
|
from tests.alembic.verifier import MigrationVerifier
|
|
|
|
APP_ID = generate_nanoid()
|
|
APP_NAME = "seed-app"
|
|
USER_ID = generate_nanoid()
|
|
USER_NAME = "seed-user"
|
|
SESSION_ID = generate_nanoid()
|
|
SESSION_NAME = "seed-session"
|
|
MESSAGE_PUBLIC_ID = generate_nanoid()
|
|
MESSAGE_CONTENT = "seed message"
|
|
METAMESSAGE_PUBLIC_ID = generate_nanoid()
|
|
METAMESSAGE_CONTENT = "seed metamessage"
|
|
COLLECTION_ID = generate_nanoid()
|
|
COLLECTION_NAME = "seed-collection"
|
|
DOCUMENT_ID = generate_nanoid()
|
|
DOCUMENT_CONTENT = "Seed document"
|
|
|
|
|
|
@register_before_upgrade("a1b2c3d4e5f6")
|
|
def prepare_initial(_verifier: MigrationVerifier) -> None:
|
|
pass
|
|
|
|
|
|
@register_after_upgrade("a1b2c3d4e5f6")
|
|
def verify_initial_schema(verifier: MigrationVerifier) -> None:
|
|
verifier.assert_table_exists("apps", exists=True)
|
|
verifier.assert_table_exists("users", exists=True)
|
|
verifier.assert_table_exists("sessions", exists=True)
|
|
verifier.assert_table_exists("messages", exists=True)
|
|
verifier.assert_table_exists("collections", exists=True)
|
|
verifier.assert_table_exists("documents", exists=True)
|
|
verifier.assert_table_exists("metamessages", exists=True)
|