/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_dirstate.py

  • Committer: Martin Pool
  • Date: 2007-03-25 12:46:04 UTC
  • mto: (2323.5.3 0.15)
  • mto: This revision was merged to the branch mainline in revision 2390.
  • Revision ID: mbp@sourcefrog.net-20070325124604-sabilbi5m4902ot8
clearer test data for test merge_swapping_rename

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2006, 2007 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""Tests of the dirstate functionality being built for WorkingTreeFormat4."""
18
18
 
19
19
import bisect
20
20
import os
 
21
import time
21
22
 
22
23
from bzrlib import (
23
24
    dirstate,
24
25
    errors,
25
 
    inventory,
26
 
    memorytree,
27
26
    osutils,
28
 
    revision as _mod_revision,
29
 
    tests,
30
27
    )
31
 
from bzrlib.tests import test_osutils
 
28
from bzrlib.memorytree import MemoryTree
 
29
from bzrlib.osutils import has_symlinks
 
30
from bzrlib.tests import (
 
31
        TestCase,
 
32
        TestCaseWithTransport,
 
33
        TestSkipped,
 
34
        )
32
35
 
33
36
 
34
37
# TODO:
44
47
# set_path_id  setting id when state is in memory modified
45
48
 
46
49
 
47
 
def load_tests(basic_tests, module, loader):
48
 
    suite = loader.suiteClass()
49
 
    dir_reader_tests, remaining_tests = tests.split_suite_by_condition(
50
 
        basic_tests, tests.condition_isinstance(TestCaseWithDirState))
51
 
    tests.multiply_tests(dir_reader_tests,
52
 
                         test_osutils.dir_reader_scenarios(), suite)
53
 
    suite.addTest(remaining_tests)
54
 
    return suite
55
 
 
56
 
 
57
 
class TestCaseWithDirState(tests.TestCaseWithTransport):
 
50
class TestCaseWithDirState(TestCaseWithTransport):
58
51
    """Helper functions for creating DirState objects with various content."""
59
52
 
60
 
    # Set by load_tests
61
 
    _dir_reader_class = None
62
 
    _native_to_unicode = None # Not used yet
63
 
 
64
 
    def setUp(self):
65
 
        tests.TestCaseWithTransport.setUp(self)
66
 
 
67
 
        self.overrideAttr(osutils,
68
 
                          '_selected_dir_reader', self._dir_reader_class())
69
 
 
70
53
    def create_empty_dirstate(self):
71
54
        """Return a locked but empty dirstate"""
72
55
        state = dirstate.DirState.initialize('dirstate')
119
102
         b/g      g-file
120
103
         b/h\xc3\xa5  h-\xc3\xa5-file  #This is u'\xe5' encoded into utf-8
121
104
 
122
 
        Notice that a/e is an empty directory.
123
 
 
124
 
        :return: The dirstate, still write-locked.
 
105
        # Notice that a/e is an empty directory.
125
106
        """
126
107
        packed_stat = 'AAAAREUHaIpFB2iKAAADAQAtkqUAAIGk'
127
108
        null_sha = 'xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx'
180
161
        """
181
162
        # The state should already be write locked, since we just had to do
182
163
        # some operation to get here.
183
 
        self.assertTrue(state._lock_token is not None)
 
164
        assert state._lock_token is not None
184
165
        try:
185
166
            self.assertEqual(expected_result[0],  state.get_parent_ids())
186
167
            # there should be no ghosts in this tree.
190
171
            state.save()
191
172
        finally:
192
173
            state.unlock()
193
 
        del state
 
174
        del state # Callers should unlock
194
175
        state = dirstate.DirState.on_file('dirstate')
195
176
        state.lock_read()
196
177
        try:
198
179
        finally:
199
180
            state.unlock()
200
181
 
201
 
    def create_basic_dirstate(self):
202
 
        """Create a dirstate with a few files and directories.
203
 
 
204
 
            a
205
 
            b/
206
 
              c
207
 
              d/
208
 
                e
209
 
            b-c
210
 
            f
211
 
        """
212
 
        tree = self.make_branch_and_tree('tree')
213
 
        paths = ['a', 'b/', 'b/c', 'b/d/', 'b/d/e', 'b-c', 'f']
214
 
        file_ids = ['a-id', 'b-id', 'c-id', 'd-id', 'e-id', 'b-c-id', 'f-id']
215
 
        self.build_tree(['tree/' + p for p in paths])
216
 
        tree.set_root_id('TREE_ROOT')
217
 
        tree.add([p.rstrip('/') for p in paths], file_ids)
218
 
        tree.commit('initial', rev_id='rev-1')
219
 
        revision_id = 'rev-1'
220
 
        # a_packed_stat = dirstate.pack_stat(os.stat('tree/a'))
221
 
        t = self.get_transport('tree')
222
 
        a_text = t.get_bytes('a')
223
 
        a_sha = osutils.sha_string(a_text)
224
 
        a_len = len(a_text)
225
 
        # b_packed_stat = dirstate.pack_stat(os.stat('tree/b'))
226
 
        # c_packed_stat = dirstate.pack_stat(os.stat('tree/b/c'))
227
 
        c_text = t.get_bytes('b/c')
228
 
        c_sha = osutils.sha_string(c_text)
229
 
        c_len = len(c_text)
230
 
        # d_packed_stat = dirstate.pack_stat(os.stat('tree/b/d'))
231
 
        # e_packed_stat = dirstate.pack_stat(os.stat('tree/b/d/e'))
232
 
        e_text = t.get_bytes('b/d/e')
233
 
        e_sha = osutils.sha_string(e_text)
234
 
        e_len = len(e_text)
235
 
        b_c_text = t.get_bytes('b-c')
236
 
        b_c_sha = osutils.sha_string(b_c_text)
237
 
        b_c_len = len(b_c_text)
238
 
        # f_packed_stat = dirstate.pack_stat(os.stat('tree/f'))
239
 
        f_text = t.get_bytes('f')
240
 
        f_sha = osutils.sha_string(f_text)
241
 
        f_len = len(f_text)
242
 
        null_stat = dirstate.DirState.NULLSTAT
243
 
        expected = {
244
 
            '':(('', '', 'TREE_ROOT'), [
245
 
                  ('d', '', 0, False, null_stat),
246
 
                  ('d', '', 0, False, revision_id),
247
 
                ]),
248
 
            'a':(('', 'a', 'a-id'), [
249
 
                   ('f', '', 0, False, null_stat),
250
 
                   ('f', a_sha, a_len, False, revision_id),
251
 
                 ]),
252
 
            'b':(('', 'b', 'b-id'), [
253
 
                  ('d', '', 0, False, null_stat),
254
 
                  ('d', '', 0, False, revision_id),
255
 
                 ]),
256
 
            'b/c':(('b', 'c', 'c-id'), [
257
 
                    ('f', '', 0, False, null_stat),
258
 
                    ('f', c_sha, c_len, False, revision_id),
259
 
                   ]),
260
 
            'b/d':(('b', 'd', 'd-id'), [
261
 
                    ('d', '', 0, False, null_stat),
262
 
                    ('d', '', 0, False, revision_id),
263
 
                   ]),
264
 
            'b/d/e':(('b/d', 'e', 'e-id'), [
265
 
                      ('f', '', 0, False, null_stat),
266
 
                      ('f', e_sha, e_len, False, revision_id),
267
 
                     ]),
268
 
            'b-c':(('', 'b-c', 'b-c-id'), [
269
 
                      ('f', '', 0, False, null_stat),
270
 
                      ('f', b_c_sha, b_c_len, False, revision_id),
271
 
                     ]),
272
 
            'f':(('', 'f', 'f-id'), [
273
 
                  ('f', '', 0, False, null_stat),
274
 
                  ('f', f_sha, f_len, False, revision_id),
275
 
                 ]),
276
 
        }
277
 
        state = dirstate.DirState.from_tree(tree, 'dirstate')
278
 
        try:
279
 
            state.save()
280
 
        finally:
281
 
            state.unlock()
282
 
        # Use a different object, to make sure nothing is pre-cached in memory.
283
 
        state = dirstate.DirState.on_file('dirstate')
284
 
        state.lock_read()
285
 
        self.addCleanup(state.unlock)
286
 
        self.assertEqual(dirstate.DirState.NOT_IN_MEMORY,
287
 
                         state._dirblock_state)
288
 
        # This is code is only really tested if we actually have to make more
289
 
        # than one read, so set the page size to something smaller.
290
 
        # We want it to contain about 2.2 records, so that we have a couple
291
 
        # records that we can read per attempt
292
 
        state._bisect_page_size = 200
293
 
        return tree, state, expected
294
 
 
295
 
    def create_duplicated_dirstate(self):
296
 
        """Create a dirstate with a deleted and added entries.
297
 
 
298
 
        This grabs a basic_dirstate, and then removes and re adds every entry
299
 
        with a new file id.
300
 
        """
301
 
        tree, state, expected = self.create_basic_dirstate()
302
 
        # Now we will just remove and add every file so we get an extra entry
303
 
        # per entry. Unversion in reverse order so we handle subdirs
304
 
        tree.unversion(['f-id', 'b-c-id', 'e-id', 'd-id', 'c-id', 'b-id', 'a-id'])
305
 
        tree.add(['a', 'b', 'b/c', 'b/d', 'b/d/e', 'b-c', 'f'],
306
 
                 ['a-id2', 'b-id2', 'c-id2', 'd-id2', 'e-id2', 'b-c-id2', 'f-id2'])
307
 
 
308
 
        # Update the expected dictionary.
309
 
        for path in ['a', 'b', 'b/c', 'b/d', 'b/d/e', 'b-c', 'f']:
310
 
            orig = expected[path]
311
 
            path2 = path + '2'
312
 
            # This record was deleted in the current tree
313
 
            expected[path] = (orig[0], [dirstate.DirState.NULL_PARENT_DETAILS,
314
 
                                        orig[1][1]])
315
 
            new_key = (orig[0][0], orig[0][1], orig[0][2]+'2')
316
 
            # And didn't exist in the basis tree
317
 
            expected[path2] = (new_key, [orig[1][0],
318
 
                                         dirstate.DirState.NULL_PARENT_DETAILS])
319
 
 
320
 
        # We will replace the 'dirstate' file underneath 'state', but that is
321
 
        # okay as lock as we unlock 'state' first.
322
 
        state.unlock()
323
 
        try:
324
 
            new_state = dirstate.DirState.from_tree(tree, 'dirstate')
325
 
            try:
326
 
                new_state.save()
327
 
            finally:
328
 
                new_state.unlock()
329
 
        finally:
330
 
            # But we need to leave state in a read-lock because we already have
331
 
            # a cleanup scheduled
332
 
            state.lock_read()
333
 
        return tree, state, expected
334
 
 
335
 
    def create_renamed_dirstate(self):
336
 
        """Create a dirstate with a few internal renames.
337
 
 
338
 
        This takes the basic dirstate, and moves the paths around.
