def test_retrieve_user(self):
with mock.patch('lizard_auth_client.client._do_post', return_value={
'success': True,
'user': {'username': '******'}}):
user_data = client.sso_get_user_django('root')
logger.debug(pprint.pformat(user_data))
self.assertEqual(user_data['username'], 'root')
def handle(self, *args, **options):
"""
"""
sso_user = options.get('sso_user')
if not sso_user:
msg = '[E] Please provide the username for the user ' \
'you are trying to sync.'
raise Exception(msg)
if V:
print("[*] About to SSO-sync data for a User "
"with username '%s'..." % sso_user)
try:
user_data = sso_get_user_django(sso_user)
if V:
print("[+] Received serialized User object: %s"
% str(user_data))
user = construct_user(user_data)
if V:
print("[+] OK, build User instance: %s" % str(user))
user.save()
if V:
print("[+] OK, succesfully saved this User instance!")
except Exception as err:
print("[E] err = '%s'" % str(err))
