def setup_sl_domain() -> SLDomain:
"""Take the first SLDomain and set its can_use_subdomain=True"""
sl_domain: SLDomain = SLDomain.first()
sl_domain.can_use_subdomain = True
Session.commit()
return sl_domain
def test_available_suffixes_default_domain(flask_client):
user = login(flask_client)
sl_domain = SLDomain.first()
CustomDomain.create(user_id=user.id,
domain="test.com",
ownership_verified=True,
commit=True)
user.default_alias_public_domain_id = sl_domain.id
# first suffix is SL Domain
first_suffix = get_available_suffixes(user)[0]
assert first_suffix.suffix.endswith(f"@{sl_domain.domain}")
user.default_alias_public_domain_id = None
# first suffix is custom domain
first_suffix = get_available_suffixes(user)[0]
assert first_suffix.suffix == "@test.com"