3252a8
5e47a1496b
Add the Remnashop legacy importer, compatibility tables, admin toggles, referral and promo lookup compatibility, and tests for the migration flow.
63 lines
2.4 KiB
Python
63 lines
2.4 KiB
Python
from db.migrator import MIGRATIONS
|
|
from db.models import (
|
|
LegacyImportMapping,
|
|
LegacyReferralCode,
|
|
SupportTicket,
|
|
SupportTicketMessage,
|
|
User,
|
|
)
|
|
|
|
|
|
def test_support_migration_is_registered_after_existing_revisions():
|
|
ids = [migration.id for migration in MIGRATIONS]
|
|
|
|
assert "0024_add_support_tickets" in ids
|
|
assert ids.index("0024_add_support_tickets") > ids.index("0023_add_email_password_auth_fields")
|
|
assert "0025_add_support_notification_timestamps" in ids
|
|
assert ids.index("0025_add_support_notification_timestamps") > ids.index(
|
|
"0024_add_support_tickets"
|
|
)
|
|
assert "0026_add_lifetime_traffic_synced_at" in ids
|
|
assert ids.index("0026_add_lifetime_traffic_synced_at") > ids.index(
|
|
"0025_add_support_notification_timestamps"
|
|
)
|
|
|
|
|
|
def test_support_models_expose_expected_tables():
|
|
assert SupportTicket.__tablename__ == "support_tickets"
|
|
assert SupportTicketMessage.__tablename__ == "support_ticket_messages"
|
|
assert "admin_last_notified_at" in SupportTicket.__table__.columns
|
|
assert "admin_last_emailed_at" in SupportTicket.__table__.columns
|
|
assert "ix_support_tickets_status_last_msg" in {
|
|
index.name for index in SupportTicket.__table__.indexes
|
|
}
|
|
|
|
|
|
def test_user_model_tracks_lifetime_traffic_sync_timestamp():
|
|
assert "lifetime_used_traffic_synced_at" in User.__table__.columns
|
|
|
|
|
|
def test_trial_eligibility_reset_migration_and_model_are_registered():
|
|
ids = [migration.id for migration in MIGRATIONS]
|
|
|
|
assert "0033_add_trial_eligibility_reset_marker" in ids
|
|
assert ids.index("0033_add_trial_eligibility_reset_marker") > ids.index(
|
|
"0032_add_telegram_notification_status"
|
|
)
|
|
assert "trial_eligibility_reset_at" in User.__table__.columns
|
|
|
|
|
|
def test_legacy_import_compatibility_migration_and_models_are_registered():
|
|
ids = [migration.id for migration in MIGRATIONS]
|
|
|
|
assert "0034_add_legacy_import_compatibility" in ids
|
|
assert ids.index("0034_add_legacy_import_compatibility") > ids.index(
|
|
"0033_add_trial_eligibility_reset_marker"
|
|
)
|
|
assert User.__table__.columns["referral_code"].type.length == 64
|
|
assert LegacyReferralCode.__tablename__ == "legacy_referral_codes"
|
|
assert LegacyImportMapping.__tablename__ == "legacy_import_mappings"
|
|
assert "uq_legacy_referral_source_code" in {
|
|
constraint.name for constraint in LegacyReferralCode.__table__.constraints
|
|
}
|