14
14
# along with this program; if not, write to the Free Software
 
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
17
19
from bzrlib import (
 
 
22
from bzrlib.workingtree import WorkingTree
 
 
23
from bzrlib.tests.blackbox import ExternalBase
 
23
25
# FIXME: These don't really look at the output of the conflict commands, just
 
24
26
# the number of lines - there should be more examination.
 
26
 
class TestConflictsBase(tests.TestCaseWithTransport):
 
 
28
class TestConflicts(ExternalBase):
 
29
 
        super(TestConflictsBase, self).setUp()
 
30
 
        self.make_tree_with_conflicts()
 
32
 
    def make_tree_with_conflicts(self):
 
 
31
        super(ExternalBase, self).setUp()
 
33
32
        a_tree = self.make_branch_and_tree('a')
 
34
33
        self.build_tree_contents([
 
35
34
            ('a/myfile', 'contentsa\n'),
 
 
54
53
        a_tree.rename_one('mydir', 'mydir3')
 
55
54
        a_tree.commit(message='change')
 
56
55
        a_tree.merge_from_branch(b_tree.branch)
 
58
 
    def run_bzr(self, cmd, working_dir='a', **kwargs):
 
59
 
        return super(TestConflictsBase, self).run_bzr(
 
60
 
            cmd, working_dir=working_dir, **kwargs)
 
63
 
class TestConflicts(TestConflictsBase):
 
65
58
    def test_conflicts(self):
 
66
 
        out, err = self.run_bzr('conflicts')
 
67
 
        self.assertEqual(3, len(out.splitlines()))
 
 
59
        conflicts, errs = self.run_bzr('conflicts')
 
 
60
        self.assertEqual(3, len(conflicts.splitlines()))
 
69
62
    def test_conflicts_text(self):
 
70
 
        out, err = self.run_bzr('conflicts --text')
 
71
 
        self.assertEqual(['my_other_file', 'myfile'], out.splitlines())
 
74
 
class TestResolve(TestConflictsBase):
 
 
63
        conflicts = self.run_bzr('conflicts --text')[0].splitlines()
 
 
64
        self.assertEqual(['my_other_file', 'myfile'], conflicts)
 
76
66
    def test_resolve(self):
 
77
67
        self.run_bzr('resolve myfile')
 
78
 
        out, err = self.run_bzr('conflicts')
 
79
 
        self.assertEqual(2, len(out.splitlines()))
 
 
68
        conflicts, errs = self.run_bzr('conflicts')
 
 
69
        self.assertEqual(2, len(conflicts.splitlines()))
 
80
70
        self.run_bzr('resolve my_other_file')
 
81
71
        self.run_bzr('resolve mydir2')
 
82
 
        out, err = self.run_bzr('conflicts')
 
83
 
        self.assertEqual(0, len(out.splitlines()))
 
 
72
        conflicts, errs = self.run_bzr('conflicts')
 
 
73
        self.assertEqual(len(conflicts.splitlines()), 0)
 
85
75
    def test_resolve_all(self):
 
86
76
        self.run_bzr('resolve --all')
 
87
 
        out, err = self.run_bzr('conflicts')
 
88
 
        self.assertEqual(0, len(out.splitlines()))
 
 
77
        conflicts, errs = self.run_bzr('conflicts')
 
 
78
        self.assertEqual(len(conflicts.splitlines()), 0)
 
90
80
    def test_resolve_in_subdir(self):
 
91
81
        """resolve when run from subdirectory should handle relative paths"""
 
92
 
        self.build_tree(["a/subdir/"])
 
93
 
        self.run_bzr("resolve ../myfile", working_dir='a/subdir')
 
94
 
        self.run_bzr("resolve ../a/myfile", working_dir='b')
 
95
 
        wt = workingtree.WorkingTree.open_containing('b')[0]
 
96
 
        conflicts = wt.conflicts()
 
97
 
        self.assertEqual(True, conflicts.is_empty(),
 
98
 
                         "tree still contains conflicts: %r" % conflicts)
 
 
82
        orig_dir = os.getcwdu()
 
 
86
            self.run_bzr("resolve ../myfile")
 
 
88
            self.run_bzr("resolve ../a/myfile")
 
 
89
            wt = WorkingTree.open_containing('.')[0]
 
 
90
            conflicts = wt.conflicts()
 
 
91
            if not conflicts.is_empty():
 
 
92
                self.fail("tree still contains conflicts: %r" % conflicts)
 
100
96
    def test_auto_resolve(self):
 
101
97
        """Text conflicts can be resolved automatically"""
 
 
106
102
        self.assertEqual(tree.kind('file_id'), 'file')
 
107
103
        file_conflict = conflicts.TextConflict('file', file_id='file_id')
 
108
104
        tree.set_conflicts(conflicts.ConflictList([file_conflict]))
 
109
 
        note = self.run_bzr('resolve', retcode=1, working_dir='tree')[1]
 
 
106
        note = self.run_bzr('resolve', retcode=1)[1]
 
110
107
        self.assertContainsRe(note, '0 conflict\\(s\\) auto-resolved.')
 
111
108
        self.assertContainsRe(note,
 
112
109
            'Remaining conflicts:\nText conflict in file')
 
113
 
        self.build_tree_contents([('tree/file', 'a\n')])
 
114
 
        note = self.run_bzr('resolve', working_dir='tree')[1]
 
 
110
        self.build_tree_contents([('file', 'a\n')])
 
 
111
        note = self.run_bzr('resolve')[1]
 
115
112
        self.assertContainsRe(note, 'All conflicts resolved.')