339
 
        """
340
 
        tree, state, expected = self.create_basic_dirstate()
341
 
        # Rename a file
342
 
        tree.rename_one('a', 'b/g')
343
 
        # And a directory
344
 
        tree.rename_one('b/d', 'h')
345
 
 
346
 
        old_a = expected['a']
347
 
        expected['a'] = (old_a[0], [('r', 'b/g', 0, False, ''), old_a[1][1]])
348
 
        expected['b/g'] = (('b', 'g', 'a-id'), [old_a[1][0],
349
 
                                                ('r', 'a', 0, False, '')])
350
 
        old_d = expected['b/d']
351
 
        expected['b/d'] = (old_d[0], [('r', 'h', 0, False, ''), old_d[1][1]])
352
 
        expected['h'] = (('', 'h', 'd-id'), [old_d[1][0],
353
 
                                             ('r', 'b/d', 0, False, '')])
354
 
 
355
 
        old_e = expected['b/d/e']
356
 
        expected['b/d/e'] = (old_e[0], [('r', 'h/e', 0, False, ''),
357
 
                             old_e[1][1]])
358
 
        expected['h/e'] = (('h', 'e', 'e-id'), [old_e[1][0],
359
 
                                                ('r', 'b/d/e', 0, False, '')])
360
 
 
361
 
        state.unlock()
362
 
        try:
363
 
            new_state = dirstate.DirState.from_tree(tree, 'dirstate')
364
 
            try:
365
 
                new_state.save()
366
 
            finally:
367
 
                new_state.unlock()
368
 
        finally:
369
 
            state.lock_read()
370
 
        return tree, state, expected
371
 
 
372
182
 
373
183
class TestTreeToDirState(TestCaseWithDirState):
374
184
 
377
187
        # There are no files on disk and no parents
378
188
        tree = self.make_branch_and_tree('tree')
379
189
        expected_result = ([], [
380
 
            (('', '', tree.get_root_id()), # common details
 
190
            (('', '', tree.path2id('')), # common details
381
191
             [('d', '', 0, False, dirstate.DirState.NULLSTAT), # current tree
382
192
             ])])
383
193
        state = dirstate.DirState.from_tree(tree, 'dirstate')
390
200
        rev_id = tree.commit('first post').encode('utf8')
391
201
        root_stat_pack = dirstate.pack_stat(os.stat(tree.basedir))
392
202
        expected_result = ([rev_id], [
393
 
            (('', '', tree.get_root_id()), # common details
 
203
            (('', '', tree.path2id('')), # common details
394
204
             [('d', '', 0, False, dirstate.DirState.NULLSTAT), # current tree
395
205
              ('d', '', 0, False, rev_id), # first parent details
396
206
             ])])
410
220
        rev_id2 = tree2.commit('second post', allow_pointless=True)
411
221
        tree.merge_from_branch(tree2.branch)
412
222
        expected_result = ([rev_id, rev_id2], [
413
 
            (('', '', tree.get_root_id()), # common details
 
223
            (('', '', tree.path2id('')), # common details
414
224
             [('d', '', 0, False, dirstate.DirState.NULLSTAT), # current tree
415
225
              ('d', '', 0, False, rev_id), # first parent details
416
 
              ('d', '', 0, False, rev_id), # second parent details
 
226
              ('d', '', 0, False, rev_id2), # second parent details
417
227
             ])])
418
228
        state = dirstate.DirState.from_tree(tree, 'dirstate')
419
229
        self.check_state_with_reopen(expected_result, state)
429
239
        tree = self.make_branch_and_tree('tree')
430
240
        self.build_tree(['tree/unknown'])
431
241
        expected_result = ([], [
432
 
            (('', '', tree.get_root_id()), # common details
 
242
            (('', '', tree.path2id('')), # common details
433
243
             [('d', '', 0, False, dirstate.DirState.NULLSTAT), # current tree
434
244
             ])])
435
245
        state = dirstate.DirState.from_tree(tree, 'dirstate')
438
248
    def get_tree_with_a_file(self):
439
249
        tree = self.make_branch_and_tree('tree')
440
250
        self.build_tree(['tree/a file'])
441
 
        tree.add('a file', 'a-file-id')
 
251
        tree.add('a file', 'a file id')
442
252
        return tree
443
253
 
444
254
    def test_non_empty_no_parents_to_dirstate(self):
446
256
        # There are files on disk and no parents
447
257
        tree = self.get_tree_with_a_file()
448
258
        expected_result = ([], [
449
 
            (('', '', tree.get_root_id()), # common details
 
259
            (('', '', tree.path2id('')), # common details
450
260
             [('d', '', 0, False, dirstate.DirState.NULLSTAT), # current tree
451
261
             ]),
452
 
            (('', 'a file', 'a-file-id'), # common
 
262
            (('', 'a file', 'a file id'), # common
453
263
             [('f', '', 0, False, dirstate.DirState.NULLSTAT), # current
454
264
             ]),
455
265
            ])
464
274
        # and length:
465
275
        self.build_tree_contents([('tree/a file', 'new content\n')])
466
276
        expected_result = ([rev_id], [
467
 
            (('', '', tree.get_root_id()), # common details
 
277
            (('', '', tree.path2id('')), # common details
468
278
             [('d', '', 0, False, dirstate.DirState.NULLSTAT), # current tree
469
279
              ('d', '', 0, False, rev_id), # first parent details
470
280
             ]),
471
 
            (('', 'a file', 'a-file-id'), # common
 
281
            (('', 'a file', 'a file id'), # common
472
282
             [('f', '', 0, False, dirstate.DirState.NULLSTAT), # current
473
283
              ('f', 'c3ed76e4bfd45ff1763ca206055bca8e9fc28aa8', 24, False,
474
284
               rev_id), # first parent
491
301
        # and length again, giving us three distinct values:
492
302
        self.build_tree_contents([('tree/a file', 'new content\n')])
493
303
        expected_result = ([rev_id, rev_id2], [
494
 
            (('', '', tree.get_root_id()), # common details
 
304
            (('', '', tree.path2id('')), # common details
495
305
             [('d', '', 0, False, dirstate.DirState.NULLSTAT), # current tree
496
306
              ('d', '', 0, False, rev_id), # first parent details
497
 
              ('d', '', 0, False, rev_id), # second parent details
 
307
              ('d', '', 0, False, rev_id2), # second parent details
498
308
             ]),
499
 
            (('', 'a file', 'a-file-id'), # common
 
309
            (('', 'a file', 'a file id'), # common
500
310
             [('f', '', 0, False, dirstate.DirState.NULLSTAT), # current
501
311
              ('f', 'c3ed76e4bfd45ff1763ca206055bca8e9fc28aa8', 24, False,
502
312
               rev_id), # first parent
543
353
        # get a state object
544
354
        # no parents, default tree content
545
355
        expected_result = ([], [
546
 
            (('', '', tree.get_root_id()), # common details
 
356
            (('', '', tree.path2id('')), # common details
547
357
             # current tree details, but new from_tree skips statting, it
548
358
             # uses set_state_from_inventory, and thus depends on the
549
359
             # inventory state.
580
390
        state.lock_read()
581
391
        try:
582
392
            entry = state._get_entry(0, path_utf8='a-file')
583
 
            # The current size should be 0 (default)
584
 
            self.assertEqual(0, entry[1][0][2])
 
393
            # The current sha1 sum should be empty
 
394
            self.assertEqual('', entry[1][0][1])
585
395
            # We should have a real entry.
586
396
            self.assertNotEqual((None, None), entry)
587
 
            # Make sure everything is old enough
588
 
            state._sha_cutoff_time()
589
 
            state._cutoff_time += 10
590
 
            # Change the file length
591
 
            self.build_tree_contents([('a-file', 'shorter')])
592
 
            sha1sum = dirstate.update_entry(state, entry, 'a-file',
593
 
                os.lstat('a-file'))
594
 
            # new file, no cached sha:
595
 
            self.assertEqual(None, sha1sum)
 
397
            sha1sum = state.update_entry(entry, 'a-file', os.lstat('a-file'))
 
398
            # We should have gotten a real sha1
 
399
            self.assertEqual('ecc5374e9ed82ad3ea3b4d452ea995a5fd3e70e3',
 
400
                             sha1sum)
596
401
 
597
402
            # The dirblock has been updated
598
 
            self.assertEqual(7, entry[1][0][2])
 
403
            self.assertEqual(sha1sum, entry[1][0][1])
599
404
            self.assertEqual(dirstate.DirState.IN_MEMORY_MODIFIED,
600
405
                             state._dirblock_state)
601
406
 
611
416
        state.lock_read()
612
417
        try:
613
418
            entry = state._get_entry(0, path_utf8='a-file')
614
 
            self.assertEqual(7, entry[1][0][2])
 
419
            self.assertEqual(sha1sum, entry[1][0][1])
615
420
        finally:
616
421
            state.unlock()
617
422
 
630
435
        state.lock_read()
631
436
        try:
632
437
            entry = state._get_entry(0, path_utf8='a-file')
633
 
            sha1sum = dirstate.update_entry(state, entry, 'a-file',
634
 
                os.lstat('a-file'))
635
 
            # No sha - too new
636
 
            self.assertEqual(None, sha1sum)
 
438
            sha1sum = state.update_entry(entry, 'a-file', os.lstat('a-file'))
 
439
            # We should have gotten a real sha1
 
440
            self.assertEqual('ecc5374e9ed82ad3ea3b4d452ea995a5fd3e70e3',
 
441
                             sha1sum)
637
442
            self.assertEqual(dirstate.DirState.IN_MEMORY_MODIFIED,
638
443
                             state._dirblock_state)
639
444
 
656
461
                state2.unlock()
657
462
        finally:
658
463
            state.unlock()
659
 
 
 
464
        
660
465
        # The file on disk should not be modified.
661
466
        state = dirstate.DirState.on_file('dirstate')
662
467
        state.lock_read()
666
471
        finally:
667
472
            state.unlock()
668
473
 
669
 
    def test_save_refuses_if_changes_aborted(self):
670
 
        self.build_tree(['a-file', 'a-dir/'])
671
 
        state = dirstate.DirState.initialize('dirstate')
672
 
        try:
673
 
            # No stat and no sha1 sum.
674
 
            state.add('a-file', 'a-file-id', 'file', None, '')
675
 
            state.save()
676
 
        finally:
677
 
            state.unlock()
678
 
 
679
 
        # The dirstate should include TREE_ROOT and 'a-file' and nothing else
680
 
        expected_blocks = [
681
 
            ('', [(('', '', 'TREE_ROOT'),
682
 
                   [('d', '', 0, False, dirstate.DirState.NULLSTAT)])]),
683
 
            ('', [(('', 'a-file', 'a-file-id'),
684
 
                   [('f', '', 0, False, dirstate.DirState.NULLSTAT)])]),
685
 
        ]
686
 
 
687
 
        state = dirstate.DirState.on_file('dirstate')
688
 
        state.lock_write()
689
 
        try:
690
 
            state._read_dirblocks_if_needed()
691
 
            self.assertEqual(expected_blocks, state._dirblocks)
692
 
 
693
 
            # Now modify the state, but mark it as inconsistent
694
 
            state.add('a-dir', 'a-dir-id', 'directory', None, '')
695
 
            state._changes_aborted = True
696
 
            state.save()
697
 
        finally:
698
 
            state.unlock()
699
 
 
700
 
        state = dirstate.DirState.on_file('dirstate')
701
 
        state.lock_read()
702
 
        try:
703
 
            state._read_dirblocks_if_needed()
704
 
            self.assertEqual(expected_blocks, state._dirblocks)
705
 
        finally:
706
 
            state.unlock()
707
 
 
708
474
 
709
475
class TestDirStateInitialize(TestCaseWithDirState):
710
476
 
718
484
        try:
719
485
            self.assertIsInstance(state, dirstate.DirState)
720
486
            lines = state.get_lines()
721
 
        finally:
 
487
            self.assertFileEqual(''.join(state.get_lines()),
 
488
                'dirstate')
 
489
            self.check_state_with_reopen(expected_result, state)
 
490
        except:
722
491
            state.unlock()
723
 
        # On win32 you can't read from a locked file, even within the same
724
 
        # process. So we have to unlock and release before we check the file
725
 
        # contents.
726
 
        self.assertFileEqual(''.join(lines), 'dirstate')
727
 
        state.lock_read() # check_state_with_reopen will unlock
728
 
        self.check_state_with_reopen(expected_result, state)
 
492
            raise
729
493
 
730
494
 
731
495
class TestDirStateManipulations(TestCaseWithDirState):
737
501
        try:
738
502
            tree1.add('')
739
503
            revid1 = tree1.commit('foo').encode('utf8')
740
 
            root_id = tree1.get_root_id()
 
504
            root_id = tree1.inventory.root.file_id
741
505
            inv = tree1.inventory
742
506
        finally:
743
507
            tree1.unlock()
758
522
            # This will unlock it
759
523
            self.check_state_with_reopen(expected_result, state)
760
524
 
761
 
    def test_set_state_from_inventory_preserves_hashcache(self):
762
 
        # https://bugs.launchpad.net/bzr/+bug/146176
763
 
        # set_state_from_inventory should preserve the stat and hash value for
764
 
        # workingtree files that are not changed by the inventory.
765
 
 
766
 
        tree = self.make_branch_and_tree('.')
767
 
        # depends on the default format using dirstate...
768
 
        tree.lock_write()
769
 
        try:
770
 
            # make a dirstate with some valid hashcache data
771
 
            # file on disk, but that's not needed for this test
772
 
            foo_contents = 'contents of foo'
773
 
            self.build_tree_contents([('foo', foo_contents)])
774
 
            tree.add('foo', 'foo-id')
775
 
 
776
 
            foo_stat = os.stat('foo')
777
 
            foo_packed = dirstate.pack_stat(foo_stat)
778
 
            foo_sha = osutils.sha_string(foo_contents)
779
 
            foo_size = len(foo_contents)
780
 
 
781
 
            # should not be cached yet, because the file's too fresh
782
 
            self.assertEqual(
783
 
                (('', 'foo', 'foo-id',),
784
 
                 [('f', '', 0, False, dirstate.DirState.NULLSTAT)]),
785
 
                tree._dirstate._get_entry(0, 'foo-id'))
786
 
            # poke in some hashcache information - it wouldn't normally be
787
 
            # stored because it's too fresh
788
 
            tree._dirstate.update_minimal(
789
 
                ('', 'foo', 'foo-id'),
790
 
                'f', False, foo_sha, foo_packed, foo_size, 'foo')
791
 
            # now should be cached
792
 
            self.assertEqual(
793
 
                (('', 'foo', 'foo-id',),
794
 
                 [('f', foo_sha, foo_size, False, foo_packed)]),
795
 
                tree._dirstate._get_entry(0, 'foo-id'))
796
 
 
797
 
            # extract the inventory, and add something to it
798
 
            inv = tree._get_inventory()
799
 
            # should see the file we poked in...
800
 
            self.assertTrue(inv.has_id('foo-id'))
801
 
            self.assertTrue(inv.has_filename('foo'))
802
 
            inv.add_path('bar', 'file', 'bar-id')
803
 
            tree._dirstate._validate()
804
 
            # this used to cause it to lose its hashcache
805
 
            tree._dirstate.set_state_from_inventory(inv)
806
 
            tree._dirstate._validate()
807
 
        finally:
808
 
            tree.unlock()
809
 
 
810
 
        tree.lock_read()
811
 
        try:
812
 
            # now check that the state still has the original hashcache value
813
 
            state = tree._dirstate
814
 
            state._validate()
815
 
            foo_tuple = state._get_entry(0, path_utf8='foo')
816
 
            self.assertEqual(
817
 
                (('', 'foo', 'foo-id',),
818
 
                 [('f', foo_sha, len(foo_contents), False,
819
 
                   dirstate.pack_stat(foo_stat))]),
820
 
                foo_tuple)
821
 
        finally:
822
 
            tree.unlock()
823
 
 
824
 
    def test_set_state_from_inventory_mixed_paths(self):
825
 
        tree1 = self.make_branch_and_tree('tree1')
826
 
        self.build_tree(['tree1/a/', 'tree1/a/b/', 'tree1/a-b/',
827
 
                         'tree1/a/b/foo', 'tree1/a-b/bar'])
828
 
        tree1.lock_write()
829
 
        try:
830
 
            tree1.add(['a', 'a/b', 'a-b', 'a/b/foo', 'a-b/bar'],
831
 
                      ['a-id', 'b-id', 'a-b-id', 'foo-id', 'bar-id'])
832
 
            tree1.commit('rev1', rev_id='rev1')
833
 
            root_id = tree1.get_root_id()
834
 
            inv = tree1.inventory
835
 
        finally:
836
 
            tree1.unlock()
837
 
        expected_result1 = [('', '', root_id, 'd'),
838
 
                            ('', 'a', 'a-id', 'd'),
839
 
                            ('', 'a-b', 'a-b-id', 'd'),
840
 
                            ('a', 'b', 'b-id', 'd'),
841
 
                            ('a/b', 'foo', 'foo-id', 'f'),
842
 
                            ('a-b', 'bar', 'bar-id', 'f'),
843
 
                           ]
844
 
        expected_result2 = [('', '', root_id, 'd'),
845
 
                            ('', 'a', 'a-id', 'd'),
846
 
                            ('', 'a-b', 'a-b-id', 'd'),
847
 
                            ('a-b', 'bar', 'bar-id', 'f'),
848
 
                           ]
849
 
        state = dirstate.DirState.initialize('dirstate')
850
 
        try:
851
 
            state.set_state_from_inventory(inv)
852
 
            values = []
853
 
            for entry in state._iter_entries():
854
 
                values.append(entry[0] + entry[1][0][:1])
855
 
            self.assertEqual(expected_result1, values)
856
 
            del inv['b-id']
857
 
            state.set_state_from_inventory(inv)
858
 
            values = []
859
 
            for entry in state._iter_entries():
860
 
                values.append(entry[0] + entry[1][0][:1])
861
 
            self.assertEqual(expected_result2, values)
862
 
        finally:
863
 
            state.unlock()
864
 
 
865
525
    def test_set_path_id_no_parents(self):
866
526
        """The id of a path can be changed trivally with no parents."""
867
527
        state = dirstate.DirState.initialize('dirstate')
868
528
        try:
869
529
            # check precondition to be sure the state does change appropriately.
870
 
            root_entry = (('', '', 'TREE_ROOT'), [('d', '', 0, False, 'x'*32)])
871
 
            self.assertEqual([root_entry], list(state._iter_entries()))
872
 
            self.assertEqual(root_entry, state._get_entry(0, path_utf8=''))
873
 
            self.assertEqual(root_entry,
874
 
                             state._get_entry(0, fileid_utf8='TREE_ROOT'))
875
 
            self.assertEqual((None, None),
876
 
                             state._get_entry(0, fileid_utf8='second-root-id'))
877
 
            state.set_path_id('', 'second-root-id')
878
 
            new_root_entry = (('', '', 'second-root-id'),
879
 
                              [('d', '', 0, False, 'x'*32)])
880
 
            expected_rows = [new_root_entry]
 
530
            self.assertEqual(
 
531
                [(('', '', 'TREE_ROOT'), [('d', '', 0, False,
 
532
                   'xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx')])],
 
533
                list(state._iter_entries()))
 
534
            state.set_path_id('', 'foobarbaz')
 
535
            expected_rows = [
 
536
                (('', '', 'foobarbaz'), [('d', '', 0, False,
 
537
                   'xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx')])]
881
538
            self.assertEqual(expected_rows, list(state._iter_entries()))
882
 
            self.assertEqual(new_root_entry, state._get_entry(0, path_utf8=''))
883
 
            self.assertEqual(new_root_entry, 
884
 
                             state._get_entry(0, fileid_utf8='second-root-id'))
885
 
            self.assertEqual((None, None),
886
 
                             state._get_entry(0, fileid_utf8='TREE_ROOT'))
887
539
            # should work across save too
888
540
            state.save()
889
541
        finally:
907
559
        state._validate()
908
560
        try:
909
561
            state.set_parent_trees([('parent-revid', rt)], ghosts=[])
910
 
            root_entry = (('', '', 'TREE_ROOT'),
911
 
                          [('d', '', 0, False, 'x'*32),
912
 
                           ('d', '', 0, False, 'parent-revid')])
913
 
            self.assertEqual(root_entry, state._get_entry(0, path_utf8=''))
914
 
            self.assertEqual(root_entry,
915
 
                             state._get_entry(0, fileid_utf8='TREE_ROOT'))
916
 
            self.assertEqual((None, None),
917
 
                             state._get_entry(0, fileid_utf8='Asecond-root-id'))
918
 
            state.set_path_id('', 'Asecond-root-id')
 
562
            state.set_path_id('', 'foobarbaz')
919
563
            state._validate()
920
564
            # now see that it is what we expected
921
 
            old_root_entry = (('', '', 'TREE_ROOT'),
922
 
                              [('a', '', 0, False, ''),
923
 
                               ('d', '', 0, False, 'parent-revid')])
924
 
            new_root_entry = (('', '', 'Asecond-root-id'),
925
 
                              [('d', '', 0, False, ''),
926
 
                               ('a', '', 0, False, '')])
927
 
            expected_rows = [new_root_entry, old_root_entry]
 
565
            expected_rows = [
 
566
                (('', '', 'TREE_ROOT'),
 
567
                    [('a', '', 0, False, ''),
 
568
                     ('d', '', 0, False, 'parent-revid'),
 
569
                     ]),
 
570
                (('', '', 'foobarbaz'),
 
571
                    [('d', '', 0, False, ''),
 
572
                     ('a', '', 0, False, ''),
 
573
                     ]),
 
574
                ]
928
575
            state._validate()
929
576
            self.assertEqual(expected_rows, list(state._iter_entries()))
930
 
            self.assertEqual(new_root_entry, state._get_entry(0, path_utf8=''))
931
 
            self.assertEqual(old_root_entry, state._get_entry(1, path_utf8=''))
932
 
            self.assertEqual((None, None),
933
 
                             state._get_entry(0, fileid_utf8='TREE_ROOT'))
934
 
            self.assertEqual(old_root_entry,
935
 
                             state._get_entry(1, fileid_utf8='TREE_ROOT'))
936
 
            self.assertEqual(new_root_entry,
937
 
                             state._get_entry(0, fileid_utf8='Asecond-root-id'))
938
 
            self.assertEqual((None, None),
939
 
                             state._get_entry(1, fileid_utf8='Asecond-root-id'))
940
577
            # should work across save too
941
578
            state.save()
942
579
        finally:
958
595
        finally:
959
596
            state.unlock()
960
597
 
 
598
 
961
599
    def test_set_parent_trees_no_content(self):
962
600
        # set_parent_trees is a slow but important api to support.
963
601
        tree1 = self.make_branch_and_memory_tree('tree1')
968
606
        finally:
969
607
            tree1.unlock()
970
608
        branch2 = tree1.branch.bzrdir.clone('tree2').open_branch()
971
 
        tree2 = memorytree.MemoryTree.create_on_branch(branch2)
 
609
        tree2 = MemoryTree.create_on_branch(branch2)
972
610
        tree2.lock_write()
973
611
        try:
974
612
            revid2 = tree2.commit('foo')
975
 
            root_id = tree2.get_root_id()
 
613
            root_id = tree2.inventory.root.file_id
976
614
        finally:
977
615
            tree2.unlock()
978
616
        state = dirstate.DirState.initialize('dirstate')
1006
644
            state.set_parent_trees(
1007
645
                ((revid1, tree1.branch.repository.revision_tree(revid1)),
1008
646
                 (revid2, tree2.branch.repository.revision_tree(revid2)),
1009
 
                 ('ghost-rev', tree2.branch.repository.revision_tree(
1010
 
                                   _mod_revision.NULL_REVISION))),
 
647
                 ('ghost-rev', tree2.branch.repository.revision_tree(None))),
1011
648
                ['ghost-rev'])
1012
649
            self.assertEqual([revid1, revid2, 'ghost-rev'],
1013
650
                             state.get_parent_ids())
1017
654
                [(('', '', root_id), [
1018
655
                  ('d', '', 0, False, dirstate.DirState.NULLSTAT),
1019
656
                  ('d', '', 0, False, revid1),
1020
 
                  ('d', '', 0, False, revid1)
 
657
                  ('d', '', 0, False, revid2)
1021
658
                  ])],
1022
659
                list(state._iter_entries()))
1023
660
        finally:
1038
675
        finally:
1039
676
            tree1.unlock()
1040
677
        branch2 = tree1.branch.bzrdir.clone('tree2').open_branch()
1041
 
        tree2 = memorytree.MemoryTree.create_on_branch(branch2)
 
678
        tree2 = MemoryTree.create_on_branch(branch2)
1042
679
        tree2.lock_write()
1043
680
        try:
1044
681
            tree2.put_file_bytes_non_atomic('file-id', 'new file-content')
1045
682
            revid2 = tree2.commit('foo')
1046
 
            root_id = tree2.get_root_id()
 
683
            root_id = tree2.inventory.root.file_id
1047
684
        finally:
1048
685
            tree2.unlock()
1049
686
        # check the layout in memory
1051
688
            (('', '', root_id), [
1052
689
             ('d', '', 0, False, dirstate.DirState.NULLSTAT),
1053
690
             ('d', '', 0, False, revid1.encode('utf8')),
1054
 
             ('d', '', 0, False, revid1.encode('utf8'))
 
691
             ('d', '', 0, False, revid2.encode('utf8'))
1055
692
             ]),
1056
693
            (('', 'a file', 'file-id'), [
1057
694
             ('a', '', 0, False, ''),
1089
726
            (('', '', 'TREE_ROOT'), [
1090
727
             ('d', '', 0, False, dirstate.DirState.NULLSTAT), # current tree
1091
728
             ]),
1092
 
            (('', 'a file', 'a-file-id'), [
 
729
            (('', 'a file', 'a file id'), [
1093
730
             ('f', '1'*20, 19, False, dirstate.pack_stat(stat)), # current tree
1094
731
             ]),
1095
732
            ]
1096
733
        try:
1097
 
            state.add('a file', 'a-file-id', 'file', stat, '1'*20)
 
734
            state.add('a file', 'a file id', 'file', stat, '1'*20)
1098
735
            # having added it, it should be in the output of iter_entries.
1099
736
            self.assertEqual(expected_entries, list(state._iter_entries()))
1100
737
            # saving and reloading should not affect this.
1103
740
            state.unlock()
1104
741
        state = dirstate.DirState.on_file('dirstate')
1105
742
        state.lock_read()
1106
 
        self.addCleanup(state.unlock)
1107
 
        self.assertEqual(expected_entries, list(state._iter_entries()))
 
743
        try:
 
744
            self.assertEqual(expected_entries, list(state._iter_entries()))
 
745
        finally:
 
746
            state.unlock()
1108
747
 
1109
748
    def test_add_path_to_unversioned_directory(self):
1110
749
        """Adding a path to an unversioned directory should error.
