Changeset 8857 for main/waeup.kofa/trunk/src/waeup/kofa/mandates

Timestamp:

30 Jun 2012, 07:03:20 (12 years ago)

Author:

Henrik Bettermann

Message:

Let's use the PasswordMandate? also for change password requests (part1).

Location:

main/waeup.kofa/trunk/src/waeup/kofa/mandates

Files:

• mandate.py (modified) (2 diffs)
• tests.py (modified) (5 diffs)

main/waeup.kofa/trunk/src/waeup/kofa/mandates/mandate.py

@@ -50 +50 @@
         return _('Nothing to do.')
 
-class StudentPasswordMandate(Mandate):
+class PasswordMandate(Mandate):
     """This is a mandate which can set a student password.
     """
 
-    def _setStudentPassword(self):
-        student_id = self.params.get('student_id', None)
+    def _setPassword(self):
+        user_id = self.params.get('user_id', None)
         pwd = self.params.get('password', None)
-        student = grok.getSite()['students'].get(student_id, None)
-        if student and pwd:
-            IUserAccount(student).setPassword(pwd)
+        user_type = self.params.get('user_type', None)
+        user = None
+        if user_type == 'student':
+            user = grok.getSite()['students'].get(user_id, None)
+        elif user_type == 'officer':
+            user = grok.getSite()['users'].get(user_id, None)
+        if user and pwd:
+            IUserAccount(user).setPassword(pwd)
             return True
         return False
@@ -67 +72 @@
         if self.expires < datetime.utcnow():
             msg = _('Mandate expired.')
-        if self._setStudentPassword():
+        if self._setPassword():
             msg = _('Password has been successfully set. '
                     'Proceed to the login page and enter your credentials.')

main/waeup.kofa/trunk/src/waeup/kofa/mandates/tests.py

@@ -31 +31 @@
     IMandatesContainer, IMandate)
 from waeup.kofa.mandates.container import MandatesContainer
-from waeup.kofa.mandates.mandate import StudentPasswordMandate
+from waeup.kofa.mandates.mandate import PasswordMandate
 from waeup.kofa.testing import (FunctionalLayer, FunctionalTestCase)
 
@@ -50 +50 @@
         self.assertTrue(
             verifyClass(
-                IMandate, StudentPasswordMandate)
+                IMandate, PasswordMandate)
             )
         self.assertTrue(
             verifyObject(
-                IMandate, StudentPasswordMandate())
+                IMandate, PasswordMandate())
             )
         return
@@ -84 +84 @@
         shutil.rmtree(self.dc_root)
 
-    def test_set_password(self):
+    def test_set_student_password(self):
         student = createObject('waeup.Student')
         self.app['students'].addStudent(student)
         # Add and execute a mandate with missing parameters.
-        mandate = StudentPasswordMandate()
+        mandate = PasswordMandate()
         self.app['mandates'].addMandate(mandate)
         msg = mandate.execute()
         self.assertEqual(msg, u'Wrong mandate parameters.')
         # Add and execute an expired mandate.
-        mandate = StudentPasswordMandate(days=0)
+        mandate = PasswordMandate(days=0)
         self.app['mandates'].addMandate(mandate)
         msg = mandate.execute()
         self.assertEqual(msg, u'Mandate expired.')
         # Add and execute a perfect mandate
-        mandate = StudentPasswordMandate()
-        mandate.params['student_id'] = student.student_id
+        mandate = PasswordMandate()
+        mandate.params['user_id'] = student.student_id
         mandate.params['password'] = 'mypwd1'
+        mandate.params['user_type'] = 'student'
         self.app['mandates'].addMandate(mandate)
         msg = mandate.execute()
@@ -110 +111 @@
         self.assertEqual(len(self.app['mandates'].keys()), 0)
 
+    def test_set_officer_password(self):
+        self.app['users'].addUser('bob', 'bobssecret')
+        officer = self.app['users']['bob']
+        mandate = PasswordMandate()
+        mandate.params['user_id'] = 'bob'
+        mandate.params['password'] = 'mypwd1'
+        mandate.params['user_type'] = 'officer'
+        self.app['mandates'].addMandate(mandate)
+        msg = mandate.execute()
+        # Password has been set.
+        self.assertEqual(msg, 'Password has been successfully set. Proceed to '
+            'the login page and enter your credentials.')
+        self.assertTrue(IUserAccount(officer).checkPassword('mypwd1'))
+
     def test_remove_expired(self):
         # mandate1 is an old mandate which just expired.
-        mandate1 = StudentPasswordMandate(days=0)
+        mandate1 = PasswordMandate(days=0)
         self.app['mandates'].addMandate(mandate1)
         # mandate2 is a new mandate with default time delta.
-        mandate2 = StudentPasswordMandate(mandate_id='23456')
+        mandate2 = PasswordMandate(mandate_id='23456')
         self.app['mandates'].addMandate(mandate2)
         self.assertEqual(len(self.app['mandates'].keys()), 2)
@@ -126 +141 @@
         student = createObject('waeup.Student')
         self.app['students'].addStudent(student)
-        mandate = StudentPasswordMandate()
-        mandate.params['student_id'] = student.student_id
+        mandate = PasswordMandate()
+        mandate.params['user_id'] = student.student_id
+        mandate.params['user_type'] = 'student'
         mandate.params['password'] = 'mypwd1'
         self.app['mandates'].addMandate(mandate)