web/test/unittest_idownloadable.py
changeset 10731 0736e31f8644
parent 10592 dfa1dcf4d7f1
equal deleted inserted replaced
10730:874ac29b515d 10731:0736e31f8644
    40     def download_file_name(self):
    40     def download_file_name(self):
    41         """return file name of the downloadable content"""
    41         """return file name of the downloadable content"""
    42         return  self.entity.name() + '.txt'
    42         return  self.entity.name() + '.txt'
    43 
    43 
    44     def download_data(self):
    44     def download_data(self):
    45         return 'Babar is not dead!'
    45         return b'Babar is not dead!'
    46 
    46 
    47 
    47 
    48 class BrokenIDownloadableGroup(IDownloadableUser):
    48 class BrokenIDownloadableGroup(IDownloadableUser):
    49     __regid__ = 'IDownloadable'
    49     __regid__ = 'IDownloadable'
    50     __select__ = is_instance('CWGroup')
    50     __select__ = is_instance('CWGroup')
    70             get = req.headers_out.getRawHeaders
    70             get = req.headers_out.getRawHeaders
    71             self.assertEqual(['attachment;filename="admin.txt"'],
    71             self.assertEqual(['attachment;filename="admin.txt"'],
    72                              get('content-disposition'))
    72                              get('content-disposition'))
    73             self.assertEqual(['text/plain;charset=ascii'],
    73             self.assertEqual(['text/plain;charset=ascii'],
    74                              get('content-type'))
    74                              get('content-type'))
    75             self.assertEqual('Babar is not dead!', data)
    75             self.assertEqual(b'Babar is not dead!', data)
    76 
    76 
    77     def test_header_with_space(self):
    77     def test_header_with_space(self):
    78         with self.admin_access.web_request() as req:
    78         with self.admin_access.web_request() as req:
    79             self.create_user(req, login=u'c c l a', password='babar')
    79             self.create_user(req, login=u'c c l a', password='babar')
    80             req.cnx.commit()
    80             req.cnx.commit()
    85             get = req.headers_out.getRawHeaders
    85             get = req.headers_out.getRawHeaders
    86             self.assertEqual(['attachment;filename="c c l a.txt"'],
    86             self.assertEqual(['attachment;filename="c c l a.txt"'],
    87                              get('content-disposition'))
    87                              get('content-disposition'))
    88             self.assertEqual(['text/plain;charset=ascii'],
    88             self.assertEqual(['text/plain;charset=ascii'],
    89                              get('content-type'))
    89                              get('content-type'))
    90             self.assertEqual('Babar is not dead!', data)
    90             self.assertEqual(b'Babar is not dead!', data)
    91 
    91 
    92     def test_header_with_space_and_comma(self):
    92     def test_header_with_space_and_comma(self):
    93         with self.admin_access.web_request() as req:
    93         with self.admin_access.web_request() as req:
    94             self.create_user(req, login=u'c " l\\ a', password='babar')
    94             self.create_user(req, login=u'c " l\\ a', password='babar')
    95             req.cnx.commit()
    95             req.cnx.commit()
   100             get = req.headers_out.getRawHeaders
   100             get = req.headers_out.getRawHeaders
   101             self.assertEqual([r'attachment;filename="c \" l\\ a.txt"'],
   101             self.assertEqual([r'attachment;filename="c \" l\\ a.txt"'],
   102                              get('content-disposition'))
   102                              get('content-disposition'))
   103             self.assertEqual(['text/plain;charset=ascii'],
   103             self.assertEqual(['text/plain;charset=ascii'],
   104                              get('content-type'))
   104                              get('content-type'))
   105             self.assertEqual('Babar is not dead!', data)
   105             self.assertEqual(b'Babar is not dead!', data)
   106 
   106 
   107     def test_header_unicode_filename(self):
   107     def test_header_unicode_filename(self):
   108         with self.admin_access.web_request() as req:
   108         with self.admin_access.web_request() as req:
   109             self.create_user(req, login=u'cécilia', password='babar')
   109             self.create_user(req, login=u'cécilia', password='babar')
   110             req.cnx.commit()
   110             req.cnx.commit()