Changeset 7101 for main/waeup.sirp

Timestamp:

13 Nov 2011, 06:36:31 (13 years ago)

Author:

Henrik Bettermann

Message:

Add test for file uploader in students.

File:

• main/waeup.sirp/trunk/src/waeup/sirp/students/tests/test_browser.py (modified) (4 diffs)

main/waeup.sirp/trunk/src/waeup/sirp/students/tests/test_browser.py

@@ -25 +25 @@
 import shutil
 import tempfile
-import cStringIO
+from StringIO import StringIO
 from datetime import datetime
 import grok
@@ -357 +357 @@
         return
 
+    def test_upload_file(self):
+        # Managers can upload a file via the StudentClearanceManageFormPage
+        # The image is stored even if form has errors
+        self.browser.addHeader('Authorization', 'Basic mgr:mgrpw')
+        self.browser.open(self.edit_clearance_student_path)
+        # Create a pseudo image file and select it to be uploaded in form
+        # as birth certificate
+        pseudo_image = StringIO('I pretend to be a graphics file')
+        ctrl = self.browser.getControl(name='form.birth_certificate')
+        file_ctrl = ctrl.mech_control
+        file_ctrl.add_file(pseudo_image, filename='myphoto.jpg')
+        self.browser.getControl("Save").click() # submit form
+        # Form has errors ...
+        self.assertTrue(
+            'Required input is missing' in self.browser.contents)
+        # ... but file has been uploaded and a correct <img> link included
+        self.assertTrue(
+            '<a target="image" href="birth_certificate.jpg">' in self.browser.contents)
+        # There is a correct <img> link included
+        self.assertTrue(
+            '<a target="image" href="birth_certificate.jpg">'
+            in self.browser.contents)
+        # Browsing the link shows a real image
+        self.browser.open('birth_certificate.jpg')
+        self.assertEqual(
+            self.browser.headers['content-type'], 'image/jpeg')
+        self.assertEqual(len(self.browser.contents), 31)
+        # Reuploading a file which is bigger than 150k will raise an error
+        self.browser.open(self.edit_clearance_student_path)
+        photo_content = 'A' * 1024 * 151  # A string of 21 KB size
+        pseudo_image = StringIO(photo_content)
+        ctrl = self.browser.getControl(name='form.birth_certificate')
+        file_ctrl = ctrl.mech_control
+        file_ctrl.add_file(pseudo_image, filename='myphoto.jpg')
+        self.browser.getControl("Save").click() # submit form
+        self.assertTrue(
+            'Uploaded image is too big' in self.browser.contents)
+
     def test_manage_course_lists(self):
         self.browser.addHeader('Authorization', 'Basic mgr:mgrpw')
@@ -468 +506 @@
         self.browser.open(datacenter_path)
         self.browser.getLink('Upload CSV file').click()
-        filecontents = cStringIO.StringIO(open('students.csv', 'rb').read())
+        filecontents = StringIO(open('students.csv', 'rb').read())
         filewidget = self.browser.getControl(name='uploadfile:file')
         filewidget.add_file(filecontents, 'text/plain', 'students.csv')
@@ -494 +532 @@
         self.browser.open(datacenter_path)
         self.browser.getLink('Upload CSV file').click()
-        filecontents = cStringIO.StringIO(open('studycourses.csv', 'rb').read())
+        filecontents = StringIO(open('studycourses.csv', 'rb').read())
         filewidget = self.browser.getControl(name='uploadfile:file')
         filewidget.add_file(filecontents, 'text/plain', 'studycourses.csv')