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

Last change on this file since 13288 was 13276, checked in by Henrik Bettermann, 9 years ago

Fix some typos.

Adjust test so that it can be processed by Sphinx.
  • Property svn:keywords set to Id
File size: 24.2 KB
RevLine 
[7195]1## $Id: tests.py 13276 2015-09-23 05:56:41Z henrik $
2##
[6951]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.
[7195]8##
[6951]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.
[7195]13##
[6951]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"""
[6988]19Tests for hostels and their UI components.
[6951]20"""
[9197]21import os
[6961]22import shutil
23import tempfile
[7003]24import grok
[8686]25import pytz
26from datetime import datetime, timedelta
[7003]27from zope.event import notify
[6951]28from zope.interface.verify import verifyClass, verifyObject
[6961]29from zope.component.hooks import setSite, clearSite
30from zope.testbrowser.testing import Browser
31from zope.security.interfaces import Unauthorized
[6972]32from zope.catalog.interfaces import ICatalog
33from zope.component import queryUtility
[7811]34from waeup.kofa.app import University
35from waeup.kofa.hostels.interfaces import (
[6963]36    IHostelsContainer, IHostel, IBed)
[7811]37from waeup.kofa.hostels.container import HostelsContainer
38from waeup.kofa.hostels.hostel import Hostel, Bed
[9202]39from waeup.kofa.hostels.batching import HostelProcessor
[9200]40from waeup.kofa.hostels.export import BedExporter, HostelExporter
[7811]41from waeup.kofa.testing import (FunctionalLayer, FunctionalTestCase)
42from waeup.kofa.students.student import Student
43from waeup.kofa.students.accommodation import BedTicket
44from waeup.kofa.university.department import Department
[6951]45
[9202]46HOSTEL_SAMPLE_DATA = open(
47    os.path.join(os.path.dirname(__file__), 'sample_hostel_data.csv'),
48    'rb').read()
49
50HOSTEL_HEADER_FIELDS = HOSTEL_SAMPLE_DATA.split(
51    '\n')[0].split(',')
52
[6951]53class HostelsContainerTestCase(FunctionalTestCase):
54
55    layer = FunctionalLayer
56
57    def test_interfaces(self):
58        # Make sure the correct interfaces are implemented.
59        self.assertTrue(
60            verifyClass(
61                IHostelsContainer, HostelsContainer)
62            )
63        self.assertTrue(
64            verifyObject(
65                IHostelsContainer, HostelsContainer())
66            )
67        self.assertTrue(
68            verifyClass(
69                IHostel, Hostel)
70            )
71        self.assertTrue(
72            verifyObject(
73                IHostel, Hostel())
74            )
[6963]75        self.assertTrue(
76            verifyClass(
77                IBed, Bed)
78            )
[13170]79        bed = Bed()
80        bed.bed_id = u'a_b_c_d'
81        bed.bed_type = u'a_b_c'
[6963]82        self.assertTrue(
83            verifyObject(
[13170]84                IBed, bed)
[6963]85            )
[6951]86        return
87
88    def test_base(self):
89        # We cannot call the fundamental methods of a base in that case
90        container = HostelsContainer()
[7077]91        hostel = Hostel()
92        # We cannot add arbitrary objects
93        department = Department()
94        self.assertRaises(
95            TypeError, container.addHostel, department)
96        self.assertRaises(
97            TypeError, hostel.addBed, department)
[8686]98        # Application is expired if startdate or enddate are not set
99        # or current datetime is outside application period.
100        self.assertTrue(container.expired)
101        delta = timedelta(days=10)
102        container.startdate = datetime.now(pytz.utc) - delta
103        self.assertTrue(container.expired)
104        container.enddate = datetime.now(pytz.utc) + delta
105        self.assertFalse(container.expired)
[6961]106
107class HostelsFullSetup(FunctionalTestCase):
108
109    def setUp(self):
110        super(HostelsFullSetup, self).setUp()
111
112        # Setup a sample site for each test
113        app = University()
114        self.dc_root = tempfile.mkdtemp()
115        app['datacenter'].setStoragePath(self.dc_root)
116
117        # Prepopulate the ZODB...
118        self.getRootFolder()['app'] = app
119        # we add the site immediately after creation to the
120        # ZODB. Catalogs and other local utilities are not setup
121        # before that step.
122        self.app = self.getRootFolder()['app']
123        # Set site here. Some of the following setup code might need
124        # to access grok.getSite() and should get our new app then
125        setSite(app)
126
[7045]127        # Add student with subobjects
128        student = Student()
[7357]129        student.firstname = u'Anna'
130        student.lastname = u'Tester'
[7045]131        student.reg_number = u'123'
132        student.matric_number = u'234'
133        student.sex = u'f'
134        self.app['students'].addStudent(student)
135        self.student_id = student.student_id
136        self.student = self.app['students'][self.student_id]
137        self.student['studycourse'].current_session = 2004
138        self.student['studycourse'].entry_session = 2004
[7068]139        # The students_catalog must be informed that the
140        # session attribute has changed
141        notify(grok.ObjectModifiedEvent(self.student))
[7045]142
143        # Set accommodation_session
[8685]144        self.app['hostels'].accommodation_session = 2004
[7045]145
[6972]146        # Create a hostel
147        hostel = Hostel()
[6973]148        hostel.hostel_id = u'hall-x'
[6972]149        self.app['hostels'][hostel.hostel_id] = hostel
150
151        # Create a bed
152        bed = Bed()
[9200]153        bed.bed_id = u'hall_block_room_bed'
[6972]154        bed.bed_number = 1
[9200]155        bed.bed_type = u'a_b_c'
[6972]156        self.app['hostels'][hostel.hostel_id][bed.bed_id] = bed
157
[6961]158        self.container_path = 'http://localhost/app/hostels'
[7068]159        self.student_path = 'http://localhost/app/students/%s' % self.student_id
[6961]160        self.manage_container_path = self.container_path + '/@@manage'
161        self.add_hostel_path = self.container_path + '/addhostel'
162
163        # Put the prepopulated site into test ZODB and prepare test
164        # browser
165        self.browser = Browser()
166        self.browser.handleErrors = False
167
[9202]168        self.logfile = os.path.join(
169            self.app['datacenter'].storage, 'logs', 'hostels.log')
170
[6961]171    def tearDown(self):
172        super(HostelsFullSetup, self).tearDown()
173        clearSite()
174        shutil.rmtree(self.dc_root)
175
[6972]176class BedCatalogTests(HostelsFullSetup):
177
178    layer = FunctionalLayer
179
180    def test_get_catalog(self):
[7045]181        # We can get a beds catalog if we wish
[6972]182        cat = queryUtility(ICatalog, name='beds_catalog')
183        assert cat is not None
184
185    def test_search_by_type(self):
186        # We can find a certain bed
187        cat = queryUtility(ICatalog, name='beds_catalog')
[9200]188        results = cat.searchResults(bed_type=(u'a_b_c', u'a_b_c'))
[6972]189        results = [x for x in results] # Turn results generator into list
190        assert len(results) == 1
[9200]191        assert results[0] is self.app['hostels']['hall-x']['hall_block_room_bed']
[6972]192
[7003]193    def test_search_by_owner(self):
194        # We can find a certain bed
[9200]195        myobj = self.app['hostels']['hall-x']['hall_block_room_bed']
[7003]196        myobj.owner = u'abc'
197        notify(grok.ObjectModifiedEvent(myobj))
198        cat = queryUtility(ICatalog, name='beds_catalog')
199        results = cat.searchResults(owner=(u'abc', u'abc'))
200        results = [x for x in results] # Turn results generator into list
201        assert len(results) == 1
[9200]202        assert results[0] is self.app['hostels']['hall-x']['hall_block_room_bed']
[7003]203
[6961]204class HostelsUITests(HostelsFullSetup):
205
206    layer = FunctionalLayer
207
208    def test_anonymous_access(self):
209        # Anonymous users can't access hostels containers
210        self.assertRaises(
211            Unauthorized, self.browser.open, self.manage_container_path)
212        return
213
[6985]214    def test_add_search_edit_delete_manage_hostels(self):
[6961]215        self.browser.addHeader('Authorization', 'Basic mgr:mgrpw')
216        self.browser.open(self.container_path)
217        self.browser.getLink("Manage accommodation").click()
218        self.assertEqual(self.browser.headers['Status'], '200 Ok')
219        self.assertEqual(self.browser.url, self.manage_container_path)
220        self.browser.getControl("Add hostel").click()
221        self.assertEqual(self.browser.headers['Status'], '200 Ok')
222        self.assertEqual(self.browser.url, self.add_hostel_path)
223        self.browser.getControl("Create hostel").click()
224        self.assertEqual(self.browser.headers['Status'], '200 Ok')
225        self.assertTrue('Hostel created' in self.browser.contents)
[6988]226        self.browser.open(self.container_path + '/addhostel')
227        self.browser.getControl("Create hostel").click()
228        self.assertTrue('The hostel already exists' in self.browser.contents)
[6973]229        hall = self.app['hostels']['hall-1']
[6961]230        hall.blocks_for_female = ['A','B']
[6973]231        self.browser.open(self.container_path + '/hall-1')
[13276]232        expected = '...<ul id="form.blocks_for_female" ><li>Block A</li>...'
[6961]233        self.assertMatches(expected,self.browser.contents)
[6973]234        self.browser.open(self.container_path + '/hall-1/manage')
[6985]235        self.browser.getControl(name="form.rooms_per_floor").value = '1'
[6962]236        self.browser.getControl("Save").click()
237        self.assertTrue('Form has been saved' in self.browser.contents)
[6985]238        # Since the testbrowser does not support Javascrip the
239        # save action cleared the settings above and we have to set them
[13176]240        # again.
[6985]241        self.assertTrue(len(hall.blocks_for_female) == 0)
242        hall.blocks_for_female = ['A','B']
[6988]243        hall.beds_for_fresh = ['A']
244        hall.beds_for_returning = ['B']
245        hall.beds_for_final = ['C']
[6985]246        hall.beds_for_all = ['D','E']
247        self.browser.getControl("Update all beds").click()
[6988]248        expected = '...0 empty beds removed, 10 beds added, 0 occupied beds modified ()...'
[6985]249        self.assertMatches(expected,self.browser.contents)
250        cat = queryUtility(ICatalog, name='beds_catalog')
251        results = cat.searchResults(
252            bed_type=('regular_female_all', 'regular_female_all'))
[9448]253        results = [(x.bed_id, x.bed_type) for x in results]
254        self.assertEqual(results,
255            [(u'hall-1_A_101_D', u'regular_female_all'),
256             (u'hall-1_A_101_E', u'regular_female_all'),
257             (u'hall-1_B_101_D', u'regular_female_all'),
258             (u'hall-1_B_101_E', u'regular_female_all')])
[13176]259        # Reserve beds.
[6988]260        self.browser.getControl("Switch reservation", index=0).click()
261        self.assertTrue('No item selected' in self.browser.contents)
[6985]262        ctrl = self.browser.getControl(name='val_id')
[6988]263        ctrl.getControl(value='hall-1_A_101_A').selected = True
264        ctrl.getControl(value='hall-1_A_101_B').selected = True
265        ctrl.getControl(value='hall-1_A_101_C').selected = True
[6985]266        ctrl.getControl(value='hall-1_A_101_D').selected = True
267        self.browser.getControl("Switch reservation", index=0).click()
[6988]268        self.assertTrue('Successfully switched beds: hall-1_A_101_A (reserved)'
[6985]269            in self.browser.contents)
[9448]270        self.assertEqual(self.app['hostels']['hall-1'][
271            'hall-1_A_101_D'].bed_type, 'regular_female_reserved')
[9534]272        self.assertTrue('A_101_A&nbsp;&nbsp;' in self.browser.contents)
[13176]273        # The catalog has been updated.
[9448]274        results = cat.searchResults(
275            bed_type=('regular_female_all', 'regular_female_all'))
276        results = [(x.bed_id, x.bed_type) for x in results]
277        self.assertEqual(results,
278            [(u'hall-1_A_101_E', u'regular_female_all'),
279             (u'hall-1_B_101_D', u'regular_female_all'),
280             (u'hall-1_B_101_E', u'regular_female_all')])
281        results = cat.searchResults(
282            bed_type=('regular_female_reserved', 'regular_female_reserved'))
283        results = [(x.bed_id, x.bed_type) for x in results]
284        self.assertEqual(results,
285            [(u'hall-1_A_101_A', u'regular_female_reserved'),
286             (u'hall-1_A_101_B', u'regular_female_reserved'),
287             (u'hall-1_A_101_C', u'regular_female_reserved'),
288             (u'hall-1_A_101_D', u'regular_female_reserved')])
[13176]289        # Change hostel configuration with one bed booked.
[9448]290        hall['hall-1_A_101_E'].owner = u'anyid'
291        notify(grok.ObjectModifiedEvent(hall['hall-1_A_101_E']))
292        hall.beds_for_fresh = ['A', 'E']
[6985]293        hall.beds_for_all = ['D']
294        self.browser.getControl("Update all beds").click()
[9448]295        expected = '...9 empty beds removed, 9 beds added, 1 occupied beds modified...'
[6985]296        self.assertMatches(expected,self.browser.contents)
[13176]297        # Updating beds (including booked beds!) does update catalog.
[6985]298        results = cat.searchResults(
299            bed_type=('regular_female_all', 'regular_female_all'))
[9448]300        results = [(x.bed_id, x.bed_type) for x in results]
301        self.assertEqual(results,
302            [(u'hall-1_B_101_D', u'regular_female_all'),])
[13176]303        # Unreserve beds.
[6988]304        ctrl = self.browser.getControl(name='val_id')
305        ctrl.getControl(value='hall-1_A_101_A').selected = True
306        ctrl.getControl(value='hall-1_A_101_B').selected = True
307        ctrl.getControl(value='hall-1_A_101_C').selected = True
308        ctrl.getControl(value='hall-1_A_101_D').selected = True
309        self.browser.getControl("Switch reservation", index=0).click()
310        assert self.app['hostels']['hall-1'][
311            'hall-1_A_101_D'].bed_type == 'regular_female_all'
312        self.assertFalse(expected in self.browser.contents)
[13176]313        # Release bed which has previously been booked.
[7045]314        bedticket = BedTicket()
[9423]315        bedticket.booking_session = 2004
[7045]316        bedticket.bed_coordinates = u'anything'
317        self.student['accommodation'].addBedTicket(bedticket)
318        self.app['hostels']['hall-1']['hall-1_A_101_D'].owner = self.student_id
319        self.browser.open(self.container_path + '/hall-1/manage')
320        ctrl = self.browser.getControl(name='val_id')
[7068]321        self.browser.getControl("Release selected beds", index=0).click()
322        self.assertMatches("...No item selected...", self.browser.contents)
323        ctrl = self.browser.getControl(name='val_id')
[7045]324        ctrl.getControl(value='hall-1_A_101_D').selected = True
325        self.browser.getControl("Release selected beds", index=0).click()
326        self.assertMatches(
327          '...Successfully released beds: hall-1_A_101_D (%s)...' % self.student_id,
328          self.browser.contents)
329        self.assertMatches(bedticket.bed_coordinates,
[8234]330          u' -- booking cancelled on <YYYY-MM-DD hh:mm:ss> UTC --')
[13176]331        # If we release a free be, nothing will happen.
[7070]332        ctrl = self.browser.getControl(name='val_id')
333        ctrl.getControl(value='hall-1_A_101_D').selected = True
334        self.browser.getControl("Release selected beds", index=0).click()
335        self.assertMatches(
336          '...No allocated bed selected...', self.browser.contents)
[13176]337        # Managers can manually allocate eligible students after cancellation.
[7068]338        self.browser.open(self.container_path + '/hall-1/hall-1_A_101_A')
[13176]339        # 'not occupied' is not accepted.
[9416]340        self.browser.getControl("Save").click()
341        self.assertMatches(
342            "...No valid student id...",
343            self.browser.contents)
[13176]344        # Invalid student ids are not accepted.
[9414]345        self.browser.getControl(name="form.owner").value = 'nonsense'
[7068]346        self.browser.getControl("Save").click()
[9414]347        self.assertMatches(
[9416]348            "...Either student does not exist or student "
349            "is not in accommodation session...",
[9414]350            self.browser.contents)
351        self.browser.getControl(name="form.owner").value = self.student_id
352        self.browser.getControl("Save").click()
[7068]353        self.assertMatches("...Form has been saved...", self.browser.contents)
[13176]354        # Students can only be allocated once.
[9414]355        self.browser.open(self.container_path + '/hall-1/hall-1_A_101_B')
356        self.browser.getControl(name="form.owner").value = self.student_id
357        self.browser.getControl("Save").click()
358        self.assertMatches(
359            "...This student resides in bed hall-1_A_101_A...",
360            self.browser.contents)
[7070]361        # If we open the same form again, we will be redirected to hostel
362        # manage page. Beds must be released first before they can be
363        # allocated to other students.
364        self.browser.open(self.container_path + '/hall-1/hall-1_A_101_A')
365        self.assertEqual(self.browser.url,
[11254]366            self.container_path + '/hall-1/@@manage#tab2')
[7070]367        # Updating the beds again will not affect the allocation and also
[13176]368        # the bed numbering remains the same.
[7070]369        old_number = self.app['hostels']['hall-1']['hall-1_A_101_A'].bed_number
370        old_owner = self.app['hostels']['hall-1']['hall-1_A_101_A'].owner
371        self.browser.getControl("Update all beds").click()
[9448]372        # 8 beds have been removed and re-added, 2 beds remains untouched
[13176]373        # because they are occupied.
[9448]374        expected = '...8 empty beds removed, 8 beds added, 0 occupied beds modified...'
[7070]375        self.assertMatches(expected,self.browser.contents)
376        new_number = self.app['hostels']['hall-1']['hall-1_A_101_A'].bed_number
377        new_owner = self.app['hostels']['hall-1']['hall-1_A_101_A'].owner
378        self.assertEqual(new_number, old_number)
379        self.assertEqual(new_owner, old_owner)
380        # If we change the bed type of an allocated bed, the modification will
[13176]381        # be indicated.
[7070]382        hall.blocks_for_female = ['B']
383        hall.blocks_for_male = ['A']
384        self.browser.getControl("Update all beds").click()
[9448]385        expected = '...8 empty beds removed, 8 beds added, ' + \
386            '2 occupied beds modified (hall-1_A_101_A, hall-1_A_101_E, )...'
[7070]387        self.assertMatches(expected,self.browser.contents)
388        new_number = self.app['hostels']['hall-1']['hall-1_A_101_A'].bed_number
389        # Also the number of the bed has changed.
390        self.assertFalse(new_number == old_number)
[9414]391        # The number of occupied beds are displayed on container page.
[9196]392        self.browser.open(self.container_path)
[9448]393        self.assertTrue('2 of 10' in self.browser.contents)
[13176]394        # Remove entire hostel.
[6962]395        self.browser.open(self.manage_container_path)
396        ctrl = self.browser.getControl(name='val_id')
397        value = ctrl.options[0]
398        ctrl.getControl(value=value).selected = True
399        self.browser.getControl("Remove selected", index=0).click()
400        self.assertTrue('Successfully removed' in self.browser.contents)
[13176]401        # Catalog is empty.
[6985]402        results = cat.searchResults(
403            bed_type=('regular_female_all', 'regular_female_all'))
404        results = [x for x in results]
405        assert len(results) == 0
[12894]406        # Actions are logged.
407        logcontent = open(self.logfile).read()
408        self.assertTrue(
409            'hall-1 - 9 empty beds removed, 9 beds added, 1 occupied '
410            'beds modified (hall-1_A_101_E, )'
411            in logcontent)
[9197]412
413    def test_clear_hostels(self):
414        self.browser.addHeader('Authorization', 'Basic mgr:mgrpw')
415        self.browser.open(self.container_path)
416        self.browser.getLink("Manage accommodation").click()
417        self.browser.getControl("Add hostel").click()
418        self.browser.getControl("Create hostel").click()
419        hall = self.app['hostels']['hall-1']
420        hall.blocks_for_female = ['A','B']
421        hall.rooms_per_floor = 1
422        hall.beds_for_fresh = ['A']
423        hall.beds_for_returning = ['B']
424        hall.beds_for_final = ['C']
425        hall.beds_for_all = ['D','E']
426        self.browser.open(self.container_path + '/hall-1/manage')
427        self.browser.getControl("Update all beds").click()
428        cat = queryUtility(ICatalog, name='beds_catalog')
429        results = cat.searchResults(bed_type=(None, None))
430        self.assertEqual(len(results), 11)
431        self.browser.getControl("Clear hostel").click()
432        self.assertEqual(len(self.app['hostels']['hall-1']), 0)
433        # Only the bed in hall-x remains in the catalog.
434        results = cat.searchResults(bed_type=(None, None))
435        self.assertEqual(len(results), 1)
436        # We can clear all hostels at the same time.
437        self.browser.open(self.manage_container_path)
438        self.browser.getControl("Clear all hostels").click()
439        results = cat.searchResults(bed_type=(None, None))
440        self.assertEqual(len(results), 0)
441        # Both actions have been logged.
[9202]442        logcontent = open(self.logfile).read()
[9197]443        self.assertTrue('INFO - zope.mgr - hostels.browser.HostelManageFormPage'
444                        ' - hall-1 - cleared' in logcontent)
445        self.assertTrue('zope.mgr - hostels.browser.HostelsContainerManagePage'
446                        ' - hostels - all hostels cleared' in logcontent)
[9200]447
448class ExportTests(HostelsFullSetup):
449
450    layer = FunctionalLayer
451
452    def setUp(self):
453        super(ExportTests, self).setUp()
454        self.workdir = tempfile.mkdtemp()
455        self.outfile = os.path.join(self.workdir, 'myoutput.csv')
456        return
457
458    def test_export_hostels(self):
459        exporter = HostelExporter()
460        exporter.export_all(self.app, self.outfile)
461        result = open(self.outfile, 'rb').read()
462        self.assertEqual(
463            result,
464            'beds_for_all,beds_for_final,beds_for_fresh,beds_for_pre,'
465            'beds_for_returning,beds_reserved,blocks_for_female,'
[10680]466            'blocks_for_male,floors_per_block,hostel_id,hostel_name,maint_fee,'
[9200]467            'rooms_per_floor,sort_id,special_handling\r\n,,,,,[],,,1,'
[10680]468            'hall-x,Hall 1,0.0,2,10,regular\r\n'
[9200]469            )
470        return
471
472    def test_export_beds(self):
473        exporter = BedExporter()
474        exporter.export_all(self.app, self.outfile)
475        result = open(self.outfile, 'rb').read()
476        self.assertEqual(
477            result,
478            'bed_id,bed_number,bed_type,owner,hall,block,room,bed,'
479            'special_handling,sex,bt\r\nhall_block_room_bed,1,a_b_c,,'
480            'hall,block,room,bed,a,b,c\r\n'
481            )
[9202]482        return
483
[9283]484    def tearDown(self):
485        super(ExportTests, self).tearDown()
486        clearSite()
487        shutil.rmtree(os.path.dirname(self.outfile))
488
[9202]489class HostelProcessorTest(HostelsFullSetup):
490
491    layer = FunctionalLayer
492
493    def test_import(self):
494        self.processor = HostelProcessor()
495        self.workdir = tempfile.mkdtemp()
[9283]496        self.csv_file = os.path.join(self.workdir, 'sample_hostel_data.csv')
[9202]497        open(self.csv_file, 'wb').write(HOSTEL_SAMPLE_DATA)
498        num, num_warns, fin_file, fail_file = self.processor.doImport(
499            self.csv_file, HOSTEL_HEADER_FIELDS)
500        self.assertEqual(num_warns,0)
501        self.assertEqual(len(self.app['hostels'].keys()), 11) # including hall-x
502        self.assertEqual(self.app['hostels'][
503            'block-a-upper-hostel'].hostel_id,'block-a-upper-hostel')
504        self.assertEqual(self.app['hostels'][
505            'block-a-upper-hostel'].beds_for_final, ['A', 'B'])
506        logcontent = open(self.logfile).read()
507        self.assertTrue(
[9706]508            "Hostel Processor - sample_hostel_data - block-a-upper-hostel - "
509            "updated: "
510            "beds_for_pre=['G'], floors_per_block=1, "
[9202]511            "beds_for_final=['A', 'B'], rooms_per_floor=32, "
[12981]512            "hostel_id=block-a-upper-hostel, "
[9202]513            "sort_id=20, beds_for_returning=['C', 'D'], "
514            "hostel_name=Block A Upper Hostel, beds_for_fresh=['E', 'F'], "
[12981]515            "blocks_for_female=['A']"
[9202]516            in logcontent)
517        shutil.rmtree(os.path.dirname(fin_file))
[9283]518        shutil.rmtree(self.workdir)
[10683]519        return
520
521    def test_import_update(self):
522        self.processor = HostelProcessor()
523        self.workdir = tempfile.mkdtemp()
524        self.csv_file = os.path.join(self.workdir, 'sample_hostel_data.csv')
525        open(self.csv_file, 'wb').write(HOSTEL_SAMPLE_DATA)
526        self.csv_file = os.path.join(self.workdir, 'sample_hostel_data.csv')
527        open(self.csv_file, 'wb').write(HOSTEL_SAMPLE_DATA)
528        num, num_warns, fin_file, fail_file = self.processor.doImport(
529            self.csv_file, HOSTEL_HEADER_FIELDS)
530        # We import the same file in update mode
531        num, num_warns, fin_file, fail_file = self.processor.doImport(
532            self.csv_file, HOSTEL_HEADER_FIELDS, 'update')
533        self.assertEqual(num_warns,0)
534        logcontent = open(self.logfile).read()
535        self.assertTrue(
536            "Hostel Processor - sample_hostel_data - block-a-upper-hostel - "
537            "updated: "
538            "beds_for_pre=['G'], floors_per_block=1, "
539            "beds_for_final=['A', 'B'], rooms_per_floor=32, "
[12981]540            "hostel_id=block-a-upper-hostel, "
[10683]541            "sort_id=20, beds_for_returning=['C', 'D'], "
542            "hostel_name=Block A Upper Hostel, beds_for_fresh=['E', 'F'], "
[12981]543            "blocks_for_female=['A']"
[10683]544            in logcontent)
545        shutil.rmtree(os.path.dirname(fin_file))
546        shutil.rmtree(self.workdir)
[9200]547        return
Note: See TracBrowser for help on using the repository browser.