Only update refresh_token if it was returned in the response (#43)

It is not returned with renewals of the id and access token (which the documentation did not make clear).
NabuCasa · Mar 3, 2021 · 8ab7653 · 8ab7653
1 parent 6a9a1f0
commit 8ab7653
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pycognito/__init__.py b/pycognito/__init__.py
@@ -706,7 +706,8 @@ def _set_tokens(self, tokens):
         AuthenticationResult.
         """
         self.verify_token(tokens["AuthenticationResult"]["IdToken"], "id_token", "id")
-        self.refresh_token = tokens["AuthenticationResult"]["RefreshToken"]
+        if "RefreshToken" in tokens["AuthenticationResult"]:
+            self.refresh_token = tokens["AuthenticationResult"]["RefreshToken"]
         self.verify_token(
             tokens["AuthenticationResult"]["AccessToken"], "access_token", "access"
         )