diff --git a/tests/test_2fa_duo.py b/tests/test_2fa_duo.py index c2c5ada..8cf4c5f 100644 --- a/tests/test_2fa_duo.py +++ b/tests/test_2fa_duo.py @@ -207,7 +207,11 @@ def test_remember_me_cookie( # shib_idp_session, shib_idp_session_ss cookie_names = {cookie['name'] for cookie in fresh_browser.get_cookies()} - assert 'shib_idp_session' in cookie_names + # prod and eval are different for about a week til the new shib is released to prod. + if test_env == 'eval': + assert '__Host-shib_idp_session' in cookie_names + else: + assert 'shib_idp_session' in cookie_names assert 'shib_idp_session_ss' in cookie_names fresh_browser.get(f'{sp_url(sp)}/Shibboleth.sso/Logout?return={idp_url}/profile/Logout') @@ -215,7 +219,11 @@ def test_remember_me_cookie( fresh_browser.wait_for_tag('span', 'Your UW NetID sign-in session has ended.') cookie_names = {cookie['name'] for cookie in fresh_browser.get_cookies()} - assert 'shib_idp_session' not in cookie_names + # prod and eval are different for about a week til the new shib is released to prod. + if test_env == 'eval': + assert '__Host-shib_idp_session' not in cookie_names + else: + assert 'shib_idp_session' not in cookie_names assert 'shib_idp_session_ss' not in cookie_names sp = ServiceProviderInstance.diafine12