4
from bzrlib.tests import TestCaseInTempDir
6
from bzrlib.commit import commit
7
from bzrlib.add import smart_add
8
from bzrlib.branch import Branch
9
from bzrlib.clone import copy_branch
10
from bzrlib.merge import merge
11
from bzrlib.workingtree import WorkingTree
12
from bzrlib.delta import compare_trees
14
class TestFileIdInvolved(TestCaseInTempDir):
16
def touch(self,filename):
17
f = file(filename,"a")
18
f.write("appended line\n")
22
def merge( self, branch_from, force=False ):
23
from bzrlib.merge_core import ApplyMerge3
25
merge([branch_from,-1],[None,None], merge_type=ApplyMerge3,
26
check_clean=(not force) )
29
super(TestFileIdInvolved, self).setUp()
30
# create three branches, and merge it
32
# /-->J ------>K (branch2)
34
# A ---> B --->C ---->D->G (main)
36
# \---> E---/----> F (branch1)
41
main_branch = Branch.initialize('.')
42
self.build_tree(["a","b","c"])
46
wt.add(['a', 'b', 'c'], ['a-file-id-2006-01-01-abcd',
47
'b-file-id-2006-01-01-defg',
48
'c-funky<file-id> quiji%bo'])
49
commit(b, "Commit one", rev_id="rev-A")
51
#-------- end A -----------
53
copy_branch(main_branch,"../branch1")
54
os.chdir("../branch1")
56
#branch1_branch = Branch.open(".")
57
self.build_tree(["d"])
59
commit(Branch.open("."), "branch1, Commit one", rev_id="rev-E")
61
#-------- end E -----------
65
commit(Branch.open("."), "Commit two", rev_id="rev-B")
67
#-------- end B -----------
69
copy_branch(Branch.open("."),"../branch2")
70
os.chdir("../branch2")
72
branch2_branch = Branch.open(".")
74
commit(Branch.open("."), "branch2, Commit one", rev_id="rev-J")
76
#-------- end J -----------
80
self.merge("../branch1")
81
commit(Branch.open("."), "merge branch1, rev-11", rev_id="rev-C")
83
#-------- end C -----------
85
os.chdir("../branch1")
86
tree = WorkingTree('.', Branch.open("."))
87
tree.rename_one("d","e")
88
commit(Branch.open("."), "branch1, commit two", rev_id="rev-F")
91
#-------- end F -----------
93
os.chdir("../branch2")
96
commit(Branch.open("."), "branch2, commit two", rev_id="rev-K")
98
#-------- end K -----------
103
self.merge("../branch1",force=True)
105
# D gets some funky characters to make sure the unescaping works
106
commit(Branch.open("."), "merge branch1, rev-12", rev_id="rev-<D>")
110
self.merge("../branch2")
111
commit(Branch.open("."), "merge branch1, rev-22", rev_id="rev-G")
115
self.branch = Branch.open(".")
118
def test_fileid_involved_all_revs(self):
120
l = self.branch.fileid_involved( )
121
self.assertEquals( sorted(map( lambda x: x[0], l )), ["a","b","c","d"])
123
def test_fileid_involved_one_rev(self):
125
l = self.branch.fileid_involved("rev-B" )
126
self.assertEquals( sorted(map( lambda x: x[0], l )), ["a","b","c"])
128
def test_fileid_involved_two_revs(self):
130
l = self.branch.fileid_involved_between_revs("rev-B","rev-K" )
131
self.assertEquals( sorted(map( lambda x: x[0], l )), ["b","c"])
133
l = self.branch.fileid_involved_between_revs("rev-C","rev-<D>" )
134
self.assertEquals( sorted(map( lambda x: x[0], l )), ["b","d"])
136
l = self.branch.fileid_involved_between_revs("rev-C","rev-G" )
137
self.assertEquals( sorted(map( lambda x: x[0], l )), ["b","c","d"])
139
l = self.branch.fileid_involved_between_revs("rev-E","rev-G" )
140
self.assertEquals( sorted(map( lambda x: x[0], l )), ["a", "b","c","d"])
143
def test_fileid_involved_sets(self):
145
l = self.branch.fileid_involved_by_set(set(["rev-B"]))
146
self.assertEquals( sorted(map( lambda x: x[0], l )), ["a"])
148
l = self.branch.fileid_involved_by_set(set(["rev-<D>"]))
149
self.assertEquals( sorted(map( lambda x: x[0], l )), ["b"])
151
def test_fileid_involved_compare(self):
153
l1 = self.branch.fileid_involved_between_revs("rev-E", "rev-<D>")
154
l2 = self.branch.fileid_involved_by_set(set(["rev-<D>","rev-F","rev-C","rev-B"]))
155
self.assertEquals( l1, l2 )
157
l1 = self.branch.fileid_involved_between_revs("rev-C", "rev-G")
158
l2 = self.branch.fileid_involved_by_set(
159
set(["rev-G","rev-<D>","rev-F","rev-K","rev-J"]))
160
self.assertEquals( l1, l2 )
162
def test_fileid_involved_full_compare(self):
163
from bzrlib.tsort import topo_sort
165
history = self.branch.revision_history( )
167
if len(history) < 2: return
169
for start in range(0,len(history)-1):
170
for end in range(start+1,len(history)):
172
l1 = self.branch.fileid_involved_between_revs(
173
history[start], history[end])
175
old_tree = self.branch.revision_tree(history[start])
176
new_tree = self.branch.revision_tree(history[end])
177
delta = compare_trees(old_tree, new_tree )
179
l2 = [ id for path, id, kind in delta.added ] + \
180
[ id for oldpath, newpath, id, kind, text_modified, \
181
meta_modified in delta.renamed ] + \
182
[ id for path, id, kind, text_modified, meta_modified in \
185
self.assertEquals( l1, set(l2) )