1115
754
        """
1116
755
        self.build_tree(['unversioned/', 'unversioned/a file'])
1117
756
        state = dirstate.DirState.initialize('dirstate')
1118
 
        self.addCleanup(state.unlock)
1119
 
        self.assertRaises(errors.NotVersionedError, state.add,
1120
 
                          'unversioned/a file', 'a-file-id', 'file', None, None)
 
757
        try:
 
758
            self.assertRaises(errors.NotVersionedError, state.add,
 
759
                'unversioned/a file', 'a file id', 'file', None, None)
 
760
        finally:
 
761
            state.unlock()
1121
762
 
1122
763
    def test_add_directory_to_root_no_parents_all_data(self):
1123
764
        # The most trivial addition of a dir is when there are no parents and
1143
784
            state.unlock()
1144
785
        state = dirstate.DirState.on_file('dirstate')
1145
786
        state.lock_read()
1146
 
        self.addCleanup(state.unlock)
1147
787
        state._validate()
1148
 
        self.assertEqual(expected_entries, list(state._iter_entries()))
 
788
        try:
 
789
            self.assertEqual(expected_entries, list(state._iter_entries()))
 
790
        finally:
 
791
            state.unlock()
1149
792
 
1150
 
    def _test_add_symlink_to_root_no_parents_all_data(self, link_name, target):
 
793
    def test_add_symlink_to_root_no_parents_all_data(self):
1151
794
        # The most trivial addition of a symlink when there are no parents and
1152
795
        # its in the root and all data about the file is supplied
1153
796
        # bzr doesn't support fake symlinks on windows, yet.
1154
 
        self.requireFeature(tests.SymlinkFeature)
1155
 
        os.symlink(target, link_name)
1156
 
        stat = os.lstat(link_name)
 
797
        if not has_symlinks():
 
798
            raise TestSkipped("No symlink support")
 
799
        os.symlink('target', 'a link')
 
800
        stat = os.lstat('a link')
1157
801
        expected_entries = [
1158
802
            (('', '', 'TREE_ROOT'), [
1159
803
             ('d', '', 0, False, dirstate.DirState.NULLSTAT), # current tree
1160
804
             ]),
1161
 
            (('', link_name.encode('UTF-8'), 'a link id'), [
1162
 
             ('l', target.encode('UTF-8'), stat[6],
1163
 
              False, dirstate.pack_stat(stat)), # current tree
 
805
            (('', 'a link', 'a link id'), [
 
806
             ('l', 'target', 6, False, dirstate.pack_stat(stat)), # current tree
1164
807
             ]),
1165
808
            ]
1166
809
        state = dirstate.DirState.initialize('dirstate')
1167
810
        try:
1168
 
            state.add(link_name, 'a link id', 'symlink', stat,
1169
 
                      target.encode('UTF-8'))
 
811
            state.add('a link', 'a link id', 'symlink', stat, 'target')
1170
812
            # having added it, it should be in the output of iter_entries.
1171
813
            self.assertEqual(expected_entries, list(state._iter_entries()))
1172
814
            # saving and reloading should not affect this.
1175
817
            state.unlock()
1176
818
        state = dirstate.DirState.on_file('dirstate')
1177
819
        state.lock_read()
1178
 
        self.addCleanup(state.unlock)
1179
 
        self.assertEqual(expected_entries, list(state._iter_entries()))
1180
 
 
1181
 
    def test_add_symlink_to_root_no_parents_all_data(self):
1182
 
        self._test_add_symlink_to_root_no_parents_all_data('a link', 'target')
1183
 
 
1184
 
    def test_add_symlink_unicode_to_root_no_parents_all_data(self):
1185
 
        self.requireFeature(tests.UnicodeFilenameFeature)
1186
 
        self._test_add_symlink_to_root_no_parents_all_data(
1187
 
            u'\N{Euro Sign}link', u'targ\N{Euro Sign}et')
 
820
        try:
 
821
            self.assertEqual(expected_entries, list(state._iter_entries()))
 
822
        finally:
 
823
            state.unlock()
1188
824
 
1189
825
    def test_add_directory_and_child_no_parents_all_data(self):
1190
826
        # after adding a directory, we should be able to add children to it.
1198
834
            (('', 'a dir', 'a dir id'), [
1199
835
             ('d', '', 0, False, dirstate.pack_stat(dirstat)), # current tree
1200
836
             ]),
1201
 
            (('a dir', 'a file', 'a-file-id'), [
 
837
            (('a dir', 'a file', 'a file id'), [
1202
838
             ('f', '1'*20, 25, False,
1203
839
              dirstate.pack_stat(filestat)), # current tree details
1204
840
             ]),
1206
842
        state = dirstate.DirState.initialize('dirstate')
1207
843
        try:
1208
844
            state.add('a dir', 'a dir id', 'directory', dirstat, None)
1209
 
            state.add('a dir/a file', 'a-file-id', 'file', filestat, '1'*20)
 
845
            state.add('a dir/a file', 'a file id', 'file', filestat, '1'*20)
1210
846
            # added it, it should be in the output of iter_entries.
1211
847
            self.assertEqual(expected_entries, list(state._iter_entries()))
1212
848
            # saving and reloading should not affect this.
1215
851
            state.unlock()
1216
852
        state = dirstate.DirState.on_file('dirstate')
1217
853
        state.lock_read()
1218
 
        self.addCleanup(state.unlock)
1219
 
        self.assertEqual(expected_entries, list(state._iter_entries()))
 
854
        try:
 
855
            self.assertEqual(expected_entries, list(state._iter_entries()))
 
856
        finally:
 
857
            state.unlock()
1220
858
 
1221
859
    def test_add_tree_reference(self):
1222
860
        # make a dirstate and add a tree reference
1236
874
            state.unlock()
1237
875
        # now check we can read it back
1238
876
        state.lock_read()
1239
 
        self.addCleanup(state.unlock)
1240
877
        state._validate()
1241
 
        entry2 = state._get_entry(0, 'subdir-id', 'subdir')
1242
 
        self.assertEqual(entry, entry2)
1243
 
        self.assertEqual(entry, expected_entry)
1244
 
        # and lookup by id should work too
1245
 
        entry2 = state._get_entry(0, fileid_utf8='subdir-id')
1246
 
        self.assertEqual(entry, expected_entry)
 
878
        try:
 
879
            entry2 = state._get_entry(0, 'subdir-id', 'subdir')
 
880
            self.assertEqual(entry, entry2)
 
881
            self.assertEqual(entry, expected_entry)
 
882
            # and lookup by id should work too
 
883
            entry2 = state._get_entry(0, fileid_utf8='subdir-id')
 
884
            self.assertEqual(entry, expected_entry)
 
885
        finally:
 
886
            state.unlock()
1247
887
 
1248
888
    def test_add_forbidden_names(self):
1249
889
        state = dirstate.DirState.initialize('dirstate')
1253
893
        self.assertRaises(errors.BzrError,
1254
894
            state.add, '..', 'ass-id', 'directory', None, None)
1255
895
 
1256
 
    def test_set_state_with_rename_b_a_bug_395556(self):
1257
 
        # bug 395556 uncovered a bug where the dirstate ends up with a false
1258
 
        # relocation record - in a tree with no parents there should be no
1259
 
        # absent or relocated records. This then leads to further corruption
1260
 
        # when a commit occurs, as the incorrect relocation gathers an
1261
 
        # incorrect absent in tree 1, and future changes go to pot.
1262
 
        tree1 = self.make_branch_and_tree('tree1')
1263
 
        self.build_tree(['tree1/b'])
1264
 
        tree1.lock_write()
1265
 
        try:
1266
 
            tree1.add(['b'], ['b-id'])
1267
 
            root_id = tree1.get_root_id()
1268
 
            inv = tree1.inventory
1269
 
            state = dirstate.DirState.initialize('dirstate')
1270
 
            try:
1271
 
                # Set the initial state with 'b'
1272
 
                state.set_state_from_inventory(inv)
1273
 
                inv.rename('b-id', root_id, 'a')
1274
 
                # Set the new state with 'a', which currently corrupts.
1275
 
                state.set_state_from_inventory(inv)
1276
 
                expected_result1 = [('', '', root_id, 'd'),
1277
 
                                    ('', 'a', 'b-id', 'f'),
1278
 
                                   ]
1279
 
                values = []
1280
 
                for entry in state._iter_entries():
1281
 
                    values.append(entry[0] + entry[1][0][:1])
1282
 
                self.assertEqual(expected_result1, values)
1283
 
            finally:
1284
 
                state.unlock()
1285
 
        finally:
1286
 
            tree1.unlock()
1287
 
 
1288
896
 
1289
897
class TestGetLines(TestCaseWithDirState):
1290
898
 
1503
1111
            state.unlock()
1504
1112
 
1505
1113
 
1506
 
class TestIterChildEntries(TestCaseWithDirState):
1507
 
 
1508
 
    def create_dirstate_with_two_trees(self):
1509
 
        """This dirstate contains multiple files and directories.
