Use same username as challenge parameters (NabuCasa#23)

* Use same username as challenge parameters * Adjusted layout * Fixes for tests * Fixes for tests
ludeeus · Jan 10, 2021 · 0a18e88 · 0a18e88
1 parent 6c173cf
commit 0a18e88
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pycognito/aws_srp.py b/pycognito/aws_srp.py
@@ -287,8 +287,12 @@ def set_new_password_challenge(self, new_password, client=None):
             )
 
             if tokens["ChallengeName"] == self.NEW_PASSWORD_REQUIRED_CHALLENGE:
+                challenge_parameters = response["ChallengeParameters"]
                 challenge_response.update(
-                    {"USERNAME": auth_params["USERNAME"], "NEW_PASSWORD": new_password}
+                    {
+                        "USERNAME": challenge_parameters["USERNAME"],
+                        "NEW_PASSWORD": new_password,
+                    }
                 )
                 new_password_response = boto_client.respond_to_auth_challenge(
                     ClientId=self.client_id,