def __init__(self, username_pattern, base_dn):
authenticator = LdapAuthenticator(
{
'url': 'unused',
'username_pattern': username_pattern,
'base_dn': base_dn
}, test_utils.temp_folder)
def connect(username, password):
server = Server('mock_server', get_info=OFFLINE_AD_2012_R2)
connection = Connection(server,
user=username,
password=password,
authentication=SIMPLE,
read_only=True,
client_strategy=MOCK_SYNC)
for dn, attrs in self._entries.items():
dn = safe_dn(dn).lower()
entry_added = connection.strategy.add_entry(dn, attrs)
if not entry_added:
raise Exception('Failed to add entry ' + dn)
lower_keys = {key.lower(): key for key in attrs.keys()}
if 'samaccountname' in lower_keys:
account_name = attrs[lower_keys['samaccountname']][0]
domain_start = dn.find('dc=') + 3
domain_end = dn.find(',', domain_start)
domain = dn[domain_start:domain_end]
connection.server.dit[
domain + '\\' +
account_name] = connection.server.dit[dn]
if 'userprincipalname' in lower_keys:
principal_name = attrs[lower_keys['userprincipalname']][0]
connection.server.dit[
principal_name] = connection.server.dit[dn]
connection.bind()
return connection
authenticator._connect = connect
self.base_dn = base_dn
self._entries = {}
self.authenticator = authenticator
def create_authenticator(auth_object, temp_folder):
auth_type = auth_object.get('type')
if not auth_type:
raise Exception('Auth type should be specified')
auth_type = auth_type.strip().lower()
if auth_type == 'ldap':
from auth.auth_ldap import LdapAuthenticator
authenticator = LdapAuthenticator(auth_object, temp_folder)
elif auth_type == 'google_oauth':
from auth.auth_google_oauth import GoogleOauthAuthenticator
authenticator = GoogleOauthAuthenticator(auth_object)
elif auth_type == 'gitlab':
from auth.auth_gitlab import GitlabOAuthAuthenticator
authenticator = GitlabOAuthAuthenticator(auth_object)
elif auth_type == 'htpasswd':
from auth.auth_htpasswd import HtpasswdAuthenticator
authenticator = HtpasswdAuthenticator(auth_object)
else:
raise Exception(auth_type + ' auth is not supported')
authenticator.auth_expiration_days = float(
auth_object.get('expiration_days', 30))
authenticator.auth_type = auth_type
return authenticator
def create_authenticator(auth_object, temp_folder):
auth_type = auth_object.get('type')
if not auth_type:
raise Exception('Auth type should be specified')
auth_type = auth_type.strip().lower()
if auth_type == 'ldap':
from auth.auth_ldap import LdapAuthenticator
authenticator = LdapAuthenticator(auth_object, temp_folder)
elif auth_type == 'google_oauth':
from auth.auth_google_oauth import GoogleOauthAuthenticator
authenticator = GoogleOauthAuthenticator(auth_object)
else:
raise Exception(auth_type + ' auth is not supported')
authenticator.auth_type = auth_type
return authenticator
def create_authenticator(auth_object, temp_folder):
auth_type = auth_object.get('type')
if not auth_type:
raise Exception('Auth type should be specified')
auth_type = auth_type.strip().lower()
if auth_type == 'ldap':
from auth.auth_ldap import LdapAuthenticator
authenticator = LdapAuthenticator(auth_object, temp_folder)
elif auth_type == 'google_oauth':
from auth.auth_google_oauth import GoogleOauthAuthenticator
authenticator = GoogleOauthAuthenticator(auth_object)
else:
raise Exception(auth_type + ' auth is not supported')
authenticator.auth_type = auth_type
return authenticator