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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20
20
from bzrlib import bzrdir
21
21
from bzrlib.tests import TestCaseWithTransport, TestCase
22
22
from bzrlib.branch import Branch
23
from bzrlib.conflicts import (MissingParent, ContentsConflict, TextConflict,
24
PathConflict, DuplicateID, DuplicateEntry, ParentLoop, UnversionedParent,
23
from bzrlib.conflicts import (
27
37
from bzrlib.errors import NotConflicted
33
43
# The order of 'path' here is important - do not let it
34
44
# be a sorted list.
35
example_conflicts = ConflictList([
36
MissingParent('Not deleting', 'pathg', 'idg'),
37
ContentsConflict('patha', 'ida'),
38
TextConflict('patha'),
39
PathConflict('pathb', 'pathc', 'idb'),
40
DuplicateID('Unversioned existing file', 'pathc', 'pathc2', 'idc', 'idc'),
41
DuplicateEntry('Moved existing file to', 'pathdd.moved', 'pathd', 'idd',
43
ParentLoop('Cancelled move', 'pathe', 'path2e', None, 'id2e'),
44
UnversionedParent('Versioned directory', 'pathf', 'idf'),
45
# u'\xe5' == a with circle
46
# '\xc3\xae' == u'\xee' == i with hat
47
# So these are u'pathg' and 'idg' only with a circle and a hat. (shappo?)
48
example_conflicts = ConflictList([
49
MissingParent('Not deleting', u'p\xe5thg', '\xc3\xaedg'),
50
ContentsConflict(u'p\xe5tha', None, '\xc3\xaeda'),
51
TextConflict(u'p\xe5tha'),
52
PathConflict(u'p\xe5thb', u'p\xe5thc', '\xc3\xaedb'),
53
DuplicateID('Unversioned existing file', u'p\xe5thc', u'p\xe5thc2',
54
'\xc3\xaedc', '\xc3\xaedc'),
55
DuplicateEntry('Moved existing file to', u'p\xe5thdd.moved', u'p\xe5thd',
57
ParentLoop('Cancelled move', u'p\xe5the', u'p\xe5th2e',
59
UnversionedParent('Versioned directory', u'p\xe5thf', '\xc3\xaedf'),
60
NonDirectoryParent('Created directory', u'p\xe5thg', '\xc3\xaedg'),
58
74
file('hello.OTHER', 'w').write('hello world3')
59
75
file('hello.sploo.BASE', 'w').write('yellow world')
60
76
file('hello.sploo.OTHER', 'w').write('yellow world2')
61
78
self.assertEqual(len(list(tree.list_files())), 6)
62
80
conflicts = tree.conflicts()
63
81
self.assertEqual(len(conflicts), 2)
64
82
self.assert_('hello' in conflicts[0].path)
67
85
restore('hello.sploo')
68
86
self.assertEqual(len(tree.conflicts()), 0)
69
87
self.assertFileEqual('hello world2', 'hello')
70
assert not os.path.lexists('hello.sploo')
88
self.assertFalse(os.path.lexists('hello.sploo'))
71
89
self.assertRaises(NotConflicted, restore, 'hello')
72
90
self.assertRaises(NotConflicted, restore, 'hello.sploo')
82
100
l = ConflictList([TextConflict('hello')])
83
101
l.remove_files(tree)
103
def test_select_conflicts(self):
104
tree = self.make_branch_and_tree('.')
105
tree_conflicts = ConflictList([ContentsConflict('foo'),
106
ContentsConflict('bar')])
107
self.assertEqual((ConflictList([ContentsConflict('bar')]),
108
ConflictList([ContentsConflict('foo')])),
109
tree_conflicts.select_conflicts(tree, ['foo']))
110
self.assertEqual((ConflictList(), tree_conflicts),
111
tree_conflicts.select_conflicts(tree, [''],
112
ignore_misses=True, recurse=True))
113
tree_conflicts = ConflictList([ContentsConflict('foo/baz'),
114
ContentsConflict('bar')])
115
self.assertEqual((ConflictList([ContentsConflict('bar')]),
116
ConflictList([ContentsConflict('foo/baz')])),
117
tree_conflicts.select_conflicts(tree, ['foo'],
120
tree_conflicts = ConflictList([PathConflict('qux', 'foo/baz')])
121
self.assertEqual((ConflictList(), tree_conflicts),
122
tree_conflicts.select_conflicts(tree, ['foo'],
125
self.assertEqual((tree_conflicts, ConflictList()),
126
tree_conflicts.select_conflicts(tree, ['foo'],
129
def test_resolve_conflicts_recursive(self):
130
tree = self.make_branch_and_tree('.')
131
self.build_tree(['dir/', 'dir/hello'])
132
tree.add(['dir', 'dir/hello'])
133
tree.set_conflicts(ConflictList([TextConflict('dir/hello')]))
134
resolve(tree, ['dir'], recursive=False, ignore_misses=True)
135
self.assertEqual(ConflictList([TextConflict('dir/hello')]),
137
resolve(tree, ['dir'], recursive=True, ignore_misses=True)
138
self.assertEqual(ConflictList([]),
86
142
class TestConflictStanzas(TestCase):
89
145
# write and read our example stanza.
90
146
stanza_iter = example_conflicts.to_stanzas()
91
147
processed = ConflictList.from_stanzas(stanza_iter)
92
for o,p in zip(processed, example_conflicts):
148
for o, p in zip(processed, example_conflicts):
93
149
self.assertEqual(o, p)
151
self.assertIsInstance(o.path, unicode)
153
if o.file_id is not None:
154
self.assertIsInstance(o.file_id, str)
156
conflict_path = getattr(o, 'conflict_path', None)
157
if conflict_path is not None:
158
self.assertIsInstance(conflict_path, unicode)
160
conflict_file_id = getattr(o, 'conflict_file_id', None)
161
if conflict_file_id is not None:
162
self.assertIsInstance(conflict_file_id, str)
95
164
def test_stanzification(self):
96
165
for stanza in example_conflicts.to_stanzas():
98
self.assertStartsWith(stanza['file_id'], 'id')
101
self.assertStartsWith(stanza['path'], 'path')
103
self.assertStartsWith(stanza['conflict_file_id'], 'id')
104
self.assertStartsWith(stanza['conflict_file_path'], 'path')
166
if 'file_id' in stanza:
167
# In Stanza form, the file_id has to be unicode.
168
self.assertStartsWith(stanza['file_id'], u'\xeed')
169
self.assertStartsWith(stanza['path'], u'p\xe5th')
170
if 'conflict_path' in stanza:
171
self.assertStartsWith(stanza['conflict_path'], u'p\xe5th')
172
if 'conflict_file_id' in stanza:
173
self.assertStartsWith(stanza['conflict_file_id'], u'\xeed')