source: main/waeup.kofa/branches/0.2/src/waeup/kofa/mandates/tests.py @ 14195

Last change on this file since 14195 was 8910, checked in by Henrik Bettermann, 12 years ago

Let removeExpired return the number of deleted mandates.

  • Property svn:keywords set to Id
File size: 7.3 KB
Line 
1## $Id: tests.py 8910 2012-07-04 07:52:21Z henrik $
2##
3## Copyright (C) 2012 Uli Fouquet & Henrik Bettermann
4## This program is free software; you can redistribute it and/or modify
5## it under the terms of the GNU General Public License as published by
6## the Free Software Foundation; either version 2 of the License, or
7## (at your option) any later version.
8##
9## This program is distributed in the hope that it will be useful,
10## but WITHOUT ANY WARRANTY; without even the implied warranty of
11## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12## GNU General Public License for more details.
13##
14## You should have received a copy of the GNU General Public License
15## along with this program; if not, write to the Free Software
16## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17##
18"""
19Tests for mandates.
20"""
21import tempfile
22import shutil
23import os
24from zope.testbrowser.testing import Browser
25from datetime import datetime, timedelta
26from zope.interface.verify import verifyClass, verifyObject
27from zope.component import createObject
28from zope.component.hooks import setSite, clearSite
29from waeup.kofa.app import University
30from waeup.kofa.interfaces import IUserAccount
31from waeup.kofa.mandates.interfaces import (
32    IMandatesContainer, IMandate)
33from waeup.kofa.mandates.container import MandatesContainer
34from waeup.kofa.mandates.mandate import PasswordMandate
35from waeup.kofa.testing import (FunctionalLayer, FunctionalTestCase)
36
37class MandatesContainerTestCase(FunctionalTestCase):
38
39    layer = FunctionalLayer
40
41    def test_interfaces(self):
42        # Make sure the correct interfaces are implemented.
43        self.assertTrue(
44            verifyClass(
45                IMandatesContainer, MandatesContainer)
46            )
47        self.assertTrue(
48            verifyObject(
49                IMandatesContainer, MandatesContainer())
50            )
51        self.assertTrue(
52            verifyClass(
53                IMandate, PasswordMandate)
54            )
55        self.assertTrue(
56            verifyObject(
57                IMandate, PasswordMandate())
58            )
59        return
60
61    def setUp(self):
62        super(MandatesContainerTestCase, self).setUp()
63
64        # Setup a sample site for each test
65        app = University()
66        self.dc_root = tempfile.mkdtemp()
67        app['datacenter'].setStoragePath(self.dc_root)
68
69        # Prepopulate the ZODB...
70        self.getRootFolder()['app'] = app
71        # we add the site immediately after creation to the
72        # ZODB. Catalogs and other local utilities are not setup
73        # before that step.
74        self.app = self.getRootFolder()['app']
75        # Set site here. Some of the following setup code might need
76        # to access grok.getSite() and should get our new app then
77        setSite(app)
78
79        self.browser = Browser()
80        self.browser.handleErrors = False
81
82    def tearDown(self):
83        super(MandatesContainerTestCase, self).tearDown()
84        clearSite()
85        shutil.rmtree(self.dc_root)
86
87    def test_set_student_password(self):
88        student = createObject('waeup.Student')
89        # Add and execute a mandate with missing parameters.
90        mandate = PasswordMandate()
91        self.app['mandates'].addMandate(mandate)
92        msg = mandate.execute()
93        self.assertEqual(msg, u'Wrong mandate parameters.')
94        # Add and execute an expired mandate.
95        mandate = PasswordMandate(days=0)
96        self.app['mandates'].addMandate(mandate)
97        msg = mandate.execute()
98        self.assertEqual(msg, u'Mandate expired.')
99        # Add and execute a perfect mandate
100        mandate = PasswordMandate()
101        mandate.params['user'] = student
102        mandate.params['password'] = 'mypwd1'
103        self.app['mandates'].addMandate(mandate)
104        msg = mandate.execute()
105        # Password has been set.
106        self.assertEqual(msg, 'Password has been successfully set. Proceed to '
107            'the login page and enter your credentials.')
108        self.assertTrue(IUserAccount(student).checkPassword('mypwd1'))
109        # All mandates have been removed.
110        self.assertEqual(len(self.app['mandates'].keys()), 0)
111        logfile = os.path.join(
112            self.app['datacenter'].storage, 'logs', 'main.log')
113        logcontent = open(logfile).read()
114        self.assertTrue('system - PasswordMandate used: K1000000' in logcontent)
115
116    def test_set_officer_password(self):
117        self.app['users'].addUser('bob', 'bobssecret')
118        officer = self.app['users']['bob']
119        mandate = PasswordMandate()
120        mandate.params['user'] = officer
121        mandate.params['password'] = 'mypwd1'
122        self.app['mandates'].addMandate(mandate)
123        msg = mandate.execute()
124        # Password has been set.
125        self.assertEqual(msg, 'Password has been successfully set. Proceed to '
126            'the login page and enter your credentials.')
127        self.assertTrue(IUserAccount(officer).checkPassword('mypwd1'))
128        logfile = os.path.join(
129            self.app['datacenter'].storage, 'logs', 'main.log')
130        logcontent = open(logfile).read()
131        self.assertTrue('system - PasswordMandate used: bob' in logcontent)
132
133    def test_set_applicant_password(self):
134        applicant = createObject('waeup.Applicant')
135        applicant.applicant_id = u'abc'
136        mandate = PasswordMandate()
137        mandate.params['user'] = applicant
138        mandate.params['password'] = 'mypwd1'
139        self.app['mandates'].addMandate(mandate)
140        msg = mandate.execute()
141        # Password has been set.
142        self.assertEqual(msg, 'Password has been successfully set. Proceed to '
143            'the login page and enter your credentials.')
144        self.assertTrue(IUserAccount(applicant).checkPassword('mypwd1'))
145        logfile = os.path.join(
146            self.app['datacenter'].storage, 'logs', 'main.log')
147        logcontent = open(logfile).read()
148        self.assertTrue('system - PasswordMandate used: abc' in logcontent)
149
150    def test_remove_expired(self):
151        # mandate1 is an old mandate which just expired.
152        mandate1 = PasswordMandate(days=0)
153        self.app['mandates'].addMandate(mandate1)
154        # mandate2 is a new mandate with default time delta.
155        mandate2 = PasswordMandate(mandate_id='23456')
156        self.app['mandates'].addMandate(mandate2)
157        self.assertEqual(len(self.app['mandates'].keys()), 2)
158        num_deleted = self.app['mandates'].removeExpired()
159        self.assertEqual(num_deleted, 1)
160        # Only the new mandate remains in the container.
161        self.assertEqual(len(self.app['mandates'].keys()), 1)
162        self.assertEqual([i for i in self.app['mandates'].keys()], [u'23456'])
163
164    def test_browser(self):
165        student = createObject('waeup.Student')
166        self.app['students'].addStudent(student)
167        mandate = PasswordMandate()
168        mandate.params['user'] = student
169        mandate.params['password'] = 'mypwd1'
170        self.app['mandates'].addMandate(mandate)
171        self.browser.open('http://localhost/app/mandate?mandate_id=%s'
172            % mandate.mandate_id)
173        # Password has been set.
174        self.assertTrue('Password has been successfully set. Proceed to '
175                        'the login page and enter your credentials.'
176            in self.browser.contents)
177        self.assertTrue(IUserAccount(student).checkPassword('mypwd1'))
178        # All mandates have been removed.
179        self.assertEqual(len(self.app['mandates'].keys()), 0)
Note: See TracBrowser for help on using the repository browser.