← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_http_response.py

  • Committer: Andrew Bennetts
  • Date: 2008-07-28 06:53:44 UTC
  • mfrom: (3581 +trunk)
  • mto: This revision was merged to the branch mainline in revision 3583.
  • Revision ID: andrew.bennetts@canonical.com-20080728065344-ocndjoycs903q6fz
Merge bzr.dev.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_http_response.py
225
225
        self.assertEquals('', f.read(0))
226
226
        self.assertEquals('', f.read(1))
227
227
 
 
228
 
228
229
class TestRangeFileSizeKnown(tests.TestCase, TestRangeFileMixin):
229
230
    """Test a RangeFile for a whole file whose size is known."""
230
231
 
253
254
        f._pos = 0 # Force an invalid pos
254
255
        self.assertRaises(errors.InvalidRange, f.read, 2)
255
256
 
 
257
 
256
258
class TestRangeFileMultipleRanges(tests.TestCase, TestRangeFileMixin):
257
259
    """Test a RangeFile for multiple ranges.
258
260
 
266
268
    fact) in real uses but may lead to hard to track bugs.
267
269
    """
268
270
 
 
271
    # The following is used to represent the boundary paramter defined
 
272
    # in HTTP response headers and the boundary lines that separate
 
273
    # multipart content.
 
274
 
 
275
    boundary = "separation"
 
276
 
269
277
    def setUp(self):
270
278
        super(TestRangeFileMultipleRanges, self).setUp()
271
279
 
272
 
        boundary = 'separation'
 
280
        boundary = self.boundary
273
281
 
274
282
        content = ''
275
283
        self.first_range_start = 25
281
289
            content += self._multipart_byterange(part, start, boundary,
282
290
                                                 file_size)
283
291
        # Final boundary
284
 
        content += self._boundary_line(boundary)
 
292
        content += self._boundary_line()
285
293
 
286
294
        self._file = response.RangeFile('Multiple_ranges_file',
287
295
                                        StringIO(content))
 
296
        self.set_file_boundary()
 
297
 
 
298
    def _boundary_line(self):
 
299
        """Helper to build the formatted boundary line."""
 
300
        return '--' + self.boundary + '\r\n'
 
301
 
 
302
    def set_file_boundary(self):
288
303
        # Ranges are set by decoding the range headers, the RangeFile user is
289
304
        # supposed to call the following before using seek or read since it
290
305
        # requires knowing the *response* headers (in that case the boundary
291
306
        # which is part of the Content-Type header).
292
 
        self._file.set_boundary(boundary)
293
 
 
294
 
    def _boundary_line(self, boundary):
295
 
        """Helper to build the formatted boundary line."""
296
 
        return '--' + boundary + '\r\n'
 
307
        self._file.set_boundary(self.boundary)
297
308
 
298
309
    def _multipart_byterange(self, data, offset, boundary, file_size='*'):
299
310
        """Encode a part of a file as a multipart/byterange MIME type.
311
322
        :return: a string containing the data encoded as it will appear in the
312
323
            HTTP response body.
313
324
        """
314
 
        bline = self._boundary_line(boundary)
 
325
        bline = self._boundary_line()
315
326
        # Each range begins with a boundary line
316
327
        range = bline
317
328
        # A range is described by a set of headers, but only 'Content-Range' is
395
406
        self.assertRaises(errors.InvalidHttpResponse, f.read, 1)
396
407
 
397
408
 
 
409
class TestRangeFileMultipleRangesQuotedBoundaries(TestRangeFileMultipleRanges):
 
410
    """Perform the same tests as TestRangeFileMultipleRanges, but uses 
 
411
    an angle-bracket quoted boundary string like IIS 6.0 and 7.0
 
412
    (but not IIS 5, which breaks the RFC in a different way
 
413
    by using square brackets, not angle brackets)
 
414
    
 
415
    This reveals a bug caused by 
 
416
    
 
417
    - The bad implementation of RFC 822 unquoting in Python (angles are not 
 
418
      quotes), coupled with 
 
419
 
 
420
    - The bad implementation of RFC 2046 in IIS (angles are not permitted chars
 
421
      in boundary lines).
 
422
 
 
423
    """
 
424
    # The boundary as it appears in boundary lines
 
425
    # IIS 6 and 7 use this value
 
426
    _boundary_trimmed = "q1w2e3r4t5y6u7i8o9p0zaxscdvfbgnhmjklkl"
 
427
    boundary = '<' + _boundary_trimmed + '>'
 
428
 
 
429
    def set_file_boundary(self):
 
430
        # Emulate broken rfc822.unquote() here by removing angles
 
431
        self._file.set_boundary(self._boundary_trimmed)
 
432
 
 
433
 
398
434
class TestRangeFileVarious(tests.TestCase):
399
435
    """Tests RangeFile aspects not covered elsewhere."""
400
436