Merge pull request #448 from aogail/bugfix/universalPromptStepUp

Add sessionToken to dict where step-up code expects it
Nike-Inc · Feb 12, 2024 · 0c61553 · 0c61553
2 parents 15f8420 + e42f409
commit 0c61553
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/gimme_aws_creds/duo_universal.py b/gimme_aws_creds/duo_universal.py
@@ -68,7 +68,8 @@ def do_auth(self):
                     "username": okta_profile_login,
                     "session": self.session.cookies['sid'],
                     "device_token": self.session.cookies['DT']
-                }
+                },
+                'sessionToken': self.session.cookies['sid']
             },
         }
 

diff --git a/tests/test_duo_universal_client.py b/tests/test_duo_universal_client.py
@@ -4,9 +4,9 @@
 
 import requests
 import responses
-from tests import read_fixture
 
 from gimme_aws_creds.duo_universal import OktaDuoUniversal
+from tests import read_fixture
 from tests.user_interface_mock import MockUserInterface
 
 
@@ -135,7 +135,8 @@ def test_universal_push(self):
                     'username': self.OKTA_LOGIN,
                     'session': self.OKTA_SID_VALUE,
                     'device_token': self.OKTA_DT_VALUE
-                }
+                },
+                'sessionToken': 'oktasidvalue',
             },
         }
 
@@ -157,7 +158,8 @@ def test_universal_phone_call(self):
                     'username': self.OKTA_LOGIN,
                     'session': self.OKTA_SID_VALUE,
                     'device_token': self.OKTA_DT_VALUE
-                }
+                },
+                'sessionToken': 'oktasidvalue',
             },
         }
 
@@ -180,7 +182,8 @@ def test_universal_passcode(self):
                     'username': self.OKTA_LOGIN,
                     'session': self.OKTA_SID_VALUE,
                     'device_token': self.OKTA_DT_VALUE
-                }
+                },
+                'sessionToken': 'oktasidvalue',
             },
         }