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()