def get_user_object(self):
cog_client = boto3.client('cognito-idp')
user = cog_client.get_user(
AccessToken=self.request.session['ACCESS_TOKEN'])
u = UserObj(username=user.get('UserAttributes')[0].get('username'),
attribute_list=user.get('UserAttributes'),
attr_map=settings.COGNITO_ATTR_MAPPING)
return u
def test_init(self):
u = UserObj('bjones', self.user_info, self.user, self.user_metadata)
self.assertEqual(u.pk, self.user_metadata.get('username'))
self.assertEqual(u.name, self.user_info[0].get('Value'))
self.assertEqual(u.user_status, self.user_metadata.get('user_status'))