1 # Copyright 2010, Google Inc.

     2 # All rights reserved.

     3 #

     4 # Redistribution and use in source and binary forms, with or without

     5 # modification, are permitted provided that the following conditions are

     6 # met:

     7 #

     8 #     * Redistributions of source code must retain the above copyright

     9 # notice, this list of conditions and the following disclaimer.

    10 #     * Redistributions in binary form must reproduce the above

    11 # copyright notice, this list of conditions and the following disclaimer

    12 # in the documentation and/or other materials provided with the

    13 # distribution.

    14 #     * Neither the name of Google Inc. nor the names of its

    15 # contributors may be used to endorse or promote products derived from

    16 # this software without specific prior written permission.

    17 

    18 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS

    19 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT

    20 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR

    21 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT

    22 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

    23 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

    24 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

    25 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

    26 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

    27 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

    28 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

    29 

    30 import unittest

    31 

    32 from httpplus import _readers

    33 

    34 def chunkedblock(x, eol='\r\n'):

    35     r"""Make a chunked transfer-encoding block.

    36 

    37     >>> chunkedblock('hi')

    38     '2\r\nhi\r\n'

    39     >>> chunkedblock('hi' * 10)

    40     '14\r\nhihihihihihihihihihi\r\n'

    41     >>> chunkedblock('hi', eol='\n')

    42     '2\nhi\n'

    43     """

    44     return ''.join((hex(len(x))[2:], eol, x, eol))

    45 

    46 corpus = 'foo\r\nbar\r\nbaz\r\n'

    47 

    48 

    49 class ChunkedReaderTest(unittest.TestCase):

    50     def test_many_block_boundaries(self):

    51         for step in xrange(1, len(corpus)):

    52             data = ''.join(chunkedblock(corpus[start:start+step]) for

    53                            start in xrange(0, len(corpus), step))

    54             for istep in xrange(1, len(data)):

    55                 rdr = _readers.ChunkedReader('\r\n')

    56                 print 'step', step, 'load', istep

    57                 for start in xrange(0, len(data), istep):

    58                     rdr._load(data[start:start+istep])

    59                 rdr._load(chunkedblock(''))

    60                 self.assertEqual(corpus, rdr.read(len(corpus) + 1))

    61 

    62     def test_small_chunk_blocks_large_wire_blocks(self):

    63         data = ''.join(map(chunkedblock, corpus)) + chunkedblock('')

    64         rdr = _readers.ChunkedReader('\r\n')

    65         for start in xrange(0, len(data), 4):

    66             d = data[start:start + 4]

    67             if d:

    68                 rdr._load(d)

    69         self.assertEqual(corpus, rdr.read(len(corpus)+100))

    70 # no-check-code