1510
 
 
1511
 
         /        a-root-value
1512
 
         a/       a-dir
1513
 
         b/       b-dir
1514
 
         c        c-file
1515
 
         d        d-file
1516
 
         a/e/     e-dir
1517
 
         a/f      f-file
1518
 
         b/g      g-file
1519
 
         b/h\xc3\xa5  h-\xc3\xa5-file  #This is u'\xe5' encoded into utf-8
1520
 
 
1521
 
        Notice that a/e is an empty directory.
1522
 
 
1523
 
        There is one parent tree, which has the same shape with the following variations:
1524
 
        b/g in the parent is gone.
1525
 
        b/h in the parent has a different id
1526
 
        b/i is new in the parent
1527
 
        c is renamed to b/j in the parent
1528
 
 
1529
 
        :return: The dirstate, still write-locked.
1530
 
        """
1531
 
        packed_stat = 'AAAAREUHaIpFB2iKAAADAQAtkqUAAIGk'
1532
 
        null_sha = 'xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx'
1533
 
        NULL_PARENT_DETAILS = dirstate.DirState.NULL_PARENT_DETAILS
1534
 
        root_entry = ('', '', 'a-root-value'), [
1535
 
            ('d', '', 0, False, packed_stat),
1536
 
            ('d', '', 0, False, 'parent-revid'),
1537
 
            ]
1538
 
        a_entry = ('', 'a', 'a-dir'), [
1539
 
            ('d', '', 0, False, packed_stat),
1540
 
            ('d', '', 0, False, 'parent-revid'),
1541
 
            ]
1542
 
        b_entry = ('', 'b', 'b-dir'), [
1543
 
            ('d', '', 0, False, packed_stat),
1544
 
            ('d', '', 0, False, 'parent-revid'),
1545
 
            ]
1546
 
        c_entry = ('', 'c', 'c-file'), [
1547
 
            ('f', null_sha, 10, False, packed_stat),
1548
 
            ('r', 'b/j', 0, False, ''),
1549
 
            ]
1550
 
        d_entry = ('', 'd', 'd-file'), [
1551
 
            ('f', null_sha, 20, False, packed_stat),
1552
 
            ('f', 'd', 20, False, 'parent-revid'),
1553
 
            ]
1554
 
        e_entry = ('a', 'e', 'e-dir'), [
1555
 
            ('d', '', 0, False, packed_stat),
1556
 
            ('d', '', 0, False, 'parent-revid'),
1557
 
            ]
1558
 
        f_entry = ('a', 'f', 'f-file'), [
1559
 
            ('f', null_sha, 30, False, packed_stat),
1560
 
            ('f', 'f', 20, False, 'parent-revid'),
1561
 
            ]
1562
 
        g_entry = ('b', 'g', 'g-file'), [
1563
 
            ('f', null_sha, 30, False, packed_stat),
1564
 
            NULL_PARENT_DETAILS,
1565
 
            ]
1566
 
        h_entry1 = ('b', 'h\xc3\xa5', 'h-\xc3\xa5-file1'), [
1567
 
            ('f', null_sha, 40, False, packed_stat),
1568
 
            NULL_PARENT_DETAILS,
1569
 
            ]
1570
 
        h_entry2 = ('b', 'h\xc3\xa5', 'h-\xc3\xa5-file2'), [
1571
 
            NULL_PARENT_DETAILS,
1572
 
            ('f', 'h', 20, False, 'parent-revid'),
1573
 
            ]
1574
 
        i_entry = ('b', 'i', 'i-file'), [
1575
 
            NULL_PARENT_DETAILS,
1576
 
            ('f', 'h', 20, False, 'parent-revid'),
1577
 
            ]
1578
 
        j_entry = ('b', 'j', 'c-file'), [
1579
 
            ('r', 'c', 0, False, ''),
1580
 
            ('f', 'j', 20, False, 'parent-revid'),
1581
 
            ]
1582
 
        dirblocks = []
1583
 
        dirblocks.append(('', [root_entry]))
1584
 
        dirblocks.append(('', [a_entry, b_entry, c_entry, d_entry]))
1585
 
        dirblocks.append(('a', [e_entry, f_entry]))
1586
 
        dirblocks.append(('b', [g_entry, h_entry1, h_entry2, i_entry, j_entry]))
1587
 
        state = dirstate.DirState.initialize('dirstate')
1588
 
        state._validate()
1589
 
        try:
1590
 
            state._set_data(['parent'], dirblocks)
1591
 
        except:
1592
 
            state.unlock()
1593
 
            raise
1594
 
        return state, dirblocks
1595
 
 
1596
 
    def test_iter_children_b(self):
1597
 
        state, dirblocks = self.create_dirstate_with_two_trees()
1598
 
        self.addCleanup(state.unlock)
1599
 
        expected_result = []
1600
 
        expected_result.append(dirblocks[3][1][2]) # h2
1601
 
        expected_result.append(dirblocks[3][1][3]) # i
1602
 
        expected_result.append(dirblocks[3][1][4]) # j
1603
 
        self.assertEqual(expected_result,
1604
 
            list(state._iter_child_entries(1, 'b')))
1605
 
 
1606
 
    def test_iter_child_root(self):
1607
 
        state, dirblocks = self.create_dirstate_with_two_trees()
1608
 
        self.addCleanup(state.unlock)
1609
 
        expected_result = []
1610
 
        expected_result.append(dirblocks[1][1][0]) # a
1611
 
        expected_result.append(dirblocks[1][1][1]) # b
1612
 
        expected_result.append(dirblocks[1][1][3]) # d
1613
 
        expected_result.append(dirblocks[2][1][0]) # e
1614
 
        expected_result.append(dirblocks[2][1][1]) # f
1615
 
        expected_result.append(dirblocks[3][1][2]) # h2
1616
 
        expected_result.append(dirblocks[3][1][3]) # i
1617
 
        expected_result.append(dirblocks[3][1][4]) # j
1618
 
        self.assertEqual(expected_result,
1619
 
            list(state._iter_child_entries(1, '')))
1620
 
 
1621
 
 
1622
 
class TestDirstateSortOrder(tests.TestCaseWithTransport):
 
1114
class TestDirstateSortOrder(TestCaseWithTransport):
1623
1115
    """Test that DirState adds entries in the right order."""
1624
1116
 
1625
1117
    def test_add_sorting(self):
1674
1166
 
1675
1167
        # *really* cheesy way to just get an empty tree
1676
1168
        repo = self.make_repository('repo')
1677
 
        empty_tree = repo.revision_tree(_mod_revision.NULL_REVISION)
 
1169
        empty_tree = repo.revision_tree(None)
1678
1170
        state.set_parent_trees([('null:', empty_tree)], [])
1679
1171
 
1680
1172
        dirblock_names = [d[0] for d in state._dirblocks]
1684
1176
class InstrumentedDirState(dirstate.DirState):
1685
1177
    """An DirState with instrumented sha1 functionality."""
1686
1178
 
1687
 
    def __init__(self, path, sha1_provider):
1688
 
        super(InstrumentedDirState, self).__init__(path, sha1_provider)
 
1179
    def __init__(self, path):
 
1180
        super(InstrumentedDirState, self).__init__(path)
1689
1181
        self._time_offset = 0
1690
1182
        self._log = []
1691
 
        # member is dynamically set in DirState.__init__ to turn on trace
1692
 
        self._sha1_provider = sha1_provider
1693
 
        self._sha1_file = self._sha1_file_and_log
1694
1183
 
1695
1184
    def _sha_cutoff_time(self):
1696
1185
        timestamp = super(InstrumentedDirState, self)._sha_cutoff_time()
1697
1186
        self._cutoff_time = timestamp + self._time_offset
1698
1187
 
1699
 
    def _sha1_file_and_log(self, abspath):
 
1188
    def _sha1_file(self, abspath, entry):
1700
1189
        self._log.append(('sha1', abspath))
1701
 
        return self._sha1_provider.sha1(abspath)
 
1190
        return super(InstrumentedDirState, self)._sha1_file(abspath, entry)
1702
1191
 
1703
1192
    def _read_link(self, abspath, old_link):
1704
1193
        self._log.append(('read_link', abspath, old_link))
1735
1224
        self.st_ino = ino
1736
1225
        self.st_mode = mode
1737
1226
 
1738
 
    @staticmethod
1739
 
    def from_stat(st):
1740
 
        return _FakeStat(st.st_size, st.st_mtime, st.st_ctime, st.st_dev,
1741
 
            st.st_ino, st.st_mode)
1742
 
 
1743
 
 
1744
 
class TestPackStat(tests.TestCaseWithTransport):
 
1227
 
 
1228
class TestUpdateEntry(TestCaseWithDirState):
 
1229
    """Test the DirState.update_entry functions"""
 
1230
 
 
1231
    def get_state_with_a(self):
 
1232
        """Create a DirState tracking a single object named 'a'"""
 
1233
        state = InstrumentedDirState.initialize('dirstate')
 
1234
        self.addCleanup(state.unlock)
 
1235
        state.add('a', 'a-id', 'file', None, '')
 
1236
        entry = state._get_entry(0, path_utf8='a')
 
1237
        return state, entry
 
1238
 
 
1239
    def test_update_entry(self):
 
1240
        state, entry = self.get_state_with_a()
 
1241
        self.build_tree(['a'])
 
1242
        # Add one where we don't provide the stat or sha already
 
1243
        self.assertEqual(('', 'a', 'a-id'), entry[0])
 
1244
        self.assertEqual([('f', '', 0, False, dirstate.DirState.NULLSTAT)],
 
1245
                         entry[1])
 
1246
        # Flush the buffers to disk
 
1247
        state.save()
 
1248
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
 
1249
                         state._dirblock_state)
 
1250
 
 
1251
        stat_value = os.lstat('a')
 
1252
        packed_stat = dirstate.pack_stat(stat_value)
 
1253
        link_or_sha1 = state.update_entry(entry, abspath='a',
 
1254
                                          stat_value=stat_value)
 
1255
        self.assertEqual('b50e5406bb5e153ebbeb20268fcf37c87e1ecfb6',
 
1256
                         link_or_sha1)
 
1257
 
 
1258
        # The dirblock entry should be updated with the new info
 
1259
        self.assertEqual([('f', link_or_sha1, 14, False, packed_stat)],
 
1260
                         entry[1])
 
1261
        self.assertEqual(dirstate.DirState.IN_MEMORY_MODIFIED,
 
1262
                         state._dirblock_state)
 
1263
        mode = stat_value.st_mode
 
1264
        self.assertEqual([('sha1', 'a'), ('is_exec', mode, False)], state._log)
 
1265
 
 
1266
        state.save()
 
1267
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
 
1268
                         state._dirblock_state)
 
1269
 
 
1270
        # If we do it again right away, we don't know if the file has changed
 
1271
        # so we will re-read the file. Roll the clock back so the file is
 
1272
        # guaranteed to look too new.
 
1273
        state.adjust_time(-10)
 
1274
 
 
1275
        link_or_sha1 = state.update_entry(entry, abspath='a',
 
1276
                                          stat_value=stat_value)
 
1277
        self.assertEqual([('sha1', 'a'), ('is_exec', mode, False),
 
1278
                          ('sha1', 'a'), ('is_exec', mode, False),
 
1279
                         ], state._log)
 
1280
        self.assertEqual('b50e5406bb5e153ebbeb20268fcf37c87e1ecfb6',
 
1281
                         link_or_sha1)
 
1282
        self.assertEqual(dirstate.DirState.IN_MEMORY_MODIFIED,
 
1283
                         state._dirblock_state)
 
1284
        state.save()
 
1285
 
 
1286
        # However, if we move the clock forward so the file is considered
 
1287
        # "stable", it should just returned the cached value.
 
1288
        state.adjust_time(20)
 
1289
        link_or_sha1 = state.update_entry(entry, abspath='a',
 
1290
                                          stat_value=stat_value)
 
1291
        self.assertEqual('b50e5406bb5e153ebbeb20268fcf37c87e1ecfb6',
 
1292
                         link_or_sha1)
 
1293
        self.assertEqual([('sha1', 'a'), ('is_exec', mode, False),
 
1294
                          ('sha1', 'a'), ('is_exec', mode, False),
 
1295
                         ], state._log)
 
1296
 
 
1297
    def test_update_entry_no_stat_value(self):
 
1298
        """Passing the stat_value is optional."""
 
1299
        state, entry = self.get_state_with_a()
 
1300
        state.adjust_time(-10) # Make sure the file looks new
 
1301
        self.build_tree(['a'])
 
1302
        # Add one where we don't provide the stat or sha already
 
1303
        link_or_sha1 = state.update_entry(entry, abspath='a')
 
1304
        self.assertEqual('b50e5406bb5e153ebbeb20268fcf37c87e1ecfb6',
 
1305
                         link_or_sha1)
 
1306
        stat_value = os.lstat('a')
 
1307
        self.assertEqual([('lstat', 'a'), ('sha1', 'a'),
 
1308
                          ('is_exec', stat_value.st_mode, False),
 
1309
                         ], state._log)
 
1310
 
 
1311
    def test_update_entry_symlink(self):
 
1312
        """Update entry should read symlinks."""
 
1313
        if not osutils.has_symlinks():
 
1314
            # PlatformDeficiency / TestSkipped
 
1315
            raise TestSkipped("No symlink support")
 
1316
        state, entry = self.get_state_with_a()
 
1317
        state.save()
 
1318
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
 
1319
                         state._dirblock_state)
 
1320
        os.symlink('target', 'a')
 
1321
 
 
1322
        state.adjust_time(-10) # Make the symlink look new
 
1323
        stat_value = os.lstat('a')
 
1324
        packed_stat = dirstate.pack_stat(stat_value)
 
1325
        link_or_sha1 = state.update_entry(entry, abspath='a',
 
1326
                                          stat_value=stat_value)
 
1327
        self.assertEqual('target', link_or_sha1)
 
1328
        self.assertEqual([('read_link', 'a', '')], state._log)
 
1329
        # Dirblock is updated
 
1330
        self.assertEqual([('l', link_or_sha1, 6, False, packed_stat)],
 
1331
                         entry[1])
 
1332
        self.assertEqual(dirstate.DirState.IN_MEMORY_MODIFIED,
 
1333
                         state._dirblock_state)
 
1334
 
 
1335
        # Because the stat_value looks new, we should re-read the target
 
1336
        link_or_sha1 = state.update_entry(entry, abspath='a',
 
1337
                                          stat_value=stat_value)
 
1338
        self.assertEqual('target', link_or_sha1)
 
1339
        self.assertEqual([('read_link', 'a', ''),
 
1340
                          ('read_link', 'a', 'target'),
 
1341
                         ], state._log)
 
1342
        state.adjust_time(+20) # Skip into the future, all files look old
 
1343
        link_or_sha1 = state.update_entry(entry, abspath='a',
 
1344
                                          stat_value=stat_value)
 
1345
        self.assertEqual('target', link_or_sha1)
 
1346
        # There should not be a new read_link call.
 
1347
        # (this is a weak assertion, because read_link is fairly inexpensive,
 
1348
        # versus the number of symlinks that we would have)
 
1349
        self.assertEqual([('read_link', 'a', ''),
 
1350
                          ('read_link', 'a', 'target'),
 
1351
                         ], state._log)
 
1352
 
 
1353
    def test_update_entry_dir(self):
 
1354
        state, entry = self.get_state_with_a()
 
1355
        self.build_tree(['a/'])
 
1356
        self.assertIs(None, state.update_entry(entry, 'a'))
 
1357
 
 
1358
    def create_and_test_file(self, state, entry):
 
1359
        """Create a file at 'a' and verify the state finds it.
 
