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"