changelist
authorOlivier Larchevêque <olivier.larcheveque@auf.org>
Thu, 31 May 2012 20:50:56 +0000 (16:50 -0400)
committerOlivier Larchevêque <olivier.larcheveque@auf.org>
Thu, 31 May 2012 20:50:56 +0000 (16:50 -0400)
project/rh/admin.py
project/rh/test/employe.py

index 77ca1ab..f00ad29 100644 (file)
@@ -114,7 +114,9 @@ class ProtectRegionMixin(object):
         if in_drh_or_admin(request.user):
             return qs
 
-        if groups.grp_correspondants_rh in user_groups:
+        if groups.grp_correspondants_rh in user_groups or\
+           groups.grp_administrateurs in user_groups or\
+           groups.grp_directeurs_bureau in user_groups:
             employe = get_employe_from_user(request.user)
             q = Q(**{self.model.prefix_implantation: \
                     employe.implantation.region})
index c4c9871..68d9892 100644 (file)
@@ -183,31 +183,39 @@ class EmployeListTest(RhTest):
         """
         self._test_correspondant_rh()
         self._test_acces_ok(self.url)
+        qs = self.client.get(self.url).context['cl'].query_set
+        self.assertEqual(len(qs), 2)
 
     def test_administrateur_regional(self):
         """
-        Un administrateur peut voir les employés
+        Un administrateur peut voir les employés de sa région
         """
         self._test_administrateur_regional()
         self._test_acces_ok(self.url)
+        qs = self.client.get(self.url).context['cl'].query_set
+        self.assertEqual(len(qs), 2)
 
     def test_directeur_bureau(self):
         """
-        Un directeur de bureau peut voir les employés
+        Un directeur de bureau peut voir les employés de sa région
         """
         self._test_directeur_bureau()
         self._test_acces_ok(self.url)
+        qs = self.client.get(self.url).context['cl'].query_set
+        self.assertEqual(len(qs), 2)
 
     def test_drh(self):
         """
-        Un DRH  peut voir les employés
+        Un DRH  peut voir tous les employés
         """
         self._test_drh()
         self._test_acces_ok(self.url)
+        qs = self.client.get(self.url).context['cl'].query_set
+        self.assertEqual(len(qs), 4)
 
     def test_drh2(self):
         """
-        Un DRH (2ieme niveau) peut voir les employés
+        Un DRH (2ieme niveau) peut voir tous les employés
         """
         self._test_drh2()
         self._test_acces_ok(self.url)