1360
 
 
1361
        The state should already be versioning *something* at 'a'. This makes
 
1362
        sure that state.update_entry recognizes it as a file.
 
1363
        """
 
1364
        self.build_tree(['a'])
 
1365
        stat_value = os.lstat('a')
 
1366
        packed_stat = dirstate.pack_stat(stat_value)
 
1367
 
 
1368
        link_or_sha1 = state.update_entry(entry, abspath='a')
 
1369
        self.assertEqual('b50e5406bb5e153ebbeb20268fcf37c87e1ecfb6',
 
1370
                         link_or_sha1)
 
1371
        self.assertEqual([('f', link_or_sha1, 14, False, packed_stat)],
 
1372
                         entry[1])
 
1373
        return packed_stat
 
1374
 
 
1375
    def create_and_test_dir(self, state, entry):
 
1376
        """Create a directory at 'a' and verify the state finds it.
 
1377
 
 
1378
        The state should already be versioning *something* at 'a'. This makes
 
1379
        sure that state.update_entry recognizes it as a directory.
 
1380
        """
 
1381
        self.build_tree(['a/'])
 
1382
        stat_value = os.lstat('a')
 
1383
        packed_stat = dirstate.pack_stat(stat_value)
 
1384
 
 
1385
        link_or_sha1 = state.update_entry(entry, abspath='a')
 
1386
        self.assertIs(None, link_or_sha1)
 
1387
        self.assertEqual([('d', '', 0, False, packed_stat)], entry[1])
 
1388
 
 
1389
        return packed_stat
 
1390
 
 
1391
    def create_and_test_symlink(self, state, entry):
 
1392
        """Create a symlink at 'a' and verify the state finds it.
 
1393
 
 
1394
        The state should already be versioning *something* at 'a'. This makes
 
1395
        sure that state.update_entry recognizes it as a symlink.
 
1396
 
 
1397
        This should not be called if this platform does not have symlink
 
1398
        support.
 
1399
        """
 
1400
        # caller should care about skipping test on platforms without symlinks
 
1401
        os.symlink('path/to/foo', 'a')
 
1402
 
 
1403
        stat_value = os.lstat('a')
 
1404
        packed_stat = dirstate.pack_stat(stat_value)
 
1405
 
 
1406
        link_or_sha1 = state.update_entry(entry, abspath='a')
 
1407
        self.assertEqual('path/to/foo', link_or_sha1)
 
1408
        self.assertEqual([('l', 'path/to/foo', 11, False, packed_stat)],
 
1409
                         entry[1])
 
1410
        return packed_stat
 
1411
 
 
1412
    def test_update_missing_file(self):
 
1413
        state, entry = self.get_state_with_a()
 
1414
        packed_stat = self.create_and_test_file(state, entry)
 
1415
        # Now if we delete the file, update_entry should recover and
 
1416
        # return None.
 
1417
        os.remove('a')
 
1418
        self.assertIs(None, state.update_entry(entry, abspath='a'))
 
1419
        # And the record shouldn't be changed.
 
1420
        digest = 'b50e5406bb5e153ebbeb20268fcf37c87e1ecfb6'
 
1421
        self.assertEqual([('f', digest, 14, False, packed_stat)],
 
1422
                         entry[1])
 
1423
 
 
1424
    def test_update_missing_dir(self):
 
1425
        state, entry = self.get_state_with_a()
 
1426
        packed_stat = self.create_and_test_dir(state, entry)
 
1427
        # Now if we delete the directory, update_entry should recover and
 
1428
        # return None.
 
1429
        os.rmdir('a')
 
1430
        self.assertIs(None, state.update_entry(entry, abspath='a'))
 
1431
        self.assertEqual([('d', '', 0, False, packed_stat)], entry[1])
 
1432
 
 
1433
    def test_update_missing_symlink(self):
 
1434
        if not osutils.has_symlinks():
 
1435
            # PlatformDeficiency / TestSkipped
 
1436
            raise TestSkipped("No symlink support")
 
1437
        state, entry = self.get_state_with_a()
 
1438
        packed_stat = self.create_and_test_symlink(state, entry)
 
1439
        os.remove('a')
 
1440
        self.assertIs(None, state.update_entry(entry, abspath='a'))
 
1441
        # And the record shouldn't be changed.
 
1442
        self.assertEqual([('l', 'path/to/foo', 11, False, packed_stat)],
 
1443
                         entry[1])
 
1444
 
 
1445
    def test_update_file_to_dir(self):
 
1446
        """If a file changes to a directory we return None for the sha.
 
1447
        We also update the inventory record.
 
