/brz/remove-bazaar

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

 

"""Tests of the dirstate functionality being built for WorkingTreeFormat4."""

 

import os

 

    errors,

    memorytree,

    osutils,

    revision as _mod_revision,

    tests,

    )

 

 

# TODO:

# set_path_id  setting id when state is in memory modified

 

 

 

 

class TestCaseWithDirState(tests.TestCaseWithTransport):

    """Helper functions for creating DirState objects with various content."""

 

    # Set by load_tests

    _dir_reader_class = None

    _native_to_unicode = None # Not used yet

 

    def setUp(self):

        self.overrideAttr(osutils,

                          '_selected_dir_reader', self._dir_reader_class())

 

        paths = ['a', 'b/', 'b/c', 'b/d/', 'b/d/e', 'b-c', 'f']

        file_ids = ['a-id', 'b-id', 'c-id', 'd-id', 'e-id', 'b-c-id', 'f-id']

        self.build_tree(['tree/' + p for p in paths])

        tree.add([p.rstrip('/') for p in paths], file_ids)

        revision_id = 'rev-1'

        # a_packed_stat = dirstate.pack_stat(os.stat('tree/a'))

        t = self.get_transport('tree')

        f_len = len(f_text)

        null_stat = dirstate.DirState.NULLSTAT

        expected = {

                  ('d', '', 0, False, null_stat),

                  ('d', '', 0, False, revision_id),

                ]),

                   ('f', '', 0, False, null_stat),

                   ('f', a_sha, a_len, False, revision_id),

                 ]),

                  ('d', '', 0, False, null_stat),

                  ('d', '', 0, False, revision_id),

                 ]),

                    ('f', '', 0, False, null_stat),

                    ('f', c_sha, c_len, False, revision_id),

                   ]),

                    ('d', '', 0, False, null_stat),

                    ('d', '', 0, False, revision_id),

                   ]),

                      ('f', '', 0, False, null_stat),

                      ('f', e_sha, e_len, False, revision_id),

                     ]),

                      ('f', '', 0, False, null_stat),

                      ('f', b_c_sha, b_c_len, False, revision_id),

                     ]),

                  ('f', '', 0, False, null_stat),

                  ('f', f_sha, f_len, False, revision_id),

                 ]),

        tree, state, expected = self.create_basic_dirstate()

        # Now we will just remove and add every file so we get an extra entry

        # per entry. Unversion in reverse order so we handle subdirs

        tree.add(['a', 'b', 'b/c', 'b/d', 'b/d/e', 'b-c', 'f'],

                 ['a-id2', 'b-id2', 'c-id2', 'd-id2', 'e-id2', 'b-c-id2', 'f-id2'])

 

 

        old_a = expected['a']

        expected['a'] = (old_a[0], [('r', 'b/g', 0, False, ''), old_a[1][1]])

                                                ('r', 'a', 0, False, '')])

        old_d = expected['b/d']

        expected['b/d'] = (old_d[0], [('r', 'h', 0, False, ''), old_d[1][1]])

                                             ('r', 'b/d', 0, False, '')])

 

        old_e = expected['b/d/e']

        expected['b/d/e'] = (old_e[0], [('r', 'h/e', 0, False, ''),

                             old_e[1][1]])

                                                ('r', 'b/d/e', 0, False, '')])

 

        state.unlock()

        # create a parent by doing a commit

        tree = self.make_branch_and_tree('tree')

        rev_id = tree.commit('first post')

        rev_id2 = tree2.commit('second post', allow_pointless=True)

        tree.merge_from_branch(tree2.branch)

        expected_result = ([rev_id, rev_id2], [

    def get_tree_with_a_file(self):

        tree = self.make_branch_and_tree('tree')

        self.build_tree(['tree/a file'])

        return tree

 

    def test_non_empty_no_parents_to_dirstate(self):

            (('', '', tree.get_root_id()), # common details

             [('d', '', 0, False, dirstate.DirState.NULLSTAT), # current tree

             ]),

             [('f', '', 0, False, dirstate.DirState.NULLSTAT), # current

             ]),

            ])

        rev_id = tree.commit('first post').encode('utf8')

        # change the current content to be different this will alter stat, sha

        # and length:

        expected_result = ([rev_id], [

            (('', '', tree.get_root_id()), # common details

             [('d', '', 0, False, dirstate.DirState.NULLSTAT), # current tree

              ('d', '', 0, False, rev_id), # first parent details

             ]),

             [('f', '', 0, False, dirstate.DirState.NULLSTAT), # current

              ('f', 'c3ed76e4bfd45ff1763ca206055bca8e9fc28aa8', 24, False,

               rev_id), # first parent

        # create a parent by doing a commit

        tree = self.get_tree_with_a_file()

        rev_id = tree.commit('first post').encode('utf8')

        # change the current content to be different this will alter stat, sha

        # and length:

        self.build_tree_contents([('tree2/a file', 'merge content\n')])

        tree.merge_from_branch(tree2.branch)

        # change the current content to be different this will alter stat, sha

        # and length again, giving us three distinct values:

        expected_result = ([rev_id, rev_id2], [

            (('', '', tree.get_root_id()), # common details

             [('d', '', 0, False, dirstate.DirState.NULLSTAT), # current tree

              ('d', '', 0, False, rev_id), # first parent details

              ('d', '', 0, False, rev_id), # second parent details

             ]),

             [('f', '', 0, False, dirstate.DirState.NULLSTAT), # current

              ('f', 'c3ed76e4bfd45ff1763ca206055bca8e9fc28aa8', 24, False,

               rev_id), # first parent

 

class TestDirStateOnFile(TestCaseWithDirState):

 

    def test_construct_with_path(self):

        tree = self.make_branch_and_tree('tree')

        state = dirstate.DirState.from_tree(tree, 'dirstate.from_tree')

        # write to disk.

        lines = state.get_lines()

        state.unlock()

        # get a state object

        # no parents, default tree content

        expected_result = ([], [

            state.unlock()

 

    def test_can_save_in_read_lock(self):

        try:

            entry = state._get_entry(0, path_utf8='a-file')

            # The current size should be 0 (default)

            self.assertEqual(0, entry[1][0][2])

            # We should have a real entry.

            self.assertNotEqual((None, None), entry)

            state._sha_cutoff_time()

 

            # The dirblock has been updated

                             state._dirblock_state)

 

            del entry

        state.lock_read()

        try:

            entry = state._get_entry(0, path_utf8='a-file')

        finally:

            state.unlock()

 

    def test_save_fails_quietly_if_locked(self):

        """If dirstate is locked, save will fail without complaining."""

        try:

            entry = state._get_entry(0, path_utf8='a-file')

                             state._dirblock_state)

 

            # Now, before we try to save, grab another dirstate, and take out a

        state = dirstate.DirState.initialize('dirstate')

        try:

            # No stat and no sha1 sum.

            state.save()

        finally:

            state.unlock()

 

        # The dirstate should include TREE_ROOT and 'a-file' and nothing else

        expected_blocks = [

                   [('d', '', 0, False, dirstate.DirState.NULLSTAT)])]),

                   [('f', '', 0, False, dirstate.DirState.NULLSTAT)])]),

        ]

 

            self.assertEqual(expected_blocks, state._dirblocks)

 

            # Now modify the state, but mark it as inconsistent

            state._changes_aborted = True

            state.save()

        finally:

        # On win32 you can't read from a locked file, even within the same

        # process. So we have to unlock and release before we check the file

        # contents.

        state.lock_read() # check_state_with_reopen will unlock

        self.check_state_with_reopen(expected_result, state)

 

 

