← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_groupcompress.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-09-29 20:43:19 UTC
  • mfrom: (5439.2.3 2.3-gcb-peak-mem)
  • Revision ID: pqm@pqm.ubuntu.com-20100929204319-s4pxrbqu9osgas9f
(jam) Reduce the peak memory in GroupCompressBlock

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_groupcompress.py
347
347
        self.assertEqual(z_content, block._z_content)
348
348
        self.assertEqual(content, block._content)
349
349
 
 
350
    def test_to_chunks(self):
 
351
        content_chunks = ['this is some content\n',
 
352
                          'this content will be compressed\n']
 
353
        content_len = sum(map(len, content_chunks))
 
354
        content = ''.join(content_chunks)
 
355
        gcb = groupcompress.GroupCompressBlock()
 
356
        gcb.set_chunked_content(content_chunks, content_len)
 
357
        total_len, block_chunks = gcb.to_chunks()
 
358
        block_bytes = ''.join(block_chunks)
 
359
        self.assertEqual(gcb._z_content_length, len(gcb._z_content))
 
360
        self.assertEqual(total_len, len(block_bytes))
 
361
        self.assertEqual(gcb._content_length, content_len)
 
362
        expected_header =('gcb1z\n' # group compress block v1 zlib
 
363
                          '%d\n' # Length of compressed content
 
364
                          '%d\n' # Length of uncompressed content
 
365
                         ) % (gcb._z_content_length, gcb._content_length)
 
366
        # The first chunk should be the header chunk. It is small, fixed size,
 
367
        # and there is no compelling reason to split it up
 
368
        self.assertEqual(expected_header, block_chunks[0])
 
369
        self.assertStartsWith(block_bytes, expected_header)
 
370
        remaining_bytes = block_bytes[len(expected_header):]
 
371
        raw_bytes = zlib.decompress(remaining_bytes)
 
372
        self.assertEqual(content, raw_bytes)
 
373
 
350
374
    def test_to_bytes(self):
351
375
        content = ('this is some content\n'
352
376
                   'this content will be compressed\n')
389
413
        z_content = zlib.compress(content)
390
414
        self.assertEqual(57182, len(z_content))
391
415
        block = groupcompress.GroupCompressBlock()
392
 
        block._z_content = z_content
 
416
        block._z_content_chunks = (z_content,)
393
417
        block._z_content_length = len(z_content)
394
418
        block._compressor_name = 'zlib'
395
419
        block._content_length = 158634
434
458
        z_content = zlib.compress(content)
435
459
        self.assertEqual(57182, len(z_content))
436
460
        block = groupcompress.GroupCompressBlock()
437
 
        block._z_content = z_content
 
461
        block._z_content_chunks = (z_content,)
438
462
        block._z_content_length = len(z_content)
439
463
        block._compressor_name = 'zlib'
440
464
        block._content_length = 158634