1448
        """
 
1449
        state, entry = self.get_state_with_a()
 
1450
        self.create_and_test_file(state, entry)
 
1451
        os.remove('a')
 
1452
        self.create_and_test_dir(state, entry)
 
1453
 
 
1454
    def test_update_file_to_symlink(self):
 
1455
        """File becomes a symlink"""
 
1456
        if not osutils.has_symlinks():
 
1457
            # PlatformDeficiency / TestSkipped
 
1458
            raise TestSkipped("No symlink support")
 
1459
        state, entry = self.get_state_with_a()
 
1460
        self.create_and_test_file(state, entry)
 
1461
        os.remove('a')
 
1462
        self.create_and_test_symlink(state, entry)
 
1463
 
 
1464
    def test_update_dir_to_file(self):
 
1465
        """Directory becoming a file updates the entry."""
 
1466
        state, entry = self.get_state_with_a()
 
1467
        self.create_and_test_dir(state, entry)
 
1468
        os.rmdir('a')
 
1469
        self.create_and_test_file(state, entry)
 
1470
 
 
1471
    def test_update_dir_to_symlink(self):
 
1472
        """Directory becomes a symlink"""
 
1473
        if not osutils.has_symlinks():
 
1474
            # PlatformDeficiency / TestSkipped
 
1475
            raise TestSkipped("No symlink support")
 
1476
        state, entry = self.get_state_with_a()
 
1477
        self.create_and_test_dir(state, entry)
 
1478
        os.rmdir('a')
 
1479
        self.create_and_test_symlink(state, entry)
 
1480
 
 
1481
    def test_update_symlink_to_file(self):
 
1482
        """Symlink becomes a file"""
 
1483
        if not has_symlinks():
 
1484
            raise TestSkipped("No symlink support")
 
1485
        state, entry = self.get_state_with_a()
 
1486
        self.create_and_test_symlink(state, entry)
 
1487
        os.remove('a')
 
1488
        self.create_and_test_file(state, entry)
 
1489
 
 
1490
    def test_update_symlink_to_dir(self):
 
1491
        """Symlink becomes a directory"""
 
1492
        if not has_symlinks():
 
1493
            raise TestSkipped("No symlink support")
 
1494
        state, entry = self.get_state_with_a()
 
1495
        self.create_and_test_symlink(state, entry)
 
1496
        os.remove('a')
 
1497
        self.create_and_test_dir(state, entry)
 
1498
 
 
1499
    def test__is_executable_win32(self):
 
1500
        state, entry = self.get_state_with_a()
 
1501
        self.build_tree(['a'])
 
1502
 
 
1503
        # Make sure we are using the win32 implementation of _is_executable
 
1504
        state._is_executable = state._is_executable_win32
 
1505
 
 
1506
        # The file on disk is not executable, but we are marking it as though
 
1507
        # it is. With _is_executable_win32 we ignore what is on disk.
 
1508
        entry[1][0] = ('f', '', 0, True, dirstate.DirState.NULLSTAT)
 
1509
 
 
1510
        stat_value = os.lstat('a')
 
1511
        packed_stat = dirstate.pack_stat(stat_value)
 
1512
 
 
1513
        state.adjust_time(-10) # Make sure everything is new
 
1514
        # Make sure it wants to kkkkkkkk
 
1515
        state.update_entry(entry, abspath='a', stat_value=stat_value)
 
1516
 
 
1517
        # The row is updated, but the executable bit stays set.
 
1518
        digest = 'b50e5406bb5e153ebbeb20268fcf37c87e1ecfb6'
 
1519
        self.assertEqual([('f', digest, 14, True, packed_stat)], entry[1])
 
1520
 
 
1521
 
 
1522
class TestPackStat(TestCaseWithTransport):
1745
1523
 
1746
1524
    def assertPackStat(self, expected, stat_value):
1747
1525
        """Check the packed and serialized form of a stat value."""
1793
1571
        self.assertPackStat('AAAbWEXm4FxF5uBmAAADCQBjLNIAAIGk', st)
1794
1572
 
1795
1573
 
1796
 
class TestBisect(TestCaseWithDirState):
 
1574
class TestBisect(TestCaseWithTransport):
1797
1575
    """Test the ability to bisect into the disk format."""
1798
1576
 
 
1577
    def create_basic_dirstate(self):
 
1578
        """Create a dirstate with a few files and directories.
 
1579
 
 
1580
            a
 
1581
            b/
 
1582
              c
 
1583
              d/
 
1584
                e
 
1585
            f
 
1586
        """
 
1587
        tree = self.make_branch_and_tree('tree')
 
1588
        paths = ['a', 'b/', 'b/c', 'b/d/', 'b/d/e', 'f']
 
1589
        file_ids = ['a-id', 'b-id', 'c-id', 'd-id', 'e-id', 'f-id']
 
1590
        self.build_tree(['tree/' + p for p in paths])
 
1591
        tree.set_root_id('TREE_ROOT')
 
1592
        tree.add([p.rstrip('/') for p in paths], file_ids)
 
1593
        tree.commit('initial', rev_id='rev-1')
 
1594
        revision_id = 'rev-1'
 
1595
        # a_packed_stat = dirstate.pack_stat(os.stat('tree/a'))
 
1596
        t = self.get_transport().clone('tree')
 
1597
        a_text = t.get_bytes('a')
 
1598
        a_sha = osutils.sha_string(a_text)
 
1599
        a_len = len(a_text)
 
1600
        # b_packed_stat = dirstate.pack_stat(os.stat('tree/b'))
 
1601
        # c_packed_stat = dirstate.pack_stat(os.stat('tree/b/c'))
 
1602
        c_text = t.get_bytes('b/c')
 
1603
        c_sha = osutils.sha_string(c_text)
 
1604
        c_len = len(c_text)
 
1605
        # d_packed_stat = dirstate.pack_stat(os.stat('tree/b/d'))
 
1606
        # e_packed_stat = dirstate.pack_stat(os.stat('tree/b/d/e'))
 
1607
        e_text = t.get_bytes('b/d/e')
 
1608
        e_sha = osutils.sha_string(e_text)
 
1609
        e_len = len(e_text)
 
1610
        # f_packed_stat = dirstate.pack_stat(os.stat('tree/f'))
 
1611
        f_text = t.get_bytes('f')
 
1612
        f_sha = osutils.sha_string(f_text)
 
1613
        f_len = len(f_text)
 
1614
        null_stat = dirstate.DirState.NULLSTAT
 
1615
        expected = {
 
1616
            '':(('', '', 'TREE_ROOT'), [
 
1617
                  ('d', '', 0, False, null_stat),
 
1618
                  ('d', '', 0, False, revision_id),
 
1619
                ]),
 
1620
            'a':(('', 'a', 'a-id'), [
 
1621
                   ('f', '', 0, False, null_stat),
 
1622
                   ('f', a_sha, a_len, False, revision_id),
 
1623
                 ]),
 
1624
            'b':(('', 'b', 'b-id'), [
 
1625
                  ('d', '', 0, False, null_stat),
 
1626
                  ('d', '', 0, False, revision_id),
 
1627
                 ]),
 
1628
            'b/c':(('b', 'c', 'c-id'), [
 
1629
                    ('f', '', 0, False, null_stat),
 
1630
                    ('f', c_sha, c_len, False, revision_id),
 
1631
                   ]),
 
1632
            'b/d':(('b', 'd', 'd-id'), [
 
1633
                    ('d', '', 0, False, null_stat),
 
1634
                    ('d', '', 0, False, revision_id),
 
1635
                   ]),
 
1636
            'b/d/e':(('b/d', 'e', 'e-id'), [
 
1637
                      ('f', '', 0, False, null_stat),
 
1638
                      ('f', e_sha, e_len, False, revision_id),
 
1639
                     ]),
 
1640
            'f':(('', 'f', 'f-id'), [
 
1641
                  ('f', '', 0, False, null_stat),
 
1642
                  ('f', f_sha, f_len, False, revision_id),
 
1643
                 ]),
 
1644
        }
 
1645
        state = dirstate.DirState.from_tree(tree, 'dirstate')
 
1646
        try:
 
1647
            state.save()
 
1648
        finally:
 
1649
            state.unlock()
 
1650
        # Use a different object, to make sure nothing is pre-cached in memory.
 
1651
        state = dirstate.DirState.on_file('dirstate')
 
1652
        state.lock_read()
 
1653
        self.addCleanup(state.unlock)
 
1654
        self.assertEqual(dirstate.DirState.NOT_IN_MEMORY,
 
1655
                         state._dirblock_state)
 
1656
        # This is code is only really tested if we actually have to make more
 
1657
        # than one read, so set the page size to something smaller.
 
1658
        # We want it to contain about 2.2 records, so that we have a couple
 
1659
        # records that we can read per attempt
 
1660
        state._bisect_page_size = 200
 
1661
        return tree, state, expected
 
1662
 
 
1663
    def create_duplicated_dirstate(self):
 
1664
        """Create a dirstate with a deleted and added entries.
 
1665
 
 
1666
        This grabs a basic_dirstate, and then removes and re adds every entry
 
1667
        with a new file id.
 
1668
        """
 
1669
        tree, state, expected = self.create_basic_dirstate()
 
1670
        # Now we will just remove and add every file so we get an extra entry
 
1671
        # per entry. Unversion in reverse order so we handle subdirs
 
1672
        tree.unversion(['f-id', 'e-id', 'd-id', 'c-id', 'b-id', 'a-id'])
 
1673
        tree.add(['a', 'b', 'b/c', 'b/d', 'b/d/e', 'f'],
 
1674
                 ['a-id2', 'b-id2', 'c-id2', 'd-id2', 'e-id2', 'f-id2'])
 
1675
 
 
1676
        # Update the expected dictionary.
 
1677
        for path in ['a', 'b', 'b/c', 'b/d', 'b/d/e', 'f']:
 
1678
            orig = expected[path]
 
1679
            path2 = path + '2'
 
1680
            # This record was deleted in the current tree
 
1681
            expected[path] = (orig[0], [dirstate.DirState.NULL_PARENT_DETAILS,
 
1682
                                        orig[1][1]])
 
1683
            new_key = (orig[0][0], orig[0][1], orig[0][2]+'2')
 
1684
            # And didn't exist in the basis tree
 
1685
            expected[path2] = (new_key, [orig[1][0],
 
1686
                                         dirstate.DirState.NULL_PARENT_DETAILS])
 
1687
 
 
1688
        # We will replace the 'dirstate' file underneath 'state', but that is
 
1689
        # okay as lock as we unlock 'state' first.
 
1690
        state.unlock()
 
1691
        try:
 
1692
            new_state = dirstate.DirState.from_tree(tree, 'dirstate')
 
1693
            try:
 
1694
                new_state.save()
 
1695
            finally:
 
1696
                new_state.unlock()
 
1697
        finally:
 
1698
            # But we need to leave state in a read-lock because we already have
 
1699
            # a cleanup scheduled
 
1700
            state.lock_read()
 
1701
        return tree, state, expected
 
1702
 
 
1703
    def create_renamed_dirstate(self):
 
1704
        """Create a dirstate with a few internal renames.
 
1705
 
 
1706
        This takes the basic dirstate, and moves the paths around.
 
1707
        """
 
1708
        tree, state, expected = self.create_basic_dirstate()
 
1709
        # Rename a file
 
1710
        tree.rename_one('a', 'b/g')
 
1711
        # And a directory
 
1712
        tree.rename_one('b/d', 'h')
 
1713
 
 
1714
        old_a = expected['a']
 
1715
        expected['a'] = (old_a[0], [('r', 'b/g', 0, False, ''), old_a[1][1]])
 
1716
        expected['b/g'] = (('b', 'g', 'a-id'), [old_a[1][0],
 
1717
                                                ('r', 'a', 0, False, '')])
 
1718
        old_d = expected['b/d']
 
1719
        expected['b/d'] = (old_d[0], [('r', 'h', 0, False, ''), old_d[1][1]])
 
1720
        expected['h'] = (('', 'h', 'd-id'), [old_d[1][0],
 
1721
                                             ('r', 'b/d', 0, False, '')])
 
1722
 
 
1723
        old_e = expected['b/d/e']
 
1724
        expected['b/d/e'] = (old_e[0], [('r', 'h/e', 0, False, ''),
 
1725
                             old_e[1][1]])
 
1726
        expected['h/e'] = (('h', 'e', 'e-id'), [old_e[1][0],
 
1727
                                                ('r', 'b/d/e', 0, False, '')])
 
1728
 
 
1729
        state.unlock()
 
1730
        try:
 
1731
            new_state = dirstate.DirState.from_tree(tree, 'dirstate')
 
1732
            try:
 
1733
                new_state.save()
 
1734
            finally:
 
1735
                new_state.unlock()
 
1736
        finally:
 
1737
            state.lock_read()
 
1738
        return tree, state, expected
 
1739
 
1799
1740
    def assertBisect(self, expected_map, map_keys, state, paths):
1800
1741
        """Assert that bisecting for paths returns the right result.
1801
1742
 
1806
1747
                      (dir, name) tuples, and sorted according to how _bisect
1807
1748
                      requires.
1808
1749
        """
1809
 
        result = state._bisect(paths)
 
1750
        dir_names = sorted(osutils.split(p) for p in paths)
 
1751
        result = state._bisect(dir_names)
1810
1752
        # For now, results are just returned in whatever order we read them.
1811
1753
        # We could sort by (dir, name, file_id) or something like that, but in
1812
1754
        # the end it would still be fairly arbitrary, and we don't want the
1813
1755
        # extra overhead if we can avoid it. So sort everything to make sure
1814
1756
        # equality is true
1815
 
        self.assertEqual(len(map_keys), len(paths))
 
1757
        assert len(map_keys) == len(dir_names)
1816
1758
        expected = {}
1817
 
        for path, keys in zip(paths, map_keys):
 
1759
        for dir_name, keys in zip(dir_names, map_keys):
1818
1760
            if keys is None:
1819
1761
                # This should not be present in the output
1820
1762
                continue
1821
 
            expected[path] = sorted(expected_map[k] for k in keys)
 
1763
            expected[dir_name] = sorted(expected_map[k] for k in keys)
1822
1764
 
1823
 
        # The returned values are just arranged randomly based on when they
1824
 
        # were read, for testing, make sure it is properly sorted.
1825
 
        for path in result:
1826
 
            result[path].sort()
 
1765
        for dir_name in result:
 
1766
            result[dir_name].sort()
1827
1767
 
1828
1768
        self.assertEqual(expected, result)
1829
1769
 
1837
1777
        :param paths: A list of directories
1838
1778
        """
