def test_set_session_expiration(self, mock_auth):
self.org = self.create_organization()
self.auth_provider = AuthProvider.objects.create(provider="saml2",
organization=self.org)
self.provider = SAML2Provider(key=self.auth_provider.provider)
self.provider.config = dummy_provider_config
self.auth_provider.get_provider = mock.MagicMock(
return_value=self.provider)
super(SAML2ACSViewTest, self).setUp()
request = self.make_request(user=None)
request.META = {
"PATH_INFO": "/",
}
test_view = SAML2ACSView()
helper = AuthHelper(request,
self.org,
AuthHelper.FLOW_LOGIN,
auth_provider=self.auth_provider)
def mock_next_step(self):
return
helper.next_step = types.MethodType(mock_next_step, helper)
instance = mock_auth.return_value
instance.get_errors.return_value = None
instance.get_attributes.return_value = {}
instance.get_session_expiration.return_value = 1591044492
test_view.dispatch(request, helper)
assert request.session.get_expiry_date() == datetime.fromtimestamp(
1591044492).replace(tzinfo=timezone.utc)
def setUp(self):
self.org = self.create_organization()
self.auth_provider = AuthProvider.objects.create(provider="saml2",
organization=self.org)
self.provider = SAML2Provider(key=self.auth_provider.provider)
super(SAML2ProviderTest, self).setUp()