plug SAML
[auf_rh_dae.git] / project / rh / test / common.py
index fb238c3..643229e 100644 (file)
@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 
 import datetime
 # -*- coding: utf-8 -*-
 
 import datetime
+from auf.django.saml.settings import SAML_REDIRECT_FIELD_NAME
 from django.contrib.auth.models import User, Group
 from project import groups
 from auf.django.references import models as ref
 from django.contrib.auth.models import User, Group
 from project import groups
 from auf.django.references import models as ref
@@ -284,7 +285,9 @@ class RhTest(TestCase):
 
     def _test_acces_ko(self, url):
         response = self.client.get(url, follow=True)
 
     def _test_acces_ko(self, url):
         response = self.client.get(url, follow=True)
-        is_ko = response.status_code in (403, 404) or 'next' in response.context
+        is_ko = response.status_code in (403, 404) or \
+                'next' in response.context or \
+                SAML_REDIRECT_FIELD_NAME in response.context['request'].GET.keys()
         self.assertEqual(is_ko, True)
 
     def _test_anonyme(self):
         self.assertEqual(is_ko, True)
 
     def _test_anonyme(self):