20
from bzrlib.tests import TestCaseInTempDir
20
from bzrlib import bzrdir
21
from bzrlib.tests import TestCaseWithTransport, TestCase
21
22
from bzrlib.branch import Branch
22
from bzrlib.commit import Commit
23
from bzrlib.conflicts import restore
23
from bzrlib.conflicts import (MissingParent, ContentsConflict, TextConflict,
24
PathConflict, DuplicateID, DuplicateEntry, ParentLoop, UnversionedParent,
24
27
from bzrlib.errors import NotConflicted
26
30
# TODO: Test commit with some added, and added-but-missing files
28
class TestConflicts(TestCaseInTempDir):
31
# RBC 20060124 is that not tested in test_commit.py ?
33
# The order of 'path' here is important - do not let it
35
# u'\xe5' == a with circle
36
# '\xc3\xae' == u'\xee' == i with hat
37
# So these are u'pathg' and 'idg' only with a circle and a hat. (shappo?)
38
example_conflicts = ConflictList([
39
MissingParent('Not deleting', u'p\xe5thg', '\xc3\xaedg'),
40
ContentsConflict(u'p\xe5tha', None, '\xc3\xaeda'),
41
TextConflict(u'p\xe5tha'),
42
PathConflict(u'p\xe5thb', u'p\xe5thc', '\xc3\xaedb'),
43
DuplicateID('Unversioned existing file', u'p\xe5thc', u'p\xe5thc2',
44
'\xc3\xaedc', '\xc3\xaedc'),
45
DuplicateEntry('Moved existing file to', u'p\xe5thdd.moved', u'p\xe5thd',
47
ParentLoop('Cancelled move', u'p\xe5the', u'p\xe5th2e',
49
UnversionedParent('Versioned directory', u'p\xe5thf', '\xc3\xaedf'),
53
class TestConflicts(TestCaseWithTransport):
30
55
def test_conflicts(self):
31
56
"""Conflicts are detected properly"""
32
b = Branch.initialize(u'.')
57
tree = self.make_branch_and_tree('.',
58
format=bzrdir.BzrDirFormat6())
33
60
file('hello', 'w').write('hello world4')
34
61
file('hello.THIS', 'w').write('hello world2')
35
62
file('hello.BASE', 'w').write('hello world1')
36
63
file('hello.OTHER', 'w').write('hello world3')
37
64
file('hello.sploo.BASE', 'w').write('yellow world')
38
65
file('hello.sploo.OTHER', 'w').write('yellow world2')
39
tree = b.working_tree()
40
67
self.assertEqual(len(list(tree.list_files())), 6)
41
conflicts = list(tree.iter_conflicts())
69
conflicts = tree.conflicts()
42
70
self.assertEqual(len(conflicts), 2)
43
self.assert_('hello' in conflicts)
44
self.assert_('hello.sploo' in conflicts)
71
self.assert_('hello' in conflicts[0].path)
72
self.assert_('hello.sploo' in conflicts[1].path)
46
74
restore('hello.sploo')
47
self.assertEqual(len(list(tree.iter_conflicts())), 0)
75
self.assertEqual(len(tree.conflicts()), 0)
48
76
self.assertFileEqual('hello world2', 'hello')
49
77
assert not os.path.lexists('hello.sploo')
50
78
self.assertRaises(NotConflicted, restore, 'hello')
51
79
self.assertRaises(NotConflicted, restore, 'hello.sploo')
81
def test_resolve_conflict_dir(self):
82
tree = self.make_branch_and_tree('.')
84
file('hello', 'w').write('hello world4')
85
tree.add('hello', 'q')
86
file('hello.THIS', 'w').write('hello world2')
87
file('hello.BASE', 'w').write('hello world1')
88
os.mkdir('hello.OTHER')
89
l = ConflictList([TextConflict('hello')])
92
def test_select_conflicts(self):
93
tree = self.make_branch_and_tree('.')
94
tree_conflicts = ConflictList([ContentsConflict('foo'),
95
ContentsConflict('bar')])
96
self.assertEqual((ConflictList([ContentsConflict('bar')]),
97
ConflictList([ContentsConflict('foo')])),
98
tree_conflicts.select_conflicts(tree, ['foo']))
99
self.assertEqual((ConflictList(), tree_conflicts),
100
tree_conflicts.select_conflicts(tree, [''],
101
ignore_misses=True, recurse=True))
102
tree_conflicts = ConflictList([ContentsConflict('foo/baz'),
103
ContentsConflict('bar')])
104
self.assertEqual((ConflictList([ContentsConflict('bar')]),
105
ConflictList([ContentsConflict('foo/baz')])),
106
tree_conflicts.select_conflicts(tree, ['foo'],
109
tree_conflicts = ConflictList([PathConflict('qux', 'foo/baz')])
110
self.assertEqual((ConflictList(), tree_conflicts),
111
tree_conflicts.select_conflicts(tree, ['foo'],
114
self.assertEqual((tree_conflicts, ConflictList()),
115
tree_conflicts.select_conflicts(tree, ['foo'],
119
class TestConflictStanzas(TestCase):
121
def test_stanza_roundtrip(self):
122
# write and read our example stanza.
123
stanza_iter = example_conflicts.to_stanzas()
124
processed = ConflictList.from_stanzas(stanza_iter)
125
for o, p in zip(processed, example_conflicts):
126
self.assertEqual(o, p)
128
self.assertIsInstance(o.path, unicode)
130
if o.file_id is not None:
131
self.assertIsInstance(o.file_id, str)
133
conflict_path = getattr(o, 'conflict_path', None)
134
if conflict_path is not None:
135
self.assertIsInstance(conflict_path, unicode)
137
conflict_file_id = getattr(o, 'conflict_file_id', None)
138
if conflict_file_id is not None:
139
self.assertIsInstance(conflict_file_id, str)
141
def test_stanzification(self):
142
for stanza in example_conflicts.to_stanzas():
143
if 'file_id' in stanza:
144
# In Stanza form, the file_id has to be unicode.
145
self.assertStartsWith(stanza['file_id'], u'\xeed')
146
self.assertStartsWith(stanza['path'], u'p\xe5th')
147
if 'conflict_path' in stanza:
148
self.assertStartsWith(stanza['conflict_path'], u'p\xe5th')
149
if 'conflict_file_id' in stanza:
150
self.assertStartsWith(stanza['conflict_file_id'], u'\xeed')