def testGetUserKwrgsExtraField(self):
with self.assertRaises(TypeError, msg="Expected arguments: username"):
with connection() as c:
c.get_user(
username = settings.LDAP_AUTH_TEST_USER_USERNAME,
foo = "foo",
)
def testGetUserBothArgsAndKwargs(self):
with self.assertRaises(TypeError, msg="Cannot use both args and kwargs to identify a user"):
with connection() as c:
c.get_user(
settings.LDAP_AUTH_TEST_USER_USERNAME,
username = settings.LDAP_AUTH_TEST_USER_USERNAME,
)
def testGetUserKwrgsExtraField(self):
with self.assertRaises(TypeError, msg="Expected arguments: username"):
with connection() as c:
c.get_user(
username=settings.LDAP_AUTH_TEST_USER_USERNAME,
foo="foo",
)
def testGetUserKwargsSuccess(self):
with connection() as c:
user = c.get_user(
username = settings.LDAP_AUTH_TEST_USER_USERNAME,
)
self.assertIsInstance(user, User)
self.assertEqual(user.username, settings.LDAP_AUTH_TEST_USER_USERNAME)
def testGetUserKwargsSuccess(self):
with connection() as c:
user = c.get_user(
username=settings.LDAP_AUTH_TEST_USER_USERNAME,
)
self.assertIsInstance(user, User)
self.assertEqual(user.username, settings.LDAP_AUTH_TEST_USER_USERNAME)
def handle_noargs(self, **kwargs):
verbosity = int(kwargs.get("verbosity", 1))
with ldap.connection(username=settings.LDAP_AUTH_CONNECTION_USERNAME, password=settings.LDAP_AUTH_CONNECTION_PASSWORD) as connection:
for user in connection.iter_users():
if verbosity >= 1:
self.stdout.write("Synced {user}".format(
user = user,
))
def handle_noargs(self, **kwargs):
verbosity = int(kwargs.get("verbosity", 1))
with ldap.connection() as connection:
for user in connection.iter_users():
if verbosity >= 1:
self.stdout.write("Synced {user}".format(
user = user,
))
def testGetUserBothArgsAndKwargs(self):
with self.assertRaises(
TypeError,
msg="Cannot use both args and kwargs to identify a user"):
with connection() as c:
c.get_user(
settings.LDAP_AUTH_TEST_USER_USERNAME,
username=settings.LDAP_AUTH_TEST_USER_USERNAME,
)
def handle(self, *args, **kwargs):
verbosity = int(kwargs.get("verbosity", 1))
with ldap.connection(
username=settings.LDAP_AUTH_CONNECTION_USERNAME,
password=settings.LDAP_AUTH_CONNECTION_PASSWORD,
) as connection:
if connection is None:
raise CommandError("Could not connect to LDAP server")
for user in connection.iter_users():
if verbosity >= 1:
self.stdout.write("Synced {user}".format(user=user, ))
def handle(self, *args, **kwargs):
verbosity = int(kwargs.get("verbosity", 1))
User = get_user_model()
auth_kwargs = {
User.USERNAME_FIELD: settings.LDAP_AUTH_CONNECTION_USERNAME,
'password': settings.LDAP_AUTH_CONNECTION_PASSWORD
}
with ldap.connection(**auth_kwargs) as connection:
if connection is None:
raise CommandError("Could not connect to LDAP server")
for user in connection.iter_users():
if verbosity >= 1:
self.stdout.write("Synced {user}".format(user=user, ))
def testGetUserKwargsIncorrectUsername(self):
with connection() as c:
user = c.get_user(username="******" +
settings.LDAP_AUTH_TEST_USER_USERNAME, )
self.assertEqual(user, None)
def testGetUserKwargsIncorrectUsername(self):
with connection() as c:
user = c.get_user(username="******" + settings.LDAP_AUTH_TEST_USER_USERNAME)
self.assertEqual(user, None)
def testGetUserMissingArgsAndKwargs(self):
with self.assertRaises(TypeError, msg="Expected arguments: username"):
with connection() as c:
c.get_user()
def testGetUserMissingArgsAndKwargs(self):
with self.assertRaises(TypeError, msg="Expected arguments: username"):
with connection() as c:
c.get_user()
def handle_noargs(self, **kwargs):
verbosity = int(kwargs.get("verbosity", 1))
with ldap.connection() as connection:
for user in connection.iter_users():
if verbosity >= 1:
self.stdout.write("Synced {user}".format(user=user, ))
