55 lines
2.3 KiB
Python
55 lines
2.3 KiB
Python
import pytest
|
|
|
|
from mock import patch
|
|
|
|
from data.database import model
|
|
from data.users.shared import can_create_user
|
|
|
|
from test.fixtures import *
|
|
|
|
@pytest.mark.parametrize('open_creation, invite_only, email, has_invite, can_create', [
|
|
# Open user creation => always allowed.
|
|
(True, False, None, False, True),
|
|
|
|
# Open user creation => always allowed.
|
|
(True, False, 'foo@example.com', False, True),
|
|
|
|
# Invite only user creation + no invite => disallowed.
|
|
(True, True, None, False, False),
|
|
|
|
# Invite only user creation + no invite => disallowed.
|
|
(True, True, 'foo@example.com', False, False),
|
|
|
|
# Invite only user creation + invite => allowed.
|
|
(True, True, 'foo@example.com', True, True),
|
|
|
|
# No open creation => Disallowed.
|
|
(False, True, 'foo@example.com', False, False),
|
|
(False, True, 'foo@example.com', True, False),
|
|
|
|
# Blacklisted emails => Disallowed.
|
|
(True, False, 'foo@blacklisted.com', False, False),
|
|
(True, False, 'foo@blacklisted.org', False, False),
|
|
(True, False, 'foo@BlAcKlIsTeD.CoM', False, False), # Verify Capitalization
|
|
(True, False, u'foo@mail.bLacklisted.Com', False, False), # Verify unicode
|
|
(True, False, 'foo@blacklisted.net', False, True), # Avoid False Positives
|
|
(True, False, 'foo@myblacklisted.com', False, True), # Avoid partial domain matches
|
|
(True, False, 'fooATblacklisted.com', False, True), # Ignore invalid email addresses
|
|
])
|
|
@pytest.mark.parametrize('blacklisting_enabled', [True, False])
|
|
def test_can_create_user(open_creation, invite_only, email, has_invite, can_create, blacklisting_enabled, app):
|
|
|
|
# Mock list of blacklisted domains
|
|
blacklisted_domains = ['blacklisted.com', 'blacklisted.org']
|
|
|
|
if has_invite:
|
|
inviter = model.user.get_user('devtable')
|
|
team = model.team.get_organization_team('buynlarge', 'owners')
|
|
model.team.add_or_invite_to_team(inviter, team, email=email)
|
|
|
|
with patch('features.USER_CREATION', open_creation):
|
|
with patch('features.INVITE_ONLY_USER_CREATION', invite_only):
|
|
with patch('features.BLACKLISTED_EMAILS', blacklisting_enabled):
|
|
if email and any(domain in email.lower() for domain in blacklisted_domains) and not blacklisting_enabled:
|
|
can_create = True # blacklisted domains can be used, if blacklisting is disabled
|
|
assert can_create_user(email, blacklisted_domains) == can_create
|