diff --git a/gimme_aws_creds/duo_universal.py b/gimme_aws_creds/duo_universal.py index e3a3c6b..eebbba0 100644 --- 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 index ac08c06..56b282d 100644 --- 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', }, }