1839
1779
        result = state._bisect_dirblocks(paths)
1840
 
        self.assertEqual(len(map_keys), len(paths))
 
1780
        assert len(map_keys) == len(paths)
 
1781
 
1841
1782
        expected = {}
1842
1783
        for path, keys in zip(paths, map_keys):
1843
1784
            if keys is None:
1865
1806
            dir_name_id, trees_info = entry
1866
1807
            expected[dir_name_id] = trees_info
1867
1808
 
1868
 
        result = state._bisect_recursive(paths)
 
1809
        dir_names = sorted(osutils.split(p) for p in paths)
 
1810
        result = state._bisect_recursive(dir_names)
1869
1811
 
1870
1812
        self.assertEqual(expected, result)
1871
1813
 
1880
1822
        self.assertBisect(expected, [['b/c']], state, ['b/c'])
1881
1823
        self.assertBisect(expected, [['b/d']], state, ['b/d'])
1882
1824
        self.assertBisect(expected, [['b/d/e']], state, ['b/d/e'])
1883
 
        self.assertBisect(expected, [['b-c']], state, ['b-c'])
1884
1825
        self.assertBisect(expected, [['f']], state, ['f'])
1885
1826
 
1886
1827
    def test_bisect_multi(self):
1889
1830
        # Bisect should be capable of finding multiple entries at the same time
1890
1831
        self.assertBisect(expected, [['a'], ['b'], ['f']],
1891
1832
                          state, ['a', 'b', 'f'])
 
1833
        # ('', 'f') sorts before the others
1892
1834
        self.assertBisect(expected, [['f'], ['b/d'], ['b/d/e']],
1893
 
                          state, ['f', 'b/d', 'b/d/e'])
1894
 
        self.assertBisect(expected, [['b'], ['b-c'], ['b/c']],
1895
 
                          state, ['b', 'b-c', 'b/c'])
 
1835
                          state, ['b/d', 'b/d/e', 'f'])
1896
1836
 
1897
1837
    def test_bisect_one_page(self):
1898
1838
        """Test bisect when there is only 1 page to read"""
1904
1844
        self.assertBisect(expected,[['b/c']], state, ['b/c'])
1905
1845
        self.assertBisect(expected,[['b/d']], state, ['b/d'])
1906
1846
        self.assertBisect(expected,[['b/d/e']], state, ['b/d/e'])
1907
 
        self.assertBisect(expected,[['b-c']], state, ['b-c'])
1908
1847
        self.assertBisect(expected,[['f']], state, ['f'])
1909
1848
        self.assertBisect(expected,[['a'], ['b'], ['f']],
1910
1849
                          state, ['a', 'b', 'f'])
1911
 
        self.assertBisect(expected, [['b/d'], ['b/d/e'], ['f']],
 
1850
        # ('', 'f') sorts before the others
 
1851
        self.assertBisect(expected, [['f'], ['b/d'], ['b/d/e']],
1912
1852
                          state, ['b/d', 'b/d/e', 'f'])
1913
 
        self.assertBisect(expected, [['b'], ['b/c'], ['b-c']],
1914
 
                          state, ['b', 'b/c', 'b-c'])
1915
1853
 
1916
1854
    def test_bisect_duplicate_paths(self):
1917
1855
        """When bisecting for a path, handle multiple entries."""
1925
1863
        self.assertBisect(expected, [['b/d', 'b/d2']], state, ['b/d'])
1926
1864
        self.assertBisect(expected, [['b/d/e', 'b/d/e2']],
1927
1865
                          state, ['b/d/e'])
1928
 
        self.assertBisect(expected, [['b-c', 'b-c2']], state, ['b-c'])
1929
1866
        self.assertBisect(expected, [['f', 'f2']], state, ['f'])
1930
1867
 
1931
1868
    def test_bisect_page_size_too_small(self):
1938
1875
        self.assertBisect(expected, [['b/c']], state, ['b/c'])
1939
1876
        self.assertBisect(expected, [['b/d']], state, ['b/d'])
1940
1877
        self.assertBisect(expected, [['b/d/e']], state, ['b/d/e'])
1941
 
        self.assertBisect(expected, [['b-c']], state, ['b-c'])
1942
1878
        self.assertBisect(expected, [['f']], state, ['f'])
1943
1879
 
1944
1880
    def test_bisect_missing(self):
1947
1883
        self.assertBisect(expected, [None], state, ['foo'])
1948
1884
        self.assertBisect(expected, [None], state, ['b/foo'])
1949
1885
        self.assertBisect(expected, [None], state, ['bar/foo'])
1950
 
        self.assertBisect(expected, [None], state, ['b-c/foo'])
1951
1886
 
1952
1887
        self.assertBisect(expected, [['a'], None, ['b/d']],
1953
1888
                          state, ['a', 'foo', 'b/d'])
1969
1904
    def test_bisect_dirblocks(self):
1970
1905
        tree, state, expected = self.create_duplicated_dirstate()
1971
1906
        self.assertBisectDirBlocks(expected,
1972
 
            [['', 'a', 'a2', 'b', 'b2', 'b-c', 'b-c2', 'f', 'f2']],
1973
 
            state, [''])
 
1907
            [['', 'a', 'a2', 'b', 'b2', 'f', 'f2']], state, [''])
1974
1908
        self.assertBisectDirBlocks(expected,
1975
1909
            [['b/c', 'b/c2', 'b/d', 'b/d2']], state, ['b'])
1976
1910
        self.assertBisectDirBlocks(expected,
1977
1911
            [['b/d/e', 'b/d/e2']], state, ['b/d'])
