k1nq
69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
from datetime import date
|
|
from pathlib import Path
|
|
|
|
from alembic.config import Config
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import Session
|
|
|
|
from alembic import command
|
|
from app.config import get_settings
|
|
from app.identity.models import User
|
|
from app.ledger.models import Account, Category
|
|
from app.spaces.models import FinancialSpace, Membership
|
|
|
|
|
|
def test_upgrade_backfills_one_pair_without_claiming_same_named_rows(
|
|
tmp_path: Path, monkeypatch,
|
|
) -> None:
|
|
database_url = f"sqlite:///{tmp_path / 'migration.db'}"
|
|
monkeypatch.setenv("DATABASE_URL", database_url)
|
|
get_settings.cache_clear()
|
|
config = Config(str(Path(__file__).parents[1] / "alembic.ini"))
|
|
config.set_main_option("script_location", str(Path(__file__).parents[1] / "alembic"))
|
|
|
|
try:
|
|
command.upgrade(config, "0002")
|
|
engine = create_engine(database_url)
|
|
with Session(engine) as db:
|
|
user = User(email="migration@example.com", password_hash="unused")
|
|
space = FinancialSpace(name="Legacy")
|
|
db.add_all([user, space])
|
|
db.flush()
|
|
db.add(Membership(space_id=space.id, user_id=user.id, role="owner"))
|
|
db.add(
|
|
Account(
|
|
space_id=space.id,
|
|
name="Savings",
|
|
kind="debit",
|
|
opened_on=date(2026, 1, 1),
|
|
)
|
|
)
|
|
db.add(
|
|
Category(
|
|
space_id=space.id,
|
|
name_override="Savings",
|
|
kind="monthly",
|
|
)
|
|
)
|
|
db.commit()
|
|
space_id = space.id
|
|
|
|
command.upgrade(config, "head")
|
|
command.upgrade(config, "head")
|
|
|
|
with Session(engine) as db:
|
|
accounts = db.query(Account).filter_by(space_id=space_id).all()
|
|
categories = db.query(Category).filter_by(space_id=space_id).all()
|
|
system_account = [item for item in accounts if item.system_key == "savings"]
|
|
system_category = [item for item in categories if item.system_key == "savings"]
|
|
assert len(system_account) == 1
|
|
assert len(system_category) == 1
|
|
assert system_category[0].linked_account_id == system_account[0].id
|
|
assert any(item.name == "Savings" and item.system_key is None for item in accounts)
|
|
assert any(
|
|
item.name_override == "Savings" and item.system_key is None
|
|
for item in categories
|
|
)
|
|
finally:
|
|
get_settings.cache_clear()
|