source: main/waeup.kofa/trunk/src/waeup/kofa/hostels/tests.py @ 17356

Last change on this file since 17356 was 17313, checked in by Henrik Bettermann, 22 months ago

Allow beds to be blocked so that no student can be allocated to such a bed space (in contrast to reserved beds)
  • Property svn:keywords set to Id
File size: 33.8 KB
Line 
1## $Id: tests.py 17313 2023-01-25 08:52:51Z henrik $
2##
3## Copyright (C) 2011 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 hostels and their UI components.
20"""
21import os
22import shutil
23import tempfile
24import grok
25import pytz
26from datetime import datetime, timedelta
27from zope.event import notify
28from zope.interface.verify import verifyClass, verifyObject
29from zope.component.hooks import setSite, clearSite
30from zope.testbrowser.testing import Browser
31from zope.security.interfaces import Unauthorized
32from zope.catalog.interfaces import ICatalog
33from zope.component import queryUtility, getUtility, createObject
34from waeup.kofa.app import University
35from waeup.kofa.hostels.interfaces import (
36    IHostelsContainer, IHostel, IBed, IHostelsUtils)
37from waeup.kofa.hostels.vocabularies import NOT_OCCUPIED
38from waeup.kofa.hostels.container import HostelsContainer
39from waeup.kofa.hostels.hostel import Hostel, Bed
40from waeup.kofa.hostels.batching import HostelProcessor, BedProcessor
41from waeup.kofa.hostels.export import BedExporter, HostelExporter
42from waeup.kofa.testing import (FunctionalLayer, FunctionalTestCase)
43from waeup.kofa.students.student import Student
44from waeup.kofa.students.accommodation import BedTicket
45from waeup.kofa.university.department import Department
46
47HOSTEL_SAMPLE_DATA = open(
48    os.path.join(os.path.dirname(__file__), 'sample_hostel_data.csv'),
49    'rb').read()
50
51HOSTEL_HEADER_FIELDS = HOSTEL_SAMPLE_DATA.split(
52    '\n')[0].split(',')
53
54BED_SAMPLE_DATA = open(
55    os.path.join(os.path.dirname(__file__), 'sample_bed_data.csv'),
56    'rb').read()
57
58BED_SAMPLE_DATA_2 = open(
59    os.path.join(os.path.dirname(__file__), 'sample_bed_data_2.csv'),
60    'rb').read()
61
62BED_HEADER_FIELDS = BED_SAMPLE_DATA.split(
63    '\n')[0].split(',')
64
65class HostelsContainerTestCase(FunctionalTestCase):
66
67    layer = FunctionalLayer
68
69    def test_interfaces(self):
70        # Make sure the correct interfaces are implemented.
71        self.assertTrue(
72            verifyClass(
73                IHostelsContainer, HostelsContainer)
74            )
75        self.assertTrue(
76            verifyObject(
77                IHostelsContainer, HostelsContainer())
78            )
79        self.assertTrue(
80            verifyClass(
81                IHostel, Hostel)
82            )
83        self.assertTrue(
84            verifyObject(
85                IHostel, Hostel())
86            )
87        self.assertTrue(
88            verifyClass(
89                IBed, Bed)
90            )
91        bed = Bed()
92        bed.bed_id = u'a_b_c_d'
93        bed.bed_type = u'a_b_c'
94        self.assertTrue(
95            verifyObject(
96                IBed, bed)
97            )
98        return
99
100    def test_base(self):
101        # We cannot call the fundamental methods of a base in that case
102        container = HostelsContainer()
103        hostel = Hostel()
104        # We cannot add arbitrary objects
105        department = Department()
106        self.assertRaises(
107            TypeError, container.addHostel, department)
108        self.assertRaises(
109            TypeError, hostel.addBed, department)
110        # Application is expired if startdate or enddate are not set
111        # or current datetime is outside application period.
112        self.assertTrue(container.expired)
113        delta = timedelta(days=10)
114        container.startdate = datetime.now(pytz.utc) - delta
115        self.assertTrue(container.expired)
116        container.enddate = datetime.now(pytz.utc) + delta
117        self.assertFalse(container.expired)
118
119class HostelsFullSetup(FunctionalTestCase):
120
121    def setUp(self):
122        super(HostelsFullSetup, self).setUp()
123
124        # Setup a sample site for each test
125        app = University()
126        self.dc_root = tempfile.mkdtemp()
127        app['datacenter'].setStoragePath(self.dc_root)
128
129        # Prepopulate the ZODB...
130        self.getRootFolder()['app'] = app
131        # we add the site immediately after creation to the
132        # ZODB. Catalogs and other local utilities are not setup
133        # before that step.
134        self.app = self.getRootFolder()['app']
135        # Set site here. Some of the following setup code might need
136        # to access grok.getSite() and should get our new app then
137        setSite(app)
138
139        # Add student with subobjects
140        student = Student()
141        student.firstname = u'Anna'
142        student.lastname = u'Tester'
143        student.reg_number = u'123'
144        student.matric_number = u'234'
145        student.sex = u'f'
146        self.app['students'].addStudent(student)
147        self.student_id = student.student_id
148        self.student = self.app['students'][self.student_id]
149        self.student['studycourse'].current_session = 2004
150        self.student['studycourse'].entry_session = 2004
151
152        # The students_catalog must be informed that the
153        # session attribute has changed
154        notify(grok.ObjectModifiedEvent(self.student))
155
156        # Set accommodation_session
157        self.app['hostels'].accommodation_session = 2004
158
159        # Create a hostel
160        hostel = Hostel()
161        hostel.hostel_id = u'hall-x'
162        self.app['hostels'][hostel.hostel_id] = hostel
163
164        # Create a bed
165        bed = Bed()
166        bed.bed_id = u'hall_block_room_bed'
167        bed.bed_number = 1
168        bed.bed_type = u'regular_male_fr'
169        self.app['hostels'][hostel.hostel_id][bed.bed_id] = bed
170
171        self.container_path = 'http://localhost/app/hostels'
172        self.student_path = 'http://localhost/app/students/%s' % self.student_id
173        self.manage_container_path = self.container_path + '/@@manage'
174        self.add_hostel_path = self.container_path + '/addhostel'
175
176        # Put the prepopulated site into test ZODB and prepare test
177        # browser
178        self.browser = Browser()
179        self.browser.handleErrors = False
180
181        self.logfile = os.path.join(
182            self.app['datacenter'].storage, 'logs', 'hostels.log')
183
184    def tearDown(self):
185        super(HostelsFullSetup, self).tearDown()
186        clearSite()
187        shutil.rmtree(self.dc_root)
188
189class HostelsContainerTests(HostelsFullSetup):
190
191    layer = FunctionalLayer
192
193    def test_release_expired_allocations(self):
194        self.app['hostels'].allocation_expiration = 7
195        self.browser.addHeader('Authorization', 'Basic mgr:mgrpw')
196        cat = queryUtility(ICatalog, name='beds_catalog')
197        bedticket = BedTicket()
198        bedticket.booking_session = 2004
199        bedticket.bed_coordinates = u'anything'
200        self.student['accommodation'].addBedTicket(bedticket)
201        self.app[
202            'hostels']['hall-x']['hall_block_room_bed'].owner = self.student_id
203        notify(grok.ObjectModifiedEvent(
204            self.app['hostels']['hall-x']['hall_block_room_bed']))
205        self.assertEqual(
206            self.app['hostels']['hall-x']['hall_block_room_bed'].bed_type,
207            'regular_male_fr')
208        results = cat.searchResults(owner=(self.student_id, self.student_id))
209        self.assertEqual(len(results), 1)
210        self.browser.open(self.container_path + '/releaseexpired')
211        self.assertTrue('Portal must be in maintenance mode for releasing expired bed allocations'
212            in self.browser.contents)
213        grok.getSite()['configuration'].maintmode_enabled_by = u'any_id'
214        self.browser.open(self.container_path + '/releaseexpired')
215        self.assertTrue('No bed released' in self.browser.contents)
216        delta = timedelta(days=10)
217        bedticket.booking_date = datetime.utcnow() - delta
218        self.browser.open(self.container_path + '/releaseexpired')
219        self.assertTrue(
220            'Successfully released beds: hall_block_room_bed (K1000000)'
221            in self.browser.contents)
222        results = cat.searchResults(owner=(self.student_id, self.student_id))
223        self.assertEqual(len(results), 0)
224        self.assertMatches(bedticket.display_coordinates,
225            '-- booking expired (2015-10-14 08:35:38 UTC) --')
226        # The owner has been removed and the bed reserved.
227        self.assertEqual(
228            self.app['hostels']['hall-x']['hall_block_room_bed'].owner,
229            NOT_OCCUPIED)
230        self.assertEqual(
231            self.app['hostels']['hall-x']['hall_block_room_bed'].bed_type,
232            'regular_male_reserved')
233        # Accommodation session can't be changed if hostels are not empty.
234        self.browser.open(self.manage_container_path)
235        self.browser.getControl(name="form.accommodation_session").value = ['2005']
236        self.browser.getControl("Save").click()
237        self.assertTrue(
238            'You can\'t change the booking session before clearing all hostels'
239            in self.browser.contents)
240        # Releasing is logged.
241        logcontent = open(self.logfile).read()
242        self.assertTrue(
243            'hostels.browser.ReleaseExpiredAllocationsPage - hostels - '
244            'released: hall_block_room_bed (K1000000)'
245            in logcontent)
246        return
247
248    def test_bed_statistics(self):
249        utils = getUtility(IHostelsUtils)
250        self.app['hostels']['hall-x'][
251            'hall_block_room_bed'].owner = NOT_OCCUPIED
252        notify(grok.ObjectModifiedEvent(
253            self.app['hostels']['hall-x']['hall_block_room_bed']))
254        stats = utils.getBedStatistics()
255        self.assertEqual(stats,
256            {'regular_male_fr': (0, 1, 1),
257             'regular_female_fi': (0, 0, 0),
258             'regular_male_re': (0, 0, 0),
259             'regular_female_fr': (0, 0, 0),
260             'regular_female_all': (0, 0, 0),
261             'regular_female_re': (0, 0, 0),
262             'regular_female_reserved': (0, 0, 0),
263             'regular_male_reserved': (0, 0, 0),
264             'regular_male_fi': (0, 0, 0),
265             'regular_male_all': (0, 0, 0)}
266             )
267        self.app[
268            'hostels']['hall-x']['hall_block_room_bed'].owner = self.student_id
269        notify(grok.ObjectModifiedEvent(
270            self.app['hostels']['hall-x']['hall_block_room_bed']))
271        stats = utils.getBedStatistics()
272        self.assertEqual(stats,
273            {'regular_male_fr': (1, 0, 1),
274             'regular_female_fi': (0, 0, 0),
275             'regular_male_re': (0, 0, 0),
276             'regular_female_fr': (0, 0, 0),
277             'regular_female_all': (0, 0, 0),
278             'regular_female_re': (0, 0, 0),
279             'regular_female_reserved': (0, 0, 0),
280             'regular_male_reserved': (0, 0, 0),
281             'regular_male_fi': (0, 0, 0),
282             'regular_male_all': (0, 0, 0)}
283             )
284
285class BedCatalogTests(HostelsFullSetup):
286
287    layer = FunctionalLayer
288
289    def test_get_catalog(self):
290        # We can get a beds catalog if we wish
291        cat = queryUtility(ICatalog, name='beds_catalog')
292        assert cat is not None
293
294    def test_search_by_type(self):
295        # We can find a certain bed
296        cat = queryUtility(ICatalog, name='beds_catalog')
297        results = cat.searchResults(
298            bed_type=(u'regular_male_fr', u'regular_male_fr'))
299        results = [x for x in results] # Turn results generator into list
300        assert len(results) == 1
301        assert results[0] is self.app['hostels']['hall-x']['hall_block_room_bed']
302
303    def test_search_by_owner(self):
304        # We can find a certain bed
305        myobj = self.app['hostels']['hall-x']['hall_block_room_bed']
306        myobj.owner = u'abc'
307        notify(grok.ObjectModifiedEvent(myobj))
308        cat = queryUtility(ICatalog, name='beds_catalog')
309        results = cat.searchResults(owner=(u'abc', u'abc'))
310        results = [x for x in results] # Turn results generator into list
311        assert len(results) == 1
312        assert results[0] is self.app['hostels']['hall-x']['hall_block_room_bed']
313
314class HostelsUITests(HostelsFullSetup):
315
316    layer = FunctionalLayer
317
318    def test_anonymous_access(self):
319        # Anonymous users can't access hostels containers
320        self.assertRaises(
321            Unauthorized, self.browser.open, self.manage_container_path)
322        return
323
324    def test_add_search_edit_delete_manage_hostels(self):
325        self.browser.addHeader('Authorization', 'Basic mgr:mgrpw')
326        self.browser.open(self.container_path)
327        self.browser.getLink("Manage accommodation").click()
328        self.assertEqual(self.browser.headers['Status'], '200 Ok')
329        self.assertEqual(self.browser.url, self.manage_container_path)
330        self.browser.getControl("Add hostel").click()
331        self.assertEqual(self.browser.headers['Status'], '200 Ok')
332        self.assertEqual(self.browser.url, self.add_hostel_path)
333        self.browser.getControl("Create hostel").click()
334        self.assertEqual(self.browser.headers['Status'], '200 Ok')
335        self.assertTrue('Hostel created' in self.browser.contents)
336        self.browser.open(self.container_path + '/addhostel')
337        self.browser.getControl("Create hostel").click()
338        self.assertTrue('The hostel already exists' in self.browser.contents)
339        hall = self.app['hostels']['hall-1']
340        hall.blocks_for_female = ['A','B']
341        self.browser.open(self.container_path + '/hall-1')
342        expected = '...<ul id="form.blocks_for_female" ><li>Block A</li>...'
343        self.assertMatches(expected,self.browser.contents)
344        self.browser.open(self.container_path + '/hall-1/manage')
345        self.browser.getControl(name="form.rooms_per_floor").value = '1'
346        self.browser.getControl("Save").click()
347        self.assertTrue('Form has been saved' in self.browser.contents)
348        # Since the testbrowser does not support Javascrip the
349        # save action cleared the settings above and we have to set them
350        # again.
351        self.assertTrue(len(hall.blocks_for_female) == 0)
352        hall.blocks_for_female = ['A','B']
353        hall.beds_for_fresh = ['A']
354        hall.beds_for_returning = ['B']
355        hall.beds_for_final = ['C']
356        hall.beds_for_all = ['D','E']
357        self.browser.getControl("Update all beds").click()
358        self.assertTrue('Portal must be in maintenance mode for bed updates'
359            in self.browser.contents)
360        grok.getSite()['configuration'].maintmode_enabled_by = u'any_id'
361        self.browser.getControl("Update all beds").click()
362        expected = '...0 empty beds removed, 10 beds added, 0 occupied beds modified ()...'
363        self.assertMatches(expected,self.browser.contents)
364        cat = queryUtility(ICatalog, name='beds_catalog')
365        results = cat.searchResults(
366            bed_type=('regular_female_all', 'regular_female_all'))
367        results = [(x.bed_id, x.bed_type) for x in results]
368        self.assertEqual(results,
369            [(u'hall-1_A_101_D', u'regular_female_all'),
370             (u'hall-1_A_101_E', u'regular_female_all'),
371             (u'hall-1_B_101_D', u'regular_female_all'),
372             (u'hall-1_B_101_E', u'regular_female_all')])
373        # Reserve beds.
374        self.browser.getControl("Switch reservation", index=0).click()
375        self.assertTrue('No item selected' in self.browser.contents)
376        ctrl = self.browser.getControl(name='val_id')
377        ctrl.getControl(value='hall-1_A_101_A').selected = True
378        ctrl.getControl(value='hall-1_A_101_B').selected = True
379        ctrl.getControl(value='hall-1_A_101_C').selected = True
380        ctrl.getControl(value='hall-1_A_101_D').selected = True
381        self.browser.getControl("Switch reservation", index=0).click()
382        self.assertTrue('Successfully switched beds: hall-1_A_101_A (reserved)'
383            in self.browser.contents)
384        self.assertEqual(self.app['hostels']['hall-1'][
385            'hall-1_A_101_D'].bed_type, 'regular_female_reserved')
386        # The catalog has been updated.
387        results = cat.searchResults(
388            bed_type=('regular_female_all', 'regular_female_all'))
389        results = [(x.bed_id, x.bed_type) for x in results]
390        self.assertEqual(results,
391            [(u'hall-1_A_101_E', u'regular_female_all'),
392             (u'hall-1_B_101_D', u'regular_female_all'),
393             (u'hall-1_B_101_E', u'regular_female_all')])
394        results = cat.searchResults(
395            bed_type=('regular_female_reserved', 'regular_female_reserved'))
396        results = [(x.bed_id, x.bed_type) for x in results]
397        self.assertEqual(results,
398            [(u'hall-1_A_101_A', u'regular_female_reserved'),
399             (u'hall-1_A_101_B', u'regular_female_reserved'),
400             (u'hall-1_A_101_C', u'regular_female_reserved'),
401             (u'hall-1_A_101_D', u'regular_female_reserved')])
402        # Block beds.
403        self.browser.getControl("Switch blockade", index=0).click()
404        self.assertTrue('No item selected' in self.browser.contents)
405        ctrl = self.browser.getControl(name='val_id')
406        ctrl.getControl(value='hall-1_A_101_A').selected = True
407        ctrl.getControl(value='hall-1_A_101_B').selected = True
408        ctrl.getControl(value='hall-1_A_101_C').selected = True
409        ctrl.getControl(value='hall-1_A_101_D').selected = True
410        self.browser.getControl("Switch blockade", index=0).click()
411        self.assertTrue('Successfully switched beds: hall-1_A_101_A (blocked)'
412            in self.browser.contents)
413        self.assertEqual(self.app['hostels']['hall-1'][
414            'hall-1_A_101_D'].bed_type, 'regular_female_blocked')
415        # The catalog has been updated.
416        results = cat.searchResults(
417            bed_type=('regular_female_all', 'regular_female_all'))
418        results = [(x.bed_id, x.bed_type) for x in results]
419        self.assertEqual(results,
420            [(u'hall-1_A_101_E', u'regular_female_all'),
421             (u'hall-1_B_101_D', u'regular_female_all'),
422             (u'hall-1_B_101_E', u'regular_female_all')])
423        results = cat.searchResults(
424            bed_type=('regular_female_blocked', 'regular_female_blocked'))
425        results = [(x.bed_id, x.bed_type) for x in results]
426        self.assertEqual(results,
427            [(u'hall-1_A_101_A', u'regular_female_blocked'),
428             (u'hall-1_A_101_B', u'regular_female_blocked'),
429             (u'hall-1_A_101_C', u'regular_female_blocked'),
430             (u'hall-1_A_101_D', u'regular_female_blocked')])
431        # Change hostel configuration with one bed booked.
432        hall['hall-1_A_101_E'].owner = u'anyid'
433        notify(grok.ObjectModifiedEvent(hall['hall-1_A_101_E']))
434        hall.beds_for_fresh = ['A', 'E']
435        hall.beds_for_all = ['D']
436        self.browser.getControl("Update all beds").click()
437        expected = '...9 empty beds removed, 9 beds added, 1 occupied beds modified...'
438        self.assertMatches(expected,self.browser.contents)
439        # Updating beds (including booked beds!) does update catalog.
440        results = cat.searchResults(
441            bed_type=('regular_female_all', 'regular_female_all'))
442        results = [(x.bed_id, x.bed_type) for x in results]
443        # The reservation of hall-1_A_101_D has been cancelled.
444        self.assertEqual(results,
445            [(u'hall-1_A_101_D', u'regular_female_all'),
446             (u'hall-1_B_101_D', u'regular_female_all')])
447        # Release bed which has previously been booked.
448        bedticket = BedTicket()
449        bedticket.booking_session = 2004
450        bedticket.bed_coordinates = u'anything'
451        self.student['accommodation'].addBedTicket(bedticket)
452        self.app['hostels']['hall-1']['hall-1_A_101_D'].owner = self.student_id
453        notify(grok.ObjectModifiedEvent(self.app['hostels']['hall-1']['hall-1_A_101_D']))
454        self.browser.open(self.container_path + '/hall-1/manage')
455        ctrl = self.browser.getControl(name='val_id')
456        self.browser.getControl("Release selected beds", index=0).click()
457        self.assertMatches("...No item selected...", self.browser.contents)
458        ctrl = self.browser.getControl(name='val_id')
459        ctrl.getControl(value='hall-1_A_101_D').selected = True
460        self.browser.getControl("Release selected beds", index=0).click()
461        self.assertMatches(
462          '...Successfully released beds: hall-1_A_101_D (%s)...' % self.student_id,
463          self.browser.contents)
464        self.assertMatches(bedticket.bed_coordinates,
465          u' -- booking cancelled on <YYYY-MM-DD hh:mm:ss> UTC --')
466        # The catalog has been updated.
467        results = cat.searchResults(owner=(self.student_id, self.student_id))
468        assert len(results) == 0
469        # If we release a free bed, nothing will happen.
470        ctrl = self.browser.getControl(name='val_id')
471        ctrl.getControl(value='hall-1_A_101_D').selected = True
472        self.browser.getControl("Release selected beds", index=0).click()
473        self.assertMatches(
474          '...No allocated bed selected...', self.browser.contents)
475        # Managers can manually allocate eligible students after cancellation.
476        self.browser.open(self.container_path + '/hall-1/hall-1_A_101_A')
477        # 'not occupied' is not accepted.
478        self.browser.getControl("Save").click()
479        self.assertMatches(
480            "...No valid student id...",
481            self.browser.contents)
482        # Invalid student ids are not accepted.
483        self.browser.getControl(name="form.owner").value = 'nonsense'
484        self.browser.getControl("Save").click()
485        self.assertMatches(
486            "...Either student does not exist or student "
487            "is not eligible to book accommodation...",
488            self.browser.contents)
489        self.browser.getControl(name="form.owner").value = self.student_id
490        self.browser.getControl("Save").click()
491        self.assertMatches("...Form has been saved...", self.browser.contents)
492        # Students can only be allocated once.
493        self.browser.open(self.container_path + '/hall-1/hall-1_A_101_B')
494        self.browser.getControl(name="form.owner").value = self.student_id
495        self.browser.getControl("Save").click()
496        self.assertMatches(
497            "...This student resides in bed hall-1_A_101_A...",
498            self.browser.contents)
499        # If we open the same form again, we will be redirected to hostel
500        # manage page. Beds must be released first before they can be
501        # allocated to other students.
502        self.browser.open(self.container_path + '/hall-1/hall-1_A_101_A')
503        self.assertEqual(self.browser.url,
504            self.container_path + '/hall-1/@@manage#tab2')
505        # Updating the beds again will not affect the allocation and also
506        # the bed numbering remains the same.
507        old_number = self.app['hostels']['hall-1']['hall-1_A_101_A'].bed_number
508        old_owner = self.app['hostels']['hall-1']['hall-1_A_101_A'].owner
509        self.browser.getControl("Update all beds").click()
510        # 8 beds have been removed and re-added, 2 beds remains untouched
511        # because they are occupied.
512        expected = '...8 empty beds removed, 8 beds added, 0 occupied beds modified...'
513        self.assertMatches(expected,self.browser.contents)
514        new_number = self.app['hostels']['hall-1']['hall-1_A_101_A'].bed_number
515        new_owner = self.app['hostels']['hall-1']['hall-1_A_101_A'].owner
516        self.assertEqual(new_number, old_number)
517        self.assertEqual(new_owner, old_owner)
518        # If we change the bed type of an allocated bed, the modification will
519        # be indicated.
520        hall.blocks_for_female = ['B']
521        hall.blocks_for_male = ['A']
522        self.browser.getControl("Update all beds").click()
523        expected = '...8 empty beds removed, 8 beds added, ' + \
524            '2 occupied beds modified (hall-1_A_101_A, hall-1_A_101_E, )...'
525        self.assertMatches(expected,self.browser.contents)
526        new_number = self.app['hostels']['hall-1']['hall-1_A_101_A'].bed_number
527        # Also the number of the bed has changed.
528        self.assertFalse(new_number == old_number)
529        # The number of occupied beds are displayed on container page.
530        self.browser.open(self.container_path)
531        self.assertTrue('2 of 10' in self.browser.contents)
532        bedticket.bed = self.app['hostels']['hall-1']['hall-1_A_101_A']
533        # Managers can open the bed statistics page
534        self.browser.getLink("Bed statistics").click()
535        self.assertTrue('Bed Statistics</h1>' in self.browser.contents)
536        # Remove entire hostel.
537        self.browser.open(self.manage_container_path)
538        ctrl = self.browser.getControl(name='val_id')
539        value = ctrl.options[0]
540        ctrl.getControl(value=value).selected = True
541        self.browser.getControl("Remove selected", index=0).click()
542        self.assertTrue('Successfully removed' in self.browser.contents)
543        # Catalog is empty.
544        results = cat.searchResults(
545            bed_type=('regular_female_all', 'regular_female_all'))
546        results = [x for x in results]
547        assert len(results) == 0
548        # Bed has been removed from bedticket
549        self.assertEqual(bedticket.bed, None)
550        # Actions are logged.
551        logcontent = open(self.logfile).read()
552        self.assertTrue(
553            'hall-1 - 9 empty beds removed, 9 beds added, 1 occupied '
554            'beds modified (hall-1_A_101_E, )'
555            in logcontent)
556
557    def test_clear_hostels(self):
558        grok.getSite()['configuration'].maintmode_enabled_by = u'any_id'
559        self.browser.addHeader('Authorization', 'Basic mgr:mgrpw')
560        self.browser.open(self.container_path)
561        self.browser.getLink("Manage accommodation").click()
562        self.browser.getControl("Add hostel").click()
563        self.browser.getControl("Create hostel").click()
564        hall = self.app['hostels']['hall-1']
565        hall.blocks_for_female = ['A','B']
566        hall.rooms_per_floor = 1
567        hall.beds_for_fresh = ['A']
568        hall.beds_for_returning = ['B']
569        hall.beds_for_final = ['C']
570        hall.beds_for_all = ['D','E']
571        self.browser.open(self.container_path + '/hall-1/manage')
572        self.browser.getControl("Update all beds").click()
573        cat = queryUtility(ICatalog, name='beds_catalog')
574        results = cat.searchResults(bed_type=(None, None))
575        self.assertEqual(len(results), 11)
576        self.browser.getControl("Clear hostel").click()
577        self.assertEqual(len(self.app['hostels']['hall-1']), 0)
578        # Only the bed in hall-x remains in the catalog.
579        results = cat.searchResults(bed_type=(None, None))
580        self.assertEqual(len(results), 1)
581        # We can clear all hostels at the same time.
582        self.browser.open(self.manage_container_path)
583        self.browser.getControl("Clear all hostels").click()
584        results = cat.searchResults(bed_type=(None, None))
585        self.assertEqual(len(results), 0)
586        # Both actions have been logged.
587        logcontent = open(self.logfile).read()
588        self.assertTrue('INFO - zope.mgr - hostels.browser.HostelManageFormPage'
589                        ' - hall-1 - cleared' in logcontent)
590        self.assertTrue('zope.mgr - hostels.browser.HostelsContainerManagePage'
591                        ' - hostels - all hostels cleared' in logcontent)
592
593class ExportTests(HostelsFullSetup):
594
595    layer = FunctionalLayer
596
597    def setUp(self):
598        super(ExportTests, self).setUp()
599        self.workdir = tempfile.mkdtemp()
600        self.outfile = os.path.join(self.workdir, 'myoutput.csv')
601        return
602
603    def test_export_hostels(self):
604        exporter = HostelExporter()
605        exporter.export_all(self.app, self.outfile)
606        result = open(self.outfile, 'rb').read()
607        self.assertEqual(
608            result,
609            'beds_for_all,beds_for_final,beds_for_fresh,beds_for_pre,'
610            'beds_for_returning,blocks_for_female,'
611            'blocks_for_male,floors_per_block,hostel_id,hostel_name,maint_fee,'
612            'rooms_per_floor,sort_id,special_handling\r\n[],[],[],[],[],[],[],'
613            '1,hall-x,Hall 1,0.0,2,10,regular\r\n'
614            )
615        return
616
617    def test_export_beds(self):
618        exporter = BedExporter()
619        exporter.export_all(self.app, self.outfile)
620        result = open(self.outfile, 'rb').read()
621        self.assertEqual(
622            result,
623            'bed_id,bed_number,bed_type,owner,hall,block,room,bed,'
624            'special_handling,sex,bt\r\nhall_block_room_bed,1,regular_male_fr,,'
625            'hall,block,room,bed,regular,male,fr\r\n'
626            )
627        return
628
629    def tearDown(self):
630        super(ExportTests, self).tearDown()
631        clearSite()
632        shutil.rmtree(os.path.dirname(self.outfile))
633
634class HostelBedProcessorTest(HostelsFullSetup):
635
636    layer = FunctionalLayer
637
638    def test_import(self):
639        self.processor = HostelProcessor()
640        self.workdir = tempfile.mkdtemp()
641        self.csv_file = os.path.join(self.workdir, 'sample_hostel_data.csv')
642        open(self.csv_file, 'wb').write(HOSTEL_SAMPLE_DATA)
643        num, num_warns, fin_file, fail_file = self.processor.doImport(
644            self.csv_file, HOSTEL_HEADER_FIELDS)
645        self.assertEqual(num_warns,0)
646        self.assertEqual(len(self.app['hostels'].keys()), 11) # including hall-x
647        self.assertEqual(self.app['hostels'][
648            'block-a-upper-hostel'].hostel_id,'block-a-upper-hostel')
649        self.assertEqual(self.app['hostels'][
650            'block-a-upper-hostel'].beds_for_final, ['A', 'B'])
651        logcontent = open(self.logfile).read()
652        self.assertTrue(
653            "Hostel Processor - sample_hostel_data - block-a-upper-hostel - "
654            "updated: "
655            "beds_for_pre=['G'], floors_per_block=1, "
656            "special_handling=regular, "
657            "beds_for_final=['A', 'B'], rooms_per_floor=32, "
658            "hostel_id=block-a-upper-hostel, "
659            "sort_id=20, beds_for_returning=['C', 'D'], "
660            "hostel_name=Block A Upper Hostel, beds_for_fresh=['E', 'F'], "
661            "blocks_for_female=['A']"
662            in logcontent)
663        shutil.rmtree(os.path.dirname(fin_file))
664        shutil.rmtree(self.workdir)
665        # The manage page can be opened.
666        self.browser.addHeader('Authorization', 'Basic mgr:mgrpw')
667        self.browser.open(self.container_path + '/block-a-upper-hostel')
668        self.browser.getLink("Manage").click()
669        self.assertEqual(self.browser.headers['Status'], '200 Ok')
670        return
671
672    def test_import_update_hostel_and_beds(self):
673        self.processor = HostelProcessor()
674        self.workdir = tempfile.mkdtemp()
675        self.csv_file = os.path.join(self.workdir, 'sample_hostel_data.csv')
676        open(self.csv_file, 'wb').write(HOSTEL_SAMPLE_DATA)
677        num, num_warns, fin_file, fail_file = self.processor.doImport(
678            self.csv_file, HOSTEL_HEADER_FIELDS)
679        # We import the same file in update mode
680        num, num_warns, fin_file, fail_file = self.processor.doImport(
681            self.csv_file, HOSTEL_HEADER_FIELDS, 'update')
682        self.assertEqual(num_warns,0)
683        logcontent = open(self.logfile).read()
684        self.assertTrue(
685            "Hostel Processor - sample_hostel_data - block-a-upper-hostel - "
686            "updated: "
687            "beds_for_pre=['G'], floors_per_block=1, "
688            "special_handling=regular, "
689            "beds_for_final=['A', 'B'], rooms_per_floor=32, "
690            "hostel_id=block-a-upper-hostel, "
691            "sort_id=20, beds_for_returning=['C', 'D'], "
692            "hostel_name=Block A Upper Hostel, beds_for_fresh=['E', 'F'], "
693            "blocks_for_female=['A']"
694            in logcontent)
695        # Update the beds of first hostel
696        self.app['hostels']['block-a-upper-hostel'].updateBeds()
697        # Import beds
698        self.processor = BedProcessor()
699        self.csv_file = os.path.join(self.workdir, 'sample_bed_data.csv')
700        open(self.csv_file, 'wb').write(BED_SAMPLE_DATA)
701        num, num_warns, fin_file, fail_file = self.processor.doImport(
702            self.csv_file, BED_HEADER_FIELDS, 'update')
703        self.assertEqual(num_warns,2)
704        fail_file_content = open(fail_file).read()
705        logcontent = open(self.logfile).read()
706        self.assertEqual(
707            'owner,bed_id,hostel_id,reserved,blocked,--ERRORS--\r\n'
708            '<IGNORE>,block-a-upper-hostel_A_101_C,block-a-upper-hostel,2,0,'
709            'reserved: invalid value\r\n'
710            'K999,block-a-upper-hostel_A_101_E,block-a-upper-hostel,0,0,'
711            'owner: student does not exist\r\n',
712            fail_file_content)
713        self.assertTrue(
714            'system - Bed Processor (update only) - sample_bed_data - '
715            'block-a-upper-hostel_A_101_A - updated: blocked'
716            in logcontent)
717        self.assertTrue(
718            'system - Bed Processor (update only) - sample_bed_data - '
719            'block-a-upper-hostel_A_101_B - updated: reserved'
720            in logcontent)
721        self.assertTrue(
722            'system - Bed Processor (update only) - sample_bed_data - '
723            'block-a-upper-hostel_A_101_D - updated: owner=K1000000'
724            in logcontent)
725        self.assertEqual(
726            self.app['hostels']['block-a-upper-hostel']
727                    ['block-a-upper-hostel_A_101_D'].owner, 'K1000000')
728        self.assertEqual(
729            self.app['hostels']['block-a-upper-hostel']
730                    ['block-a-upper-hostel_A_101_B'].bed_type,
731                    'regular_female_reserved')
732        # Add another student K1000001
733        student = createObject('waeup.Student')
734        student.firstname = u'Claus'
735        student.lastname = u'Tester'
736        self.app['students'].addStudent(student)
737        # Try to allocate this student to a blocked bed
738
739        open(self.csv_file, 'wb').write(BED_SAMPLE_DATA_2)
740        num, num_warns, fin_file, fail_file = self.processor.doImport(
741            self.csv_file, BED_HEADER_FIELDS, 'update')
742        fail_file_content = open(fail_file).read()
743        self.assertEqual(
744            'owner,bed_id,hostel_id,reserved,blocked,--ERRORS--\r\n'
745            'K1000001,block-a-upper-hostel_A_101_A,block-a-upper-hostel,'
746            '<IGNORE>,<IGNORE>,Bed is blocked.\r\n',
747            fail_file_content)
748        shutil.rmtree(os.path.dirname(fin_file))
749        shutil.rmtree(self.workdir)
750        return
Note: See TracBrowser for help on using the repository browser.