~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/blackbox/test_conflicts.py

(gz) Fix unicode handling in bzrlib.conflicts ui code (Martin [gz])

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006, 2007, 2009, 2010 Canonical Ltd
 
1
# Copyright (C) 2006, 2007, 2009, 2010, 2011 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
21
21
    )
22
22
from bzrlib.tests import script
23
23
 
24
 
def make_tree_with_conflicts(test, this_path='this', other_path='other'):
 
24
 
 
25
def make_tree_with_conflicts(test, this_path='this', other_path='other',
 
26
        prefix='my'):
25
27
    this_tree = test.make_branch_and_tree(this_path)
26
28
    test.build_tree_contents([
27
 
        ('%s/myfile' % (this_path,), 'this content\n'),
28
 
        ('%s/my_other_file' % (this_path,), 'this content\n'),
29
 
        ('%s/mydir/' % (this_path,),),
 
29
        ('%s/%sfile' % (this_path, prefix), 'this content\n'),
 
30
        ('%s/%s_other_file' % (this_path, prefix), 'this content\n'),
 
31
        ('%s/%sdir/' % (this_path, prefix),),
30
32
        ])
31
 
    this_tree.add('myfile')
32
 
    this_tree.add('my_other_file')
33
 
    this_tree.add('mydir')
 
33
    this_tree.add(prefix+'file')
 
34
    this_tree.add(prefix+'_other_file')
 
35
    this_tree.add(prefix+'dir')
34
36
    this_tree.commit(message="new")
35
37
    other_tree = this_tree.bzrdir.sprout(other_path).open_workingtree()
36
38
    test.build_tree_contents([
37
 
        ('%s/myfile' % (other_path,), 'contentsb\n'),
38
 
        ('%s/my_other_file' % (other_path,), 'contentsb\n'),
 
39
        ('%s/%sfile' % (other_path, prefix), 'contentsb\n'),
 
40
        ('%s/%s_other_file' % (other_path, prefix), 'contentsb\n'),
39
41
        ])
40
 
    other_tree.rename_one('mydir', 'mydir2')
 
42
    other_tree.rename_one(prefix+'dir', prefix+'dir2')
41
43
    other_tree.commit(message="change")
42
44
    test.build_tree_contents([
43
 
        ('%s/myfile' % (this_path,), 'contentsa2\n'),
44
 
        ('%s/my_other_file' % (this_path,), 'contentsa2\n'),
 
45
        ('%s/%sfile' % (this_path, prefix), 'contentsa2\n'),
 
46
        ('%s/%s_other_file' % (this_path, prefix), 'contentsa2\n'),
45
47
        ])
46
 
    this_tree.rename_one('mydir', 'mydir3')
 
48
    this_tree.rename_one(prefix+'dir', prefix+'dir3')
47
49
    this_tree.commit(message='change')
48
50
    this_tree.merge_from_branch(other_tree.branch)
49
51
    return this_tree, other_tree
79
81
Path conflict: mydir3 / mydir2
80
82
Text conflict in myfile
81
83
""")
 
84
 
 
85
 
 
86
class TestUnicodePaths(tests.TestCaseWithTransport):
 
87
    """Unicode characters in conflicts should be displayed properly"""
 
88
 
 
89
    encoding = "UTF-8"
 
90
 
 
91
    def _as_output(self, text):
 
92
        return text
 
93
 
 
94
    def test_messages(self):
 
95
        """Conflict messages involving non-ascii paths are displayed okay"""
 
96
        make_tree_with_conflicts(self, "branch", prefix=u"\xA7")
 
97
        out, err = self.run_bzr(["conflicts", "-d", "branch"],
 
98
            encoding=self.encoding)
 
99
        self.assertEqual(out.decode(self.encoding),
 
100
            u"Text conflict in \xA7_other_file\n"
 
101
            u"Path conflict: \xA7dir3 / \xA7dir2\n"
 
102
            u"Text conflict in \xA7file\n")
 
103
        self.assertEqual(err, "")
 
104
 
 
105
    def test_text_conflict_paths(self):
 
106
        """Text conflicts on non-ascii paths are displayed okay"""
 
107
        make_tree_with_conflicts(self, "branch", prefix=u"\xA7")
 
108
        out, err = self.run_bzr(["conflicts", "-d", "branch", "--text"],
 
109
            encoding=self.encoding)
 
110
        self.assertEqual(out.decode(self.encoding),
 
111
            u"\xA7_other_file\n"
 
112
            u"\xA7file\n")
 
113
        self.assertEqual(err, "")
 
114
 
 
115
 
 
116
class TestUnicodePathsOnAsciiTerminal(TestUnicodePaths):
 
117
    """Undisplayable unicode characters in conflicts should be escaped"""
 
118
 
 
119
    encoding = "ascii"
 
120
 
 
121
    def setUp(self):
 
122
        self.skip("Need to decide if replacing is the desired behaviour")
 
123
 
 
124
    def _as_output(self, text):
 
125
        return text.encode(self.encoding, "replace")