← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test__chk_map.py

  • Committer: John Arbash Meinel
  • Date: 2010-05-11 10:45:26 UTC
  • mto: This revision was merged to the branch mainline in revision 5225.
  • Revision ID: john@arbash-meinel.com-20100511104526-zxnstcxta22hzw2n
Implement a compiled extension for parsing the text key out of a CHKInventory value.

Related to bug #562666. This seems to shave 5-10% out of the time spent doing a complete
branch of bzr.dev/launchpad/etc.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test__chk_map.py
18
18
 
19
19
from bzrlib import (
20
20
    chk_map,
 
21
    inventory,
21
22
    tests,
22
23
    )
23
24
from bzrlib.static_tuple import StaticTuple
236
237
        self.assertEqual(("sha1:1234",), node.key())
237
238
        self.assertEqual('pref\x00fo', node._search_prefix)
238
239
        self.assertEqual({'pref\x00fo\x00': ('sha1:abcd',)}, node._items)
 
240
 
 
241
 
 
242
class Test_BytesToTextKey(tests.TestCase):
 
243
 
 
244
    def assertBytesToTextKey(self, key, bytes):
 
245
        self.assertEqual(key,
 
246
                         self.module._bytes_to_text_key(bytes))
 
247
 
 
248
    def assertBytesToTextKeyRaises(self, bytes):
 
249
        # These are invalid bytes, and we want to make sure the code under test
 
250
        # raises an exception rather than segfaults, etc. We don't particularly
 
251
        # care what exception.
 
252
        self.assertRaises(Exception, self.module._bytes_to_text_key, bytes)
 
253
 
 
254
    def test_file(self):
 
255
        self.assertBytesToTextKey(('file-id', 'revision-id'),
 
256
                 'file: file-id\nparent-id\nname\nrevision-id\n'
 
257
                 'da39a3ee5e6b4b0d3255bfef95601890afd80709\n100\nN')
 
258
 
 
259
    def test_invalid_no_kind(self):
 
260
        self.assertBytesToTextKeyRaises(
 
261
                 'file  file-id\nparent-id\nname\nrevision-id\n'
 
262
                 'da39a3ee5e6b4b0d3255bfef95601890afd80709\n100\nN')
 
263
 
 
264
    def test_invalid_no_space(self):
 
265
        self.assertBytesToTextKeyRaises(
 
266
                 'file:file-id\nparent-id\nname\nrevision-id\n'
 
267
                 'da39a3ee5e6b4b0d3255bfef95601890afd80709\n100\nN')
 
268
 
 
269
    def test_invalid_too_short_file_id(self):
 
270
        self.assertBytesToTextKeyRaises('file:file-id')
 
271
 
 
272
    def test_invalid_too_short_parent_id(self):
 
273
        self.assertBytesToTextKeyRaises('file:file-id\nparent-id')
 
274
 
 
275
    def test_invalid_too_short_name(self):
 
276
        self.assertBytesToTextKeyRaises('file:file-id\nparent-id\nname')
 
277
 
 
278
    def test_dir(self):
 
279
        self.assertBytesToTextKey(('dir-id', 'revision-id'),
 
280
                 'dir: dir-id\nparent-id\nname\nrevision-id')