1978
1912
        self.assertBisectDirBlocks(expected,
1979
 
            [['', 'a', 'a2', 'b', 'b2', 'b-c', 'b-c2', 'f', 'f2'],
 
1913
            [['', 'a', 'a2', 'b', 'b2', 'f', 'f2'],
1980
1914
             ['b/c', 'b/c2', 'b/d', 'b/d2'],
1981
1915
             ['b/d/e', 'b/d/e2'],
1982
1916
            ], state, ['', 'b', 'b/d'])
1997
1931
        self.assertBisectRecursive(expected, ['a'], state, ['a'])
1998
1932
        self.assertBisectRecursive(expected, ['b/c'], state, ['b/c'])
1999
1933
        self.assertBisectRecursive(expected, ['b/d/e'], state, ['b/d/e'])
2000
 
        self.assertBisectRecursive(expected, ['b-c'], state, ['b-c'])
2001
1934
        self.assertBisectRecursive(expected, ['b/d', 'b/d/e'],
2002
1935
                                   state, ['b/d'])
2003
1936
        self.assertBisectRecursive(expected, ['b', 'b/c', 'b/d', 'b/d/e'],
2004
1937
                                   state, ['b'])
2005
 
        self.assertBisectRecursive(expected, ['', 'a', 'b', 'b-c', 'f', 'b/c',
 
1938
        self.assertBisectRecursive(expected, ['', 'a', 'b', 'f', 'b/c',
2006
1939
                                              'b/d', 'b/d/e'],
2007
1940
                                   state, [''])
2008
1941
 
2032
1965
                                   state, ['b'])
2033
1966
 
2034
1967
 
2035
 
class TestDirstateValidation(TestCaseWithDirState):
2036
 
 
2037
 
    def test_validate_correct_dirstate(self):
2038
 
        state = self.create_complex_dirstate()
2039
 
        state._validate()
2040
 
        state.unlock()
2041
 
        # and make sure we can also validate with a read lock
2042
 
        state.lock_read()
2043
 
        try:
2044
 
            state._validate()
2045
 
        finally:
2046
 
            state.unlock()
2047
 
 
2048
 
    def test_dirblock_not_sorted(self):
2049
 
        tree, state, expected = self.create_renamed_dirstate()
2050
 
        state._read_dirblocks_if_needed()
2051
 
        last_dirblock = state._dirblocks[-1]
2052
 
        # we're appending to the dirblock, but this name comes before some of
2053
 
        # the existing names; that's wrong
2054
 
        last_dirblock[1].append(
2055
 
            (('h', 'aaaa', 'a-id'),
2056
 
             [('a', '', 0, False, ''),
2057
 
              ('a', '', 0, False, '')]))
2058
 
        e = self.assertRaises(AssertionError,
2059
 
            state._validate)
2060
 
        self.assertContainsRe(str(e), 'not sorted')
2061
 
 
2062
 
    def test_dirblock_name_mismatch(self):
2063
 
        tree, state, expected = self.create_renamed_dirstate()
2064
 
        state._read_dirblocks_if_needed()
2065
 
        last_dirblock = state._dirblocks[-1]
2066
 
        # add an entry with the wrong directory name
2067
 
        last_dirblock[1].append(
2068
 
            (('', 'z', 'a-id'),
2069
 
             [('a', '', 0, False, ''),
2070
 
              ('a', '', 0, False, '')]))
2071
 
        e = self.assertRaises(AssertionError,
2072
 
            state._validate)
2073
 
        self.assertContainsRe(str(e),
2074
 
            "doesn't match directory name")
2075
 
 
2076
 
    def test_dirblock_missing_rename(self):
2077
 
        tree, state, expected = self.create_renamed_dirstate()
2078
 
        state._read_dirblocks_if_needed()
2079
 
        last_dirblock = state._dirblocks[-1]
2080
 
        # make another entry for a-id, without a correct 'r' pointer to
2081
 
        # the real occurrence in the working tree
2082
 
        last_dirblock[1].append(
2083
 
            (('h', 'z', 'a-id'),
2084
 
             [('a', '', 0, False, ''),
2085
 
              ('a', '', 0, False, '')]))
2086
 
        e = self.assertRaises(AssertionError,
2087
 
            state._validate)
2088
 
        self.assertContainsRe(str(e),
2089
 
            'file a-id is absent in row')
2090
 
 
2091
 
 
2092
 
class TestDirstateTreeReference(TestCaseWithDirState):
2093
 
 
2094
 
    def test_reference_revision_is_none(self):
2095
 
        tree = self.make_branch_and_tree('tree', format='dirstate-with-subtree')
2096
 
        subtree = self.make_branch_and_tree('tree/subtree',
2097
 
                            format='dirstate-with-subtree')
2098
 
        subtree.set_root_id('subtree')
2099
 
        tree.add_reference(subtree)
2100
 
        tree.add('subtree')
2101
 
        state = dirstate.DirState.from_tree(tree, 'dirstate')
2102
 
        key = ('', 'subtree', 'subtree')
2103
 
        expected = ('', [(key,
2104
 
            [('t', '', 0, False, 'xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx')])])
2105
 
 
2106
 
        try:
2107
 
            self.assertEqual(expected, state._find_block(key))
2108
 
        finally:
2109
 
            state.unlock()
2110
 
 
2111
 
 
2112
 
class TestDiscardMergeParents(TestCaseWithDirState):
2113
 
 
2114
 
    def test_discard_no_parents(self):
2115
 
        # This should be a no-op
2116
 
        state = self.create_empty_dirstate()
2117
 
        self.addCleanup(state.unlock)
2118
 
        state._discard_merge_parents()
2119
 
        state._validate()
2120
 
 
2121
 
    def test_discard_one_parent(self):
2122
 
        # No-op
2123
 
        packed_stat = 'AAAAREUHaIpFB2iKAAADAQAtkqUAAIGk'
2124
 
        root_entry_direntry = ('', '', 'a-root-value'), [
2125
 
            ('d', '', 0, False, packed_stat),
2126
 
            ('d', '', 0, False, packed_stat),
2127
 
            ]
2128
 
        dirblocks = []
2129
 
        dirblocks.append(('', [root_entry_direntry]))
2130
 
        dirblocks.append(('', []))
2131
 
 
2132
 
        state = self.create_empty_dirstate()
2133
 
        self.addCleanup(state.unlock)
2134
 
        state._set_data(['parent-id'], dirblocks[:])
2135
 
        state._validate()
2136
 
 
2137
 
        state._discard_merge_parents()
2138
 
        state._validate()
2139
 
        self.assertEqual(dirblocks, state._dirblocks)
2140
 
 
2141
 
    def test_discard_simple(self):
2142
 
        # No-op
2143
 
        packed_stat = 'AAAAREUHaIpFB2iKAAADAQAtkqUAAIGk'
2144
 
        root_entry_direntry = ('', '', 'a-root-value'), [
2145
 
            ('d', '', 0, False, packed_stat),
2146
 
            ('d', '', 0, False, packed_stat),
2147
 
            ('d', '', 0, False, packed_stat),
2148
 
            ]
2149
 
        expected_root_entry_direntry = ('', '', 'a-root-value'), [
2150
 
            ('d', '', 0, False, packed_stat),
2151
 
            ('d', '', 0, False, packed_stat),
2152
 
            ]
2153
 
        dirblocks = []
2154
 
        dirblocks.append(('', [root_entry_direntry]))
2155
 
        dirblocks.append(('', []))
2156
 
 
2157
 
        state = self.create_empty_dirstate()
2158
 
        self.addCleanup(state.unlock)
2159
 
        state._set_data(['parent-id', 'merged-id'], dirblocks[:])
2160
 
        state._validate()
2161
 
 
2162
 
        # This should strip of the extra column
2163
 
        state._discard_merge_parents()
2164
 
        state._validate()
2165
 
        expected_dirblocks = [('', [expected_root_entry_direntry]), ('', [])]
2166
 
        self.assertEqual(expected_dirblocks, state._dirblocks)
2167
 
 
2168
 
    def test_discard_absent(self):
2169
 
        """If entries are only in a merge, discard should remove the entries"""
2170
 
        null_stat = dirstate.DirState.NULLSTAT
2171
 
        present_dir = ('d', '', 0, False, null_stat)
2172
 
        present_file = ('f', '', 0, False, null_stat)
2173
 
        absent = dirstate.DirState.NULL_PARENT_DETAILS
2174
 
        root_key = ('', '', 'a-root-value')
2175
 
        file_in_root_key = ('', 'file-in-root', 'a-file-id')
2176
 
        file_in_merged_key = ('', 'file-in-merged', 'b-file-id')
2177
 
        dirblocks = [('', [(root_key, [present_dir, present_dir, present_dir])]),
2178
 
                     ('', [(file_in_merged_key,
2179
 
                            [absent, absent, present_file]),
2180
 
                           (file_in_root_key,
2181
 
                            [present_file, present_file, present_file]),
2182
 
                          ]),
2183
 
                    ]
2184
 
 
2185
 
        state = self.create_empty_dirstate()
2186
 
        self.addCleanup(state.unlock)
2187
 
        state._set_data(['parent-id', 'merged-id'], dirblocks[:])
2188
 
        state._validate()
2189
 
 
2190
 
        exp_dirblocks = [('', [(root_key, [present_dir, present_dir])]),
2191
 
                         ('', [(file_in_root_key,
2192
 
                                [present_file, present_file]),
2193
 
                              ]),
2194
 
                        ]
2195
 
        state._discard_merge_parents()
2196
 
        state._validate()
2197
 
        self.assertEqual(exp_dirblocks, state._dirblocks)
2198
 
 
2199
 
    def test_discard_renamed(self):
2200
 
        null_stat = dirstate.DirState.NULLSTAT
2201
 
        present_dir = ('d', '', 0, False, null_stat)
2202
 
        present_file = ('f', '', 0, False, null_stat)
2203
 
        absent = dirstate.DirState.NULL_PARENT_DETAILS
2204
 
        root_key = ('', '', 'a-root-value')
2205
 
        file_in_root_key = ('', 'file-in-root', 'a-file-id')
2206
 
        # Renamed relative to parent
2207
 
        file_rename_s_key = ('', 'file-s', 'b-file-id')
2208
 
        file_rename_t_key = ('', 'file-t', 'b-file-id')
2209
 
        # And one that is renamed between the parents, but absent in this
2210
 
        key_in_1 = ('', 'file-in-1', 'c-file-id')
2211
 
        key_in_2 = ('', 'file-in-2', 'c-file-id')
2212
 
 
2213
 
        dirblocks = [
2214
 
            ('', [(root_key, [present_dir, present_dir, present_dir])]),
2215
 
            ('', [(key_in_1,
2216
 
                   [absent, present_file, ('r', 'file-in-2', 'c-file-id')]),
2217
 
                  (key_in_2,
2218
 
                   [absent, ('r', 'file-in-1', 'c-file-id'), present_file]),
2219
 
                  (file_in_root_key,
2220
 
                   [present_file, present_file, present_file]),
2221
 
                  (file_rename_s_key,
2222
 
                   [('r', 'file-t', 'b-file-id'), absent, present_file]),
2223
 
                  (file_rename_t_key,
2224
 
                   [present_file, absent, ('r', 'file-s', 'b-file-id')]),
2225
 
                 ]),
2226
 
        ]
2227
 
        exp_dirblocks = [
2228
 
            ('', [(root_key, [present_dir, present_dir])]),
2229
 
            ('', [(key_in_1, [absent, present_file]),
2230
 
                  (file_in_root_key, [present_file, present_file]),
2231
 
                  (file_rename_t_key, [present_file, absent]),
2232
 
                 ]),
2233
 
        ]
2234
 
        state = self.create_empty_dirstate()
2235
 
        self.addCleanup(state.unlock)
2236
 
        state._set_data(['parent-id', 'merged-id'], dirblocks[:])
2237
 
        state._validate()
2238
 
 
2239
 
        state._discard_merge_parents()
2240
 
        state._validate()
2241
 
        self.assertEqual(exp_dirblocks, state._dirblocks)
2242
 
 
2243
 
    def test_discard_all_subdir(self):
2244
 
        null_stat = dirstate.DirState.NULLSTAT
2245
 
        present_dir = ('d', '', 0, False, null_stat)
2246
 
        present_file = ('f', '', 0, False, null_stat)
2247
 
        absent = dirstate.DirState.NULL_PARENT_DETAILS
2248
 
        root_key = ('', '', 'a-root-value')
2249
 
        subdir_key = ('', 'sub', 'dir-id')
2250
 
        child1_key = ('sub', 'child1', 'child1-id')
2251
 
        child2_key = ('sub', 'child2', 'child2-id')
2252
 
        child3_key = ('sub', 'child3', 'child3-id')
2253
 
 
2254
 
        dirblocks = [
2255
 
            ('', [(root_key, [present_dir, present_dir, present_dir])]),
2256
 
            ('', [(subdir_key, [present_dir, present_dir, present_dir])]),
2257
 
            ('sub', [(child1_key, [absent, absent, present_file]),
2258
 
                     (child2_key, [absent, absent, present_file]),
2259
 
                     (child3_key, [absent, absent, present_file]),
2260
 
                    ]),
2261
 
        ]
2262
 
        exp_dirblocks = [
2263
 
            ('', [(root_key, [present_dir, present_dir])]),
2264
 
            ('', [(subdir_key, [present_dir, present_dir])]),
2265
 
            ('sub', []),
2266
 
        ]
2267
 
        state = self.create_empty_dirstate()
2268
 
        self.addCleanup(state.unlock)
2269
 
        state._set_data(['parent-id', 'merged-id'], dirblocks[:])
2270
 
        state._validate()
2271
 
 
2272
 
        state._discard_merge_parents()
2273
 
        state._validate()
2274
 
        self.assertEqual(exp_dirblocks, state._dirblocks)
2275
 
 
2276
 
 
2277
 
class Test_InvEntryToDetails(tests.TestCase):
2278
 
 
2279
 
    def assertDetails(self, expected, inv_entry):
2280
 
        details = dirstate.DirState._inv_entry_to_details(inv_entry)
2281
 
        self.assertEqual(expected, details)
2282
 
        # details should always allow join() and always be a plain str when
2283
 
        # finished
2284
 
        (minikind, fingerprint, size, executable, tree_data) = details
2285
 
        self.assertIsInstance(minikind, str)
2286
 
        self.assertIsInstance(fingerprint, str)
2287
 
        self.assertIsInstance(tree_data, str)
2288
 
 
2289
 
    def test_unicode_symlink(self):
2290
 
        inv_entry = inventory.InventoryLink('link-file-id',
2291
 
                                            u'nam\N{Euro Sign}e',
2292
 
                                            'link-parent-id')
2293
 
        inv_entry.revision = 'link-revision-id'
2294
 
        target = u'link-targ\N{Euro Sign}t'
2295
 
        inv_entry.symlink_target = target
2296
 
        self.assertDetails(('l', target.encode('UTF-8'), 0, False,
2297
 
                            'link-revision-id'), inv_entry)
2298
 
 
2299
 
 
2300
 
class TestSHA1Provider(tests.TestCaseInTempDir):
2301
 
 
2302
 
    def test_sha1provider_is_an_interface(self):
2303
 
        p = dirstate.SHA1Provider()
2304
 
        self.assertRaises(NotImplementedError, p.sha1, "foo")
2305
 
        self.assertRaises(NotImplementedError, p.stat_and_sha1, "foo")
2306
 
 
2307
 
    def test_defaultsha1provider_sha1(self):
2308
 
        text = 'test\r\nwith\nall\rpossible line endings\r\n'
2309
 
        self.build_tree_contents([('foo', text)])
2310
 
        expected_sha = osutils.sha_string(text)
2311
 
        p = dirstate.DefaultSHA1Provider()
2312
 
        self.assertEqual(expected_sha, p.sha1('foo'))
2313
 
 
2314
 
    def test_defaultsha1provider_stat_and_sha1(self):
2315
 
        text = 'test\r\nwith\nall\rpossible line endings\r\n'
2316
 
        self.build_tree_contents([('foo', text)])
2317
 
        expected_sha = osutils.sha_string(text)
2318
 
        p = dirstate.DefaultSHA1Provider()
2319
 
        statvalue, sha1 = p.stat_and_sha1('foo')
2320
 
        self.assertTrue(len(statvalue) >= 10)
2321
 
        self.assertEqual(len(text), statvalue.st_size)
2322
 
        self.assertEqual(expected_sha, sha1)
 
1968
class TestBisectDirblock(TestCase):
 
1969
    """Test that bisect_dirblock() returns the expected values.
 
1970
 
 
1971
    bisect_dirblock is intended to work like bisect.bisect_left() except it
 
1972
    knows it is working on dirblocks and that dirblocks are sorted by ('path',
 
1973
    'to', 'foo') chunks rather than by raw 'path/to/foo'.
 
1974
    """
 
1975
 
 
1976
    def assertBisect(self, dirblocks, split_dirblocks, path, *args, **kwargs):
 
1977
        """Assert that bisect_split works like bisect_left on the split paths.
 
1978
 
 
1979
        :param dirblocks: A list of (path, [info]) pairs.
 
1980
        :param split_dirblocks: A list of ((split, path), [info]) pairs.
 
1981
        :param path: The path we are indexing.
 
1982
 
 
1983
        All other arguments will be passed along.
 
1984
        """
 
1985
        bisect_split_idx = dirstate.bisect_dirblock(dirblocks, path,
 
1986
                                                 *args, **kwargs)
 
1987
        split_dirblock = (path.split('/'), [])
 
1988
        bisect_left_idx = bisect.bisect_left(split_dirblocks, split_dirblock,
 
1989
                                             *args)
 
1990
        self.assertEqual(bisect_left_idx, bisect_split_idx,
 
1991
                         'bisect_split disagreed. %s != %s'
 
1992
                         ' for key %s'
 
1993
                         % (bisect_left_idx, bisect_split_idx, path)
 
1994
                         )
 
1995
 
 
1996
    def paths_to_dirblocks(self, paths):
 
1997
        """Convert a list of paths into dirblock form.
 
1998
 
 
1999
        Also, ensure that the paths are in proper sorted order.
 
2000
        """
 
2001
        dirblocks = [(path, []) for path in paths]
 
2002
        split_dirblocks = [(path.split('/'), []) for path in paths]
 
2003
        self.assertEqual(sorted(split_dirblocks), split_dirblocks)
 
2004
        return dirblocks, split_dirblocks
 
2005
 
 
2006
    def test_simple(self):
 
2007
        """In the simple case it works just like bisect_left"""
 
2008
        paths = ['', 'a', 'b', 'c', 'd']
 
2009
        dirblocks, split_dirblocks = self.paths_to_dirblocks(paths)
 
2010
        for path in paths:
 
2011
            self.assertBisect(dirblocks, split_dirblocks, path)
 
2012
        self.assertBisect(dirblocks, split_dirblocks, '_')
 
2013
        self.assertBisect(dirblocks, split_dirblocks, 'aa')
 
2014
        self.assertBisect(dirblocks, split_dirblocks, 'bb')
 
2015
        self.assertBisect(dirblocks, split_dirblocks, 'cc')
 
2016
        self.assertBisect(dirblocks, split_dirblocks, 'dd')
 
2017
        self.assertBisect(dirblocks, split_dirblocks, 'a/a')
 
2018
        self.assertBisect(dirblocks, split_dirblocks, 'b/b')
 
2019
        self.assertBisect(dirblocks, split_dirblocks, 'c/c')
 
2020
        self.assertBisect(dirblocks, split_dirblocks, 'd/d')
 
2021
 
 
2022
    def test_involved(self):
 
2023
        """This is where bisect_left diverges slightly."""
 
2024
        paths = ['', 'a',
 
2025
                 'a/a', 'a/a/a', 'a/a/z', 'a/a-a', 'a/a-z',
 
2026
                 'a/z', 'a/z/a', 'a/z/z', 'a/z-a', 'a/z-z',
 
2027
                 'a-a', 'a-z',
 
2028
                 'z', 'z/a/a', 'z/a/z', 'z/a-a', 'z/a-z',
 
2029
                 'z/z', 'z/z/a', 'z/z/z', 'z/z-a', 'z/z-z',
 
2030
                 'z-a', 'z-z',
 
2031
                ]
 
2032
        dirblocks, split_dirblocks = self.paths_to_dirblocks(paths)
 
2033
        for path in paths:
 
2034
            self.assertBisect(dirblocks, split_dirblocks, path)
 
2035
 
 
2036
    def test_involved_cached(self):
 
2037
        """This is where bisect_left diverges slightly."""
 
2038
        paths = ['', 'a',
 
2039
                 'a/a', 'a/a/a', 'a/a/z', 'a/a-a', 'a/a-z',
 
2040
                 'a/z', 'a/z/a', 'a/z/z', 'a/z-a', 'a/z-z',
 
2041
                 'a-a', 'a-z',
 
2042
                 'z', 'z/a/a', 'z/a/z', 'z/a-a', 'z/a-z',
 
2043
                 'z/z', 'z/z/a', 'z/z/z', 'z/z-a', 'z/z-z',
 
2044
                 'z-a', 'z-z',
 
2045
                ]
 
2046
        cache = {}
 
2047
        dirblocks, split_dirblocks = self.paths_to_dirblocks(paths)
 
2048
        for path in paths:
 
2049
            self.assertBisect(dirblocks, split_dirblocks, path, cache=cache)
 
2050