class TestDirStateManipulations(TestCaseWithDirState):

 

    def test_set_state_from_inventory_no_content_no_parents(self):

        # setting the current inventory is a slow but important api to support.

        expected_result = [], [

            (('', '', root_id), [

             ('d', '', 0, False, dirstate.DirState.NULLSTAT)])]

            # This will unlock it

            self.check_state_with_reopen(expected_result, state)

 

    def test_set_state_from_inventory_preserves_hashcache(self):

        # https://bugs.launchpad.net/bzr/+bug/146176

        # set_state_from_inventory should preserve the stat and hash value for

        try:

            # make a dirstate with some valid hashcache data

            # file on disk, but that's not needed for this test

            self.build_tree_contents([('foo', foo_contents)])

 

            foo_stat = os.stat('foo')

            foo_packed = dirstate.pack_stat(foo_stat)

                tree._dirstate._get_entry(0, 'foo-id'))

 

            # extract the inventory, and add something to it

            # should see the file we poked in...

            self.assertTrue(inv.has_filename('foo'))

            tree._dirstate._validate()

            # this used to cause it to lose its hashcache

            tree._dirstate.set_state_from_inventory(inv)

        try:

            tree1.add(['a', 'a/b', 'a-b', 'a/b/foo', 'a-b/bar'],

                      ['a-id', 'b-id', 'a-b-id', 'foo-id', 'bar-id'])

            root_id = tree1.get_root_id()

        finally:

            tree1.unlock()

        expected_result1 = [('', '', root_id, 'd'),

            for entry in state._iter_entries():

                values.append(entry[0] + entry[1][0][:1])

            self.assertEqual(expected_result1, values)

            state.set_state_from_inventory(inv)

            values = []

            for entry in state._iter_entries():

        """Set the root file id in a dirstate with parents"""

        mt = self.make_branch_and_tree('mt')

        # in case the default tree format uses a different root id

        state = dirstate.DirState.initialize('dirstate')

        state._validate()

        try:

            revid1 = tree1.commit('foo')

        finally:

            tree1.unlock()

        tree2 = memorytree.MemoryTree.create_on_branch(branch2)

        tree2.lock_write()

        try:

        tree1.lock_write()

        try:

            tree1.add('')

            revid1 = tree1.commit('foo')

        finally:

            tree1.unlock()

        tree2 = memorytree.MemoryTree.create_on_branch(branch2)

        tree2.lock_write()

        try:

            revid2 = tree2.commit('foo')

            root_id = tree2.get_root_id()

        finally:

             ('d', '', 0, False, revid1.encode('utf8')),

             ('d', '', 0, False, revid1.encode('utf8'))

             ]),

             ('a', '', 0, False, ''),

             ('f', '2439573625385400f2a669657a7db6ae7515d371', 12, False,

              revid1.encode('utf8')),

        # the 1*20 is the sha1 pretend value.

        state = dirstate.DirState.initialize('dirstate')

        expected_entries = [

             ('d', '', 0, False, dirstate.DirState.NULLSTAT), # current tree

             ]),

             ('f', '1'*20, 19, False, dirstate.pack_stat(stat)), # current tree

             ]),

            ]

        try:

            # having added it, it should be in the output of iter_entries.

            self.assertEqual(expected_entries, list(state._iter_entries()))

            # saving and reloading should not affect this.

        # The most trivial addition of a symlink when there are no parents and

        # its in the root and all data about the file is supplied

        # bzr doesn't support fake symlinks on windows, yet.

        os.symlink(target, link_name)

        stat = os.lstat(link_name)

        expected_entries = [

        self._test_add_symlink_to_root_no_parents_all_data('a link', 'target')

 

    def test_add_symlink_unicode_to_root_no_parents_all_data(self):