1301
1726
# children of non-root directories should not be renamed
1302
1727
self.assertEqual(2, transform_result.rename_count)
1729
def create_ab_tree(self):
1730
"""Create a committed test tree with two files"""
1731
source = self.make_branch_and_tree('source')
1732
self.build_tree_contents([('source/file1', 'A')])
1733
self.build_tree_contents([('source/file2', 'B')])
1734
source.add(['file1', 'file2'], ['file1-id', 'file2-id'])
1735
source.commit('commit files')
1737
self.addCleanup(source.unlock)
1740
def test_build_tree_accelerator_tree(self):
1741
source = self.create_ab_tree()
1742
self.build_tree_contents([('source/file2', 'C')])
1744
real_source_get_file = source.get_file
1745
def get_file(file_id, path=None):
1746
calls.append(file_id)
1747
return real_source_get_file(file_id, path)
1748
source.get_file = get_file
1749
target = self.make_branch_and_tree('target')
1750
revision_tree = source.basis_tree()
1751
revision_tree.lock_read()
1752
self.addCleanup(revision_tree.unlock)
1753
build_tree(revision_tree, target, source)
1754
self.assertEqual(['file1-id'], calls)
1756
self.addCleanup(target.unlock)
1757
self.assertEqual([], list(target.iter_changes(revision_tree)))
1759
def test_build_tree_accelerator_tree_missing_file(self):
1760
source = self.create_ab_tree()
1761
os.unlink('source/file1')
1762
source.remove(['file2'])
1763
target = self.make_branch_and_tree('target')
1764
revision_tree = source.basis_tree()
1765
revision_tree.lock_read()
1766
self.addCleanup(revision_tree.unlock)
1767
build_tree(revision_tree, target, source)
1769
self.addCleanup(target.unlock)
1770
self.assertEqual([], list(target.iter_changes(revision_tree)))
1772
def test_build_tree_accelerator_wrong_kind(self):
1773
self.requireFeature(SymlinkFeature)
1774
source = self.make_branch_and_tree('source')
1775
self.build_tree_contents([('source/file1', '')])
1776
self.build_tree_contents([('source/file2', '')])
1777
source.add(['file1', 'file2'], ['file1-id', 'file2-id'])
1778
source.commit('commit files')
1779
os.unlink('source/file2')
1780
self.build_tree_contents([('source/file2/', 'C')])
1781
os.unlink('source/file1')
1782
os.symlink('file2', 'source/file1')
1784
real_source_get_file = source.get_file
1785
def get_file(file_id, path=None):
1786
calls.append(file_id)
1787
return real_source_get_file(file_id, path)
1788
source.get_file = get_file
1789
target = self.make_branch_and_tree('target')
1790
revision_tree = source.basis_tree()
1791
revision_tree.lock_read()
1792
self.addCleanup(revision_tree.unlock)
1793
build_tree(revision_tree, target, source)
1794
self.assertEqual([], calls)
1796
self.addCleanup(target.unlock)
1797
self.assertEqual([], list(target.iter_changes(revision_tree)))
1799
def test_build_tree_hardlink(self):
1800
self.requireFeature(HardlinkFeature)
1801
source = self.create_ab_tree()
1802
target = self.make_branch_and_tree('target')
1803
revision_tree = source.basis_tree()
1804
revision_tree.lock_read()
1805
self.addCleanup(revision_tree.unlock)
1806
build_tree(revision_tree, target, source, hardlink=True)
1808
self.addCleanup(target.unlock)
1809
self.assertEqual([], list(target.iter_changes(revision_tree)))
1810
source_stat = os.stat('source/file1')
1811
target_stat = os.stat('target/file1')
1812
self.assertEqual(source_stat, target_stat)
1814
# Explicitly disallowing hardlinks should prevent them.
1815
target2 = self.make_branch_and_tree('target2')
1816
build_tree(revision_tree, target2, source, hardlink=False)
1818
self.addCleanup(target2.unlock)
1819
self.assertEqual([], list(target2.iter_changes(revision_tree)))
1820
source_stat = os.stat('source/file1')
1821
target2_stat = os.stat('target2/file1')
1822
self.assertNotEqual(source_stat, target2_stat)
1824
def test_build_tree_accelerator_tree_moved(self):
1825
source = self.make_branch_and_tree('source')
1826
self.build_tree_contents([('source/file1', 'A')])
1827
source.add(['file1'], ['file1-id'])
1828
source.commit('commit files')
1829
source.rename_one('file1', 'file2')
1831
self.addCleanup(source.unlock)
1832
target = self.make_branch_and_tree('target')
1833
revision_tree = source.basis_tree()
1834
revision_tree.lock_read()
1835
self.addCleanup(revision_tree.unlock)
1836
build_tree(revision_tree, target, source)
1838
self.addCleanup(target.unlock)
1839
self.assertEqual([], list(target.iter_changes(revision_tree)))
1841
def test_build_tree_hardlinks_preserve_execute(self):
1842
self.requireFeature(HardlinkFeature)
1843
source = self.create_ab_tree()
1844
tt = TreeTransform(source)
1845
trans_id = tt.trans_id_tree_file_id('file1-id')
1846
tt.set_executability(True, trans_id)
1848
self.assertTrue(source.is_executable('file1-id'))
1849
target = self.make_branch_and_tree('target')
1850
revision_tree = source.basis_tree()
1851
revision_tree.lock_read()
1852
self.addCleanup(revision_tree.unlock)
1853
build_tree(revision_tree, target, source, hardlink=True)
1855
self.addCleanup(target.unlock)
1856
self.assertEqual([], list(target.iter_changes(revision_tree)))
1857
self.assertTrue(source.is_executable('file1-id'))
1859
def test_case_insensitive_build_tree_inventory(self):
1860
if (tests.CaseInsensitiveFilesystemFeature.available()
1861
or tests.CaseInsCasePresFilenameFeature.available()):
1862
raise tests.UnavailableFeature('Fully case sensitive filesystem')
1863
source = self.make_branch_and_tree('source')
1864
self.build_tree(['source/file', 'source/FILE'])
1865
source.add(['file', 'FILE'], ['lower-id', 'upper-id'])
1866
source.commit('added files')
1867
# Don't try this at home, kids!
1868
# Force the tree to report that it is case insensitive
1869
target = self.make_branch_and_tree('target')
1870
target.case_sensitive = False
1871
build_tree(source.basis_tree(), target, source, delta_from_tree=True)
1872
self.assertEqual('file.moved', target.id2path('lower-id'))
1873
self.assertEqual('FILE', target.id2path('upper-id'))
1876
class TestCommitTransform(tests.TestCaseWithTransport):
1878
def get_branch(self):
1879
tree = self.make_branch_and_tree('tree')
1881
self.addCleanup(tree.unlock)
1882
tree.commit('empty commit')
1885
def get_branch_and_transform(self):
1886
branch = self.get_branch()
1887
tt = TransformPreview(branch.basis_tree())
1888
self.addCleanup(tt.finalize)
1891
def test_commit_wrong_basis(self):
1892
branch = self.get_branch()
1893
basis = branch.repository.revision_tree(
1894
_mod_revision.NULL_REVISION)
1895
tt = TransformPreview(basis)
1896
self.addCleanup(tt.finalize)
1897
e = self.assertRaises(ValueError, tt.commit, branch, '')
1898
self.assertEqual('TreeTransform not based on branch basis: null:',
1901
def test_empy_commit(self):
1902
branch, tt = self.get_branch_and_transform()
1903
rev = tt.commit(branch, 'my message')
1904
self.assertEqual(2, branch.revno())
1905
repo = branch.repository
1906
self.assertEqual('my message', repo.get_revision(rev).message)
1908
def test_merge_parents(self):
1909
branch, tt = self.get_branch_and_transform()
1910
rev = tt.commit(branch, 'my message', ['rev1b', 'rev1c'])
1911
self.assertEqual(['rev1b', 'rev1c'],
1912
branch.basis_tree().get_parent_ids()[1:])
1914
def test_first_commit(self):
1915
branch = self.make_branch('branch')
1917
self.addCleanup(branch.unlock)
1918
tt = TransformPreview(branch.basis_tree())
1919
tt.new_directory('', ROOT_PARENT, 'TREE_ROOT')
1920
rev = tt.commit(branch, 'my message')
1921
self.assertEqual([], branch.basis_tree().get_parent_ids())
1922
self.assertNotEqual(_mod_revision.NULL_REVISION,
1923
branch.last_revision())
1925
def test_first_commit_with_merge_parents(self):
1926
branch = self.make_branch('branch')
1928
self.addCleanup(branch.unlock)
1929
tt = TransformPreview(branch.basis_tree())
1930
e = self.assertRaises(ValueError, tt.commit, branch,
1931
'my message', ['rev1b-id'])
1932
self.assertEqual('Cannot supply merge parents for first commit.',
1934
self.assertEqual(_mod_revision.NULL_REVISION, branch.last_revision())
1936
def test_add_files(self):
1937
branch, tt = self.get_branch_and_transform()
1938
tt.new_file('file', tt.root, 'contents', 'file-id')
1939
trans_id = tt.new_directory('dir', tt.root, 'dir-id')
1940
tt.new_symlink('symlink', trans_id, 'target', 'symlink-id')
1941
rev = tt.commit(branch, 'message')
1942
tree = branch.basis_tree()
1943
self.assertEqual('file', tree.id2path('file-id'))
1944
self.assertEqual('contents', tree.get_file_text('file-id'))
1945
self.assertEqual('dir', tree.id2path('dir-id'))
1946
self.assertEqual('dir/symlink', tree.id2path('symlink-id'))
1947
self.assertEqual('target', tree.get_symlink_target('symlink-id'))
1949
def test_add_unversioned(self):
1950
branch, tt = self.get_branch_and_transform()
1951
tt.new_file('file', tt.root, 'contents')
1952
self.assertRaises(errors.StrictCommitFailed, tt.commit, branch,
1953
'message', strict=True)
1955
def test_modify_strict(self):
1956
branch, tt = self.get_branch_and_transform()
1957
tt.new_file('file', tt.root, 'contents', 'file-id')
1958
tt.commit(branch, 'message', strict=True)
1959
tt = TransformPreview(branch.basis_tree())
1960
trans_id = tt.trans_id_file_id('file-id')
1961
tt.delete_contents(trans_id)
1962
tt.create_file('contents', trans_id)
1963
tt.commit(branch, 'message', strict=True)
1965
def test_commit_malformed(self):
1966
"""Committing a malformed transform should raise an exception.
1968
In this case, we are adding a file without adding its parent.
1970
branch, tt = self.get_branch_and_transform()
1971
parent_id = tt.trans_id_file_id('parent-id')
1972
tt.new_file('file', parent_id, 'contents', 'file-id')
1973
self.assertRaises(errors.MalformedTransform, tt.commit, branch,
1305
1977
class MockTransform(object):
1332
2005
self.assertEqual(name, 'name.~1~')
1333
2006
name = get_backup_name(MockEntry(), {'a':['1', '2', '3']}, 'a', tt)
1334
2007
self.assertEqual(name, 'name.~4~')
2010
class TestFileMover(tests.TestCaseWithTransport):
2012
def test_file_mover(self):
2013
self.build_tree(['a/', 'a/b', 'c/', 'c/d'])
2014
mover = _FileMover()
2015
mover.rename('a', 'q')
2016
self.failUnlessExists('q')
2017
self.failIfExists('a')
2018
self.failUnlessExists('q/b')
2019
self.failUnlessExists('c')
2020
self.failUnlessExists('c/d')
2022
def test_pre_delete_rollback(self):
2023
self.build_tree(['a/'])
2024
mover = _FileMover()
2025
mover.pre_delete('a', 'q')
2026
self.failUnlessExists('q')
2027
self.failIfExists('a')
2029
self.failIfExists('q')
2030
self.failUnlessExists('a')
2032
def test_apply_deletions(self):
2033
self.build_tree(['a/', 'b/'])
2034
mover = _FileMover()
2035
mover.pre_delete('a', 'q')
2036
mover.pre_delete('b', 'r')
2037
self.failUnlessExists('q')
2038
self.failUnlessExists('r')
2039
self.failIfExists('a')
2040
self.failIfExists('b')
2041
mover.apply_deletions()
2042
self.failIfExists('q')
2043
self.failIfExists('r')
2044
self.failIfExists('a')
2045
self.failIfExists('b')
2047
def test_file_mover_rollback(self):
2048
self.build_tree(['a/', 'a/b', 'c/', 'c/d/', 'c/e/'])
2049
mover = _FileMover()
2050
mover.rename('c/d', 'c/f')
2051
mover.rename('c/e', 'c/d')
2053
mover.rename('a', 'c')
2054
except errors.FileExists, e:
2056
self.failUnlessExists('a')
2057
self.failUnlessExists('c/d')
2060
class Bogus(Exception):
2064
class TestTransformRollback(tests.TestCaseWithTransport):
2066
class ExceptionFileMover(_FileMover):
2068
def __init__(self, bad_source=None, bad_target=None):
2069
_FileMover.__init__(self)
2070
self.bad_source = bad_source
2071
self.bad_target = bad_target
2073
def rename(self, source, target):
2074
if (self.bad_source is not None and
2075
source.endswith(self.bad_source)):
2077
elif (self.bad_target is not None and
2078
target.endswith(self.bad_target)):
2081
_FileMover.rename(self, source, target)
2083
def test_rollback_rename(self):
2084
tree = self.make_branch_and_tree('.')
2085
self.build_tree(['a/', 'a/b'])
2086
tt = TreeTransform(tree)
2087
self.addCleanup(tt.finalize)
2088
a_id = tt.trans_id_tree_path('a')
2089
tt.adjust_path('c', tt.root, a_id)
2090
tt.adjust_path('d', a_id, tt.trans_id_tree_path('a/b'))
2091
self.assertRaises(Bogus, tt.apply,
2092
_mover=self.ExceptionFileMover(bad_source='a'))
2093
self.failUnlessExists('a')
2094
self.failUnlessExists('a/b')
2096
self.failUnlessExists('c')
2097
self.failUnlessExists('c/d')
2099
def test_rollback_rename_into_place(self):
2100
tree = self.make_branch_and_tree('.')
2101
self.build_tree(['a/', 'a/b'])
2102
tt = TreeTransform(tree)
2103
self.addCleanup(tt.finalize)
2104
a_id = tt.trans_id_tree_path('a')
2105
tt.adjust_path('c', tt.root, a_id)
2106
tt.adjust_path('d', a_id, tt.trans_id_tree_path('a/b'))
2107
self.assertRaises(Bogus, tt.apply,
2108
_mover=self.ExceptionFileMover(bad_target='c/d'))
2109
self.failUnlessExists('a')
2110
self.failUnlessExists('a/b')
2112
self.failUnlessExists('c')
2113
self.failUnlessExists('c/d')
2115
def test_rollback_deletion(self):
2116
tree = self.make_branch_and_tree('.')
2117
self.build_tree(['a/', 'a/b'])
2118
tt = TreeTransform(tree)
2119
self.addCleanup(tt.finalize)
2120
a_id = tt.trans_id_tree_path('a')
2121
tt.delete_contents(a_id)
2122
tt.adjust_path('d', tt.root, tt.trans_id_tree_path('a/b'))
2123
self.assertRaises(Bogus, tt.apply,
2124
_mover=self.ExceptionFileMover(bad_target='d'))
2125
self.failUnlessExists('a')
2126
self.failUnlessExists('a/b')
2128
def test_resolve_no_parent(self):
2129
wt = self.make_branch_and_tree('.')
2130
tt = TreeTransform(wt)
2131
self.addCleanup(tt.finalize)
2132
parent = tt.trans_id_file_id('parent-id')
2133
tt.new_file('file', parent, 'Contents')
2134
resolve_conflicts(tt)
2137
A_ENTRY = ('a-id', ('a', 'a'), True, (True, True),
2138
('TREE_ROOT', 'TREE_ROOT'), ('a', 'a'), ('file', 'file'),
2140
ROOT_ENTRY = ('TREE_ROOT', ('', ''), False, (True, True), (None, None),
2141
('', ''), ('directory', 'directory'), (False, None))
2144
class TestTransformPreview(tests.TestCaseWithTransport):
2146
def create_tree(self):
2147
tree = self.make_branch_and_tree('.')
2148
self.build_tree_contents([('a', 'content 1')])
2149
tree.add('a', 'a-id')
2150
tree.commit('rev1', rev_id='rev1')
2151
return tree.branch.repository.revision_tree('rev1')
2153
def get_empty_preview(self):
2154
repository = self.make_repository('repo')
2155
tree = repository.revision_tree(_mod_revision.NULL_REVISION)
2156
preview = TransformPreview(tree)
2157
self.addCleanup(preview.finalize)
2160
def test_transform_preview(self):
2161
revision_tree = self.create_tree()
2162
preview = TransformPreview(revision_tree)
2163
self.addCleanup(preview.finalize)
2165
def test_transform_preview_tree(self):
2166
revision_tree = self.create_tree()
2167
preview = TransformPreview(revision_tree)
2168
self.addCleanup(preview.finalize)
2169
preview.get_preview_tree()
2171
def test_transform_new_file(self):
2172
revision_tree = self.create_tree()
2173
preview = TransformPreview(revision_tree)
2174
self.addCleanup(preview.finalize)
2175
preview.new_file('file2', preview.root, 'content B\n', 'file2-id')
2176
preview_tree = preview.get_preview_tree()
2177
self.assertEqual(preview_tree.kind('file2-id'), 'file')
2179
preview_tree.get_file('file2-id').read(), 'content B\n')
2181
def test_diff_preview_tree(self):
2182
revision_tree = self.create_tree()
2183
preview = TransformPreview(revision_tree)
2184
self.addCleanup(preview.finalize)
2185
preview.new_file('file2', preview.root, 'content B\n', 'file2-id')
2186
preview_tree = preview.get_preview_tree()
2188
show_diff_trees(revision_tree, preview_tree, out)
2189
lines = out.getvalue().splitlines()
2190
self.assertEqual(lines[0], "=== added file 'file2'")
2191
# 3 lines of diff administrivia
2192
self.assertEqual(lines[4], "+content B")
2194
def test_transform_conflicts(self):
2195
revision_tree = self.create_tree()
2196
preview = TransformPreview(revision_tree)
2197
self.addCleanup(preview.finalize)
2198
preview.new_file('a', preview.root, 'content 2')
2199
resolve_conflicts(preview)
2200
trans_id = preview.trans_id_file_id('a-id')
2201
self.assertEqual('a.moved', preview.final_name(trans_id))
2203
def get_tree_and_preview_tree(self):
2204
revision_tree = self.create_tree()
2205
preview = TransformPreview(revision_tree)
2206
self.addCleanup(preview.finalize)
2207
a_trans_id = preview.trans_id_file_id('a-id')
2208
preview.delete_contents(a_trans_id)
2209
preview.create_file('b content', a_trans_id)
2210
preview_tree = preview.get_preview_tree()
2211
return revision_tree, preview_tree
2213
def test_iter_changes(self):
2214
revision_tree, preview_tree = self.get_tree_and_preview_tree()
2215
root = revision_tree.inventory.root.file_id
2216
self.assertEqual([('a-id', ('a', 'a'), True, (True, True),
2217
(root, root), ('a', 'a'), ('file', 'file'),
2219
list(preview_tree.iter_changes(revision_tree)))
2221
def test_include_unchanged_succeeds(self):
2222
revision_tree, preview_tree = self.get_tree_and_preview_tree()
2223
changes = preview_tree.iter_changes(revision_tree,
2224
include_unchanged=True)
2225
root = revision_tree.inventory.root.file_id
2227
self.assertEqual([ROOT_ENTRY, A_ENTRY], list(changes))
2229
def test_specific_files(self):
2230
revision_tree, preview_tree = self.get_tree_and_preview_tree()
2231
changes = preview_tree.iter_changes(revision_tree,
2232
specific_files=[''])
2233
self.assertEqual([ROOT_ENTRY, A_ENTRY], list(changes))
2235
def test_want_unversioned(self):
2236
revision_tree, preview_tree = self.get_tree_and_preview_tree()
2237
changes = preview_tree.iter_changes(revision_tree,
2238
want_unversioned=True)
2239
self.assertEqual([ROOT_ENTRY, A_ENTRY], list(changes))
2241
def test_ignore_extra_trees_no_specific_files(self):
2242
# extra_trees is harmless without specific_files, so we'll silently
2243
# accept it, even though we won't use it.
2244
revision_tree, preview_tree = self.get_tree_and_preview_tree()
2245
preview_tree.iter_changes(revision_tree, extra_trees=[preview_tree])
2247
def test_ignore_require_versioned_no_specific_files(self):
2248
# require_versioned is meaningless without specific_files.
2249
revision_tree, preview_tree = self.get_tree_and_preview_tree()
2250
preview_tree.iter_changes(revision_tree, require_versioned=False)
2252
def test_ignore_pb(self):
2253
# pb could be supported, but TT.iter_changes doesn't support it.
2254
revision_tree, preview_tree = self.get_tree_and_preview_tree()
2255
preview_tree.iter_changes(revision_tree, pb=progress.DummyProgress())
2257
def test_kind(self):
2258
revision_tree = self.create_tree()
2259
preview = TransformPreview(revision_tree)
2260
self.addCleanup(preview.finalize)
2261
preview.new_file('file', preview.root, 'contents', 'file-id')
2262
preview.new_directory('directory', preview.root, 'dir-id')
2263
preview_tree = preview.get_preview_tree()
2264
self.assertEqual('file', preview_tree.kind('file-id'))
2265
self.assertEqual('directory', preview_tree.kind('dir-id'))
2267
def test_get_file_mtime(self):
2268
preview = self.get_empty_preview()
2269
file_trans_id = preview.new_file('file', preview.root, 'contents',
2271
limbo_path = preview._limbo_name(file_trans_id)
2272
preview_tree = preview.get_preview_tree()
2273
self.assertEqual(os.stat(limbo_path).st_mtime,
2274
preview_tree.get_file_mtime('file-id'))
2276
def test_get_file(self):
2277
preview = self.get_empty_preview()
2278
preview.new_file('file', preview.root, 'contents', 'file-id')
2279
preview_tree = preview.get_preview_tree()
2280
tree_file = preview_tree.get_file('file-id')
2282
self.assertEqual('contents', tree_file.read())
2286
def test_get_symlink_target(self):
2287
self.requireFeature(SymlinkFeature)
2288
preview = self.get_empty_preview()
2289
preview.new_symlink('symlink', preview.root, 'target', 'symlink-id')
2290
preview_tree = preview.get_preview_tree()
2291
self.assertEqual('target',
2292
preview_tree.get_symlink_target('symlink-id'))
2294
def test_all_file_ids(self):
2295
tree = self.make_branch_and_tree('tree')
2296
self.build_tree(['tree/a', 'tree/b', 'tree/c'])
2297
tree.add(['a', 'b', 'c'], ['a-id', 'b-id', 'c-id'])
2298
preview = TransformPreview(tree)
2299
self.addCleanup(preview.finalize)
2300
preview.unversion_file(preview.trans_id_file_id('b-id'))
2301
c_trans_id = preview.trans_id_file_id('c-id')
2302
preview.unversion_file(c_trans_id)
2303
preview.version_file('c-id', c_trans_id)
2304
preview_tree = preview.get_preview_tree()
2305
self.assertEqual(set(['a-id', 'c-id', tree.get_root_id()]),
2306
preview_tree.all_file_ids())
2308
def test_path2id_deleted_unchanged(self):
2309
tree = self.make_branch_and_tree('tree')
2310
self.build_tree(['tree/unchanged', 'tree/deleted'])
2311
tree.add(['unchanged', 'deleted'], ['unchanged-id', 'deleted-id'])
2312
preview = TransformPreview(tree)
2313
self.addCleanup(preview.finalize)
2314
preview.unversion_file(preview.trans_id_file_id('deleted-id'))
2315
preview_tree = preview.get_preview_tree()
2316
self.assertEqual('unchanged-id', preview_tree.path2id('unchanged'))
2317
self.assertIs(None, preview_tree.path2id('deleted'))
2319
def test_path2id_created(self):
2320
tree = self.make_branch_and_tree('tree')
2321
self.build_tree(['tree/unchanged'])
2322
tree.add(['unchanged'], ['unchanged-id'])
2323
preview = TransformPreview(tree)
2324
self.addCleanup(preview.finalize)
2325
preview.new_file('new', preview.trans_id_file_id('unchanged-id'),
2326
'contents', 'new-id')
2327
preview_tree = preview.get_preview_tree()
2328
self.assertEqual('new-id', preview_tree.path2id('unchanged/new'))
2330
def test_path2id_moved(self):
2331
tree = self.make_branch_and_tree('tree')
2332
self.build_tree(['tree/old_parent/', 'tree/old_parent/child'])
2333
tree.add(['old_parent', 'old_parent/child'],
2334
['old_parent-id', 'child-id'])
2335
preview = TransformPreview(tree)
2336
self.addCleanup(preview.finalize)
2337
new_parent = preview.new_directory('new_parent', preview.root,
2339
preview.adjust_path('child', new_parent,
2340
preview.trans_id_file_id('child-id'))
2341
preview_tree = preview.get_preview_tree()
2342
self.assertIs(None, preview_tree.path2id('old_parent/child'))
2343
self.assertEqual('child-id', preview_tree.path2id('new_parent/child'))
2345
def test_path2id_renamed_parent(self):
2346
tree = self.make_branch_and_tree('tree')
2347
self.build_tree(['tree/old_name/', 'tree/old_name/child'])
2348
tree.add(['old_name', 'old_name/child'],
2349
['parent-id', 'child-id'])
2350
preview = TransformPreview(tree)
2351
self.addCleanup(preview.finalize)
2352
preview.adjust_path('new_name', preview.root,
2353
preview.trans_id_file_id('parent-id'))
2354
preview_tree = preview.get_preview_tree()
2355
self.assertIs(None, preview_tree.path2id('old_name/child'))
2356
self.assertEqual('child-id', preview_tree.path2id('new_name/child'))
2358
def assertMatchingIterEntries(self, tt, specific_file_ids=None):
2359
preview_tree = tt.get_preview_tree()
2360
preview_result = list(preview_tree.iter_entries_by_dir(
2364
actual_result = list(tree.iter_entries_by_dir(specific_file_ids))
2365
self.assertEqual(actual_result, preview_result)
2367
def test_iter_entries_by_dir_new(self):
2368
tree = self.make_branch_and_tree('tree')
2369
tt = TreeTransform(tree)
2370
tt.new_file('new', tt.root, 'contents', 'new-id')
2371
self.assertMatchingIterEntries(tt)
2373
def test_iter_entries_by_dir_deleted(self):
2374
tree = self.make_branch_and_tree('tree')
2375
self.build_tree(['tree/deleted'])
2376
tree.add('deleted', 'deleted-id')
2377
tt = TreeTransform(tree)
2378
tt.delete_contents(tt.trans_id_file_id('deleted-id'))
2379
self.assertMatchingIterEntries(tt)
2381
def test_iter_entries_by_dir_unversioned(self):
2382
tree = self.make_branch_and_tree('tree')
2383
self.build_tree(['tree/removed'])
2384
tree.add('removed', 'removed-id')
2385
tt = TreeTransform(tree)
2386
tt.unversion_file(tt.trans_id_file_id('removed-id'))
2387
self.assertMatchingIterEntries(tt)
2389
def test_iter_entries_by_dir_moved(self):
2390
tree = self.make_branch_and_tree('tree')
2391
self.build_tree(['tree/moved', 'tree/new_parent/'])
2392
tree.add(['moved', 'new_parent'], ['moved-id', 'new_parent-id'])
2393
tt = TreeTransform(tree)
2394
tt.adjust_path('moved', tt.trans_id_file_id('new_parent-id'),
2395
tt.trans_id_file_id('moved-id'))
2396
self.assertMatchingIterEntries(tt)
2398
def test_iter_entries_by_dir_specific_file_ids(self):
2399
tree = self.make_branch_and_tree('tree')
2400
tree.set_root_id('tree-root-id')
2401
self.build_tree(['tree/parent/', 'tree/parent/child'])
2402
tree.add(['parent', 'parent/child'], ['parent-id', 'child-id'])
2403
tt = TreeTransform(tree)
2404
self.assertMatchingIterEntries(tt, ['tree-root-id', 'child-id'])
2406
def test_symlink_content_summary(self):
2407
self.requireFeature(SymlinkFeature)
2408
preview = self.get_empty_preview()
2409
preview.new_symlink('path', preview.root, 'target', 'path-id')
2410
summary = preview.get_preview_tree().path_content_summary('path')
2411
self.assertEqual(('symlink', None, None, 'target'), summary)
2413
def test_missing_content_summary(self):
2414
preview = self.get_empty_preview()
2415
summary = preview.get_preview_tree().path_content_summary('path')
2416
self.assertEqual(('missing', None, None, None), summary)
2418
def test_deleted_content_summary(self):
2419
tree = self.make_branch_and_tree('tree')
2420
self.build_tree(['tree/path/'])
2422
preview = TransformPreview(tree)
2423
self.addCleanup(preview.finalize)
2424
preview.delete_contents(preview.trans_id_tree_path('path'))
2425
summary = preview.get_preview_tree().path_content_summary('path')
2426
self.assertEqual(('missing', None, None, None), summary)
2428
def test_file_content_summary_executable(self):
2429
if not osutils.supports_executable():
2430
raise TestNotApplicable()
2431
preview = self.get_empty_preview()
2432
path_id = preview.new_file('path', preview.root, 'contents', 'path-id')
2433
preview.set_executability(True, path_id)
2434
summary = preview.get_preview_tree().path_content_summary('path')
2435
self.assertEqual(4, len(summary))
2436
self.assertEqual('file', summary[0])
2437
# size must be known
2438
self.assertEqual(len('contents'), summary[1])
2440
self.assertEqual(True, summary[2])
2441
# will not have hash (not cheap to determine)
2442
self.assertIs(None, summary[3])
2444
def test_change_executability(self):
2445
if not osutils.supports_executable():
2446
raise TestNotApplicable()
2447
tree = self.make_branch_and_tree('tree')
2448
self.build_tree(['tree/path'])
2450
preview = TransformPreview(tree)
2451
self.addCleanup(preview.finalize)
2452
path_id = preview.trans_id_tree_path('path')
2453
preview.set_executability(True, path_id)
2454
summary = preview.get_preview_tree().path_content_summary('path')
2455
self.assertEqual(True, summary[2])
2457
def test_file_content_summary_non_exec(self):
2458
preview = self.get_empty_preview()
2459
preview.new_file('path', preview.root, 'contents', 'path-id')
2460
summary = preview.get_preview_tree().path_content_summary('path')
2461
self.assertEqual(4, len(summary))
2462
self.assertEqual('file', summary[0])
2463
# size must be known
2464
self.assertEqual(len('contents'), summary[1])
2466
if osutils.supports_executable():
2467
self.assertEqual(False, summary[2])
2469
self.assertEqual(None, summary[2])
2470
# will not have hash (not cheap to determine)
2471
self.assertIs(None, summary[3])
2473
def test_dir_content_summary(self):
2474
preview = self.get_empty_preview()
2475
preview.new_directory('path', preview.root, 'path-id')
2476
summary = preview.get_preview_tree().path_content_summary('path')
2477
self.assertEqual(('directory', None, None, None), summary)
2479
def test_tree_content_summary(self):
2480
preview = self.get_empty_preview()
2481
path = preview.new_directory('path', preview.root, 'path-id')
2482
preview.set_tree_reference('rev-1', path)
2483
summary = preview.get_preview_tree().path_content_summary('path')
2484
self.assertEqual(4, len(summary))
2485
self.assertEqual('tree-reference', summary[0])
2487
def test_annotate(self):
2488
tree = self.make_branch_and_tree('tree')
2489
self.build_tree_contents([('tree/file', 'a\n')])
2490
tree.add('file', 'file-id')
2491
tree.commit('a', rev_id='one')
2492
self.build_tree_contents([('tree/file', 'a\nb\n')])
2493
preview = TransformPreview(tree)
2494
self.addCleanup(preview.finalize)
2495
file_trans_id = preview.trans_id_file_id('file-id')
2496
preview.delete_contents(file_trans_id)
2497
preview.create_file('a\nb\nc\n', file_trans_id)
2498
preview_tree = preview.get_preview_tree()
2504
annotation = preview_tree.annotate_iter('file-id', 'me:')
2505
self.assertEqual(expected, annotation)
2507
def test_annotate_missing(self):
2508
preview = self.get_empty_preview()
2509
preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
2510
preview_tree = preview.get_preview_tree()
2516
annotation = preview_tree.annotate_iter('file-id', 'me:')
2517
self.assertEqual(expected, annotation)
2519
def test_annotate_rename(self):
2520
tree = self.make_branch_and_tree('tree')
2521
self.build_tree_contents([('tree/file', 'a\n')])
2522
tree.add('file', 'file-id')
2523
tree.commit('a', rev_id='one')
2524
preview = TransformPreview(tree)
2525
self.addCleanup(preview.finalize)
2526
file_trans_id = preview.trans_id_file_id('file-id')
2527
preview.adjust_path('newname', preview.root, file_trans_id)
2528
preview_tree = preview.get_preview_tree()
2532
annotation = preview_tree.annotate_iter('file-id', 'me:')
2533
self.assertEqual(expected, annotation)
2535
def test_annotate_deleted(self):
2536
tree = self.make_branch_and_tree('tree')
2537
self.build_tree_contents([('tree/file', 'a\n')])
2538
tree.add('file', 'file-id')
2539
tree.commit('a', rev_id='one')
2540
self.build_tree_contents([('tree/file', 'a\nb\n')])
2541
preview = TransformPreview(tree)
2542
self.addCleanup(preview.finalize)
2543
file_trans_id = preview.trans_id_file_id('file-id')
2544
preview.delete_contents(file_trans_id)
2545
preview_tree = preview.get_preview_tree()
2546
annotation = preview_tree.annotate_iter('file-id', 'me:')
2547
self.assertIs(None, annotation)
2549
def test_stored_kind(self):
2550
preview = self.get_empty_preview()
2551
preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
2552
preview_tree = preview.get_preview_tree()
2553
self.assertEqual('file', preview_tree.stored_kind('file-id'))
2555
def test_is_executable(self):
2556
preview = self.get_empty_preview()
2557
preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
2558
preview.set_executability(True, preview.trans_id_file_id('file-id'))
2559
preview_tree = preview.get_preview_tree()
2560
self.assertEqual(True, preview_tree.is_executable('file-id'))
2562
def test_get_set_parent_ids(self):
2563
revision_tree, preview_tree = self.get_tree_and_preview_tree()
2564
self.assertEqual([], preview_tree.get_parent_ids())
2565
preview_tree.set_parent_ids(['rev-1'])
2566
self.assertEqual(['rev-1'], preview_tree.get_parent_ids())
2568
def test_plan_file_merge(self):
2569
work_a = self.make_branch_and_tree('wta')
2570
self.build_tree_contents([('wta/file', 'a\nb\nc\nd\n')])
2571
work_a.add('file', 'file-id')
2572
base_id = work_a.commit('base version')
2573
tree_b = work_a.bzrdir.sprout('wtb').open_workingtree()
2574
preview = TransformPreview(work_a)
2575
self.addCleanup(preview.finalize)
2576
trans_id = preview.trans_id_file_id('file-id')
2577
preview.delete_contents(trans_id)
2578
preview.create_file('b\nc\nd\ne\n', trans_id)
2579
self.build_tree_contents([('wtb/file', 'a\nc\nd\nf\n')])
2580
tree_a = preview.get_preview_tree()
2581
tree_a.set_parent_ids([base_id])
2583
('killed-a', 'a\n'),
2584
('killed-b', 'b\n'),
2585
('unchanged', 'c\n'),
2586
('unchanged', 'd\n'),
2589
], list(tree_a.plan_file_merge('file-id', tree_b)))
2591
def test_plan_file_merge_revision_tree(self):
2592
work_a = self.make_branch_and_tree('wta')
2593
self.build_tree_contents([('wta/file', 'a\nb\nc\nd\n')])
2594
work_a.add('file', 'file-id')
2595
base_id = work_a.commit('base version')
2596
tree_b = work_a.bzrdir.sprout('wtb').open_workingtree()
2597
preview = TransformPreview(work_a.basis_tree())
2598
self.addCleanup(preview.finalize)
2599
trans_id = preview.trans_id_file_id('file-id')
2600
preview.delete_contents(trans_id)
2601
preview.create_file('b\nc\nd\ne\n', trans_id)
2602
self.build_tree_contents([('wtb/file', 'a\nc\nd\nf\n')])
2603
tree_a = preview.get_preview_tree()
2604
tree_a.set_parent_ids([base_id])
2606
('killed-a', 'a\n'),
2607
('killed-b', 'b\n'),
2608
('unchanged', 'c\n'),
2609
('unchanged', 'd\n'),
2612
], list(tree_a.plan_file_merge('file-id', tree_b)))
2614
def test_walkdirs(self):
2615
preview = self.get_empty_preview()
2616
preview.version_file('tree-root', preview.root)
2617
preview_tree = preview.get_preview_tree()
2618
file_trans_id = preview.new_file('a', preview.root, 'contents',
2620
expected = [(('', 'tree-root'),
2621
[('a', 'a', 'file', None, 'a-id', 'file')])]
2622
self.assertEqual(expected, list(preview_tree.walkdirs()))
2624
def test_extras(self):
2625
work_tree = self.make_branch_and_tree('tree')
2626
self.build_tree(['tree/removed-file', 'tree/existing-file',
2627
'tree/not-removed-file'])
2628
work_tree.add(['removed-file', 'not-removed-file'])
2629
preview = TransformPreview(work_tree)
2630
self.addCleanup(preview.finalize)
2631
preview.new_file('new-file', preview.root, 'contents')
2632
preview.new_file('new-versioned-file', preview.root, 'contents',
2634
tree = preview.get_preview_tree()
2635
preview.unversion_file(preview.trans_id_tree_path('removed-file'))
2636
self.assertEqual(set(['new-file', 'removed-file', 'existing-file']),
2639
def test_merge_into_preview(self):
2640
work_tree = self.make_branch_and_tree('tree')
2641
self.build_tree_contents([('tree/file','b\n')])
2642
work_tree.add('file', 'file-id')
2643
work_tree.commit('first commit')
2644
child_tree = work_tree.bzrdir.sprout('child').open_workingtree()
2645
self.build_tree_contents([('child/file','b\nc\n')])
2646
child_tree.commit('child commit')
2647
child_tree.lock_write()
2648
self.addCleanup(child_tree.unlock)
2649
work_tree.lock_write()
2650
self.addCleanup(work_tree.unlock)
2651
preview = TransformPreview(work_tree)
2652
self.addCleanup(preview.finalize)
2653
preview_tree = preview.get_preview_tree()
2654
file_trans_id = preview.trans_id_file_id('file-id')
2655
preview.delete_contents(file_trans_id)
2656
preview.create_file('a\nb\n', file_trans_id)
2657
pb = progress.DummyProgress()
2658
merger = Merger.from_revision_ids(pb, preview_tree,
2659
child_tree.branch.last_revision(),
2660
other_branch=child_tree.branch,
2661
tree_branch=work_tree.branch)
2662
merger.merge_type = Merge3Merger
2663
tt = merger.make_merger().make_preview_transform()
2664
self.addCleanup(tt.finalize)
2665
final_tree = tt.get_preview_tree()
2666
self.assertEqual('a\nb\nc\n', final_tree.get_file_text('file-id'))
2668
def test_merge_preview_into_workingtree(self):
2669
tree = self.make_branch_and_tree('tree')
2670
tt = TransformPreview(tree)
2671
self.addCleanup(tt.finalize)
2672
tt.new_file('name', tt.root, 'content', 'file-id')
2673
tree2 = self.make_branch_and_tree('tree2')
2674
pb = progress.DummyProgress()
2675
merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
2676
pb, tree.basis_tree())
2677
merger.merge_type = Merge3Merger
2680
def test_merge_preview_into_workingtree_handles_conflicts(self):
2681
tree = self.make_branch_and_tree('tree')
2682
self.build_tree_contents([('tree/foo', 'bar')])
2683
tree.add('foo', 'foo-id')
2685
tt = TransformPreview(tree)
2686
self.addCleanup(tt.finalize)
2687
trans_id = tt.trans_id_file_id('foo-id')
2688
tt.delete_contents(trans_id)
2689
tt.create_file('baz', trans_id)
2690
tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
2691
self.build_tree_contents([('tree2/foo', 'qux')])
2692
pb = progress.DummyProgress()
2693
merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
2694
pb, tree.basis_tree())
2695
merger.merge_type = Merge3Merger
2698
def test_is_executable(self):
2699
tree = self.make_branch_and_tree('tree')
2700
preview = TransformPreview(tree)
2701
self.addCleanup(preview.finalize)
2702
preview.new_file('foo', preview.root, 'bar', 'baz-id')
2703
preview_tree = preview.get_preview_tree()
2704
self.assertEqual(False, preview_tree.is_executable('baz-id',
2706
self.assertEqual(False, preview_tree.is_executable('baz-id'))
2708
def test_commit_preview_tree(self):
2709
tree = self.make_branch_and_tree('tree')
2710
rev_id = tree.commit('rev1')
2711
tree.branch.lock_write()
2712
self.addCleanup(tree.branch.unlock)
2713
tt = TransformPreview(tree)
2714
tt.new_file('file', tt.root, 'contents', 'file_id')
2715
self.addCleanup(tt.finalize)
2716
preview = tt.get_preview_tree()
2717
preview.set_parent_ids([rev_id])
2718
builder = tree.branch.get_commit_builder([rev_id])
2719
list(builder.record_iter_changes(preview, rev_id, tt.iter_changes()))
2720
builder.finish_inventory()
2721
rev2_id = builder.commit('rev2')
2722
rev2_tree = tree.branch.repository.revision_tree(rev2_id)
2723
self.assertEqual('contents', rev2_tree.get_file_text('file_id'))
2726
class FakeSerializer(object):
2727
"""Serializer implementation that simply returns the input.
2729
The input is returned in the order used by pack.ContainerPushParser.
2732
def bytes_record(bytes, names):
2736
class TestSerializeTransform(tests.TestCaseWithTransport):
2738
_test_needs_features = [tests.UnicodeFilenameFeature]
2740
def get_preview(self, tree=None):
2742
tree = self.make_branch_and_tree('tree')
2743
tt = TransformPreview(tree)
2744
self.addCleanup(tt.finalize)
2747
def assertSerializesTo(self, expected, tt):
2748
records = list(tt.serialize(FakeSerializer()))
2749
self.assertEqual(expected, records)
2752
def default_attribs():
2757
'_new_executability': {},
2759
'_tree_path_ids': {'': 'new-0'},
2761
'_removed_contents': [],
2762
'_non_present_ids': {},
2765
def make_records(self, attribs, contents):
2767
(((('attribs'),),), bencode.bencode(attribs))]
2768
records.extend([(((n, k),), c) for n, k, c in contents])
2771
def creation_records(self):
2772
attribs = self.default_attribs()
2773
attribs['_id_number'] = 3
2774
attribs['_new_name'] = {
2775
'new-1': u'foo\u1234'.encode('utf-8'), 'new-2': 'qux'}
2776
attribs['_new_id'] = {'new-1': 'baz', 'new-2': 'quxx'}
2777
attribs['_new_parent'] = {'new-1': 'new-0', 'new-2': 'new-0'}
2778
attribs['_new_executability'] = {'new-1': 1}
2780
('new-1', 'file', 'i 1\nbar\n'),
2781
('new-2', 'directory', ''),
2783
return self.make_records(attribs, contents)
2785
def test_serialize_creation(self):
2786
tt = self.get_preview()
2787
tt.new_file(u'foo\u1234', tt.root, 'bar', 'baz', True)
2788
tt.new_directory('qux', tt.root, 'quxx')
2789
self.assertSerializesTo(self.creation_records(), tt)
2791
def test_deserialize_creation(self):
2792
tt = self.get_preview()
2793
tt.deserialize(iter(self.creation_records()))
2794
self.assertEqual(3, tt._id_number)
2795
self.assertEqual({'new-1': u'foo\u1234',
2796
'new-2': 'qux'}, tt._new_name)
2797
self.assertEqual({'new-1': 'baz', 'new-2': 'quxx'}, tt._new_id)
2798
self.assertEqual({'new-1': tt.root, 'new-2': tt.root}, tt._new_parent)
2799
self.assertEqual({'baz': 'new-1', 'quxx': 'new-2'}, tt._r_new_id)
2800
self.assertEqual({'new-1': True}, tt._new_executability)
2801
self.assertEqual({'new-1': 'file',
2802
'new-2': 'directory'}, tt._new_contents)
2803
foo_limbo = open(tt._limbo_name('new-1'), 'rb')
2805
foo_content = foo_limbo.read()
2808
self.assertEqual('bar', foo_content)
2810
def symlink_creation_records(self):
2811
attribs = self.default_attribs()
2812
attribs['_id_number'] = 2
2813
attribs['_new_name'] = {'new-1': u'foo\u1234'.encode('utf-8')}
2814
attribs['_new_parent'] = {'new-1': 'new-0'}
2815
contents = [('new-1', 'symlink', u'bar\u1234'.encode('utf-8'))]
2816
return self.make_records(attribs, contents)
2818
def test_serialize_symlink_creation(self):
2819
self.requireFeature(tests.SymlinkFeature)
2820
tt = self.get_preview()
2821
tt.new_symlink(u'foo\u1234', tt.root, u'bar\u1234')
2822
self.assertSerializesTo(self.symlink_creation_records(), tt)
2824
def test_deserialize_symlink_creation(self):
2825
self.requireFeature(tests.SymlinkFeature)
2826
tt = self.get_preview()
2827
tt.deserialize(iter(self.symlink_creation_records()))
2828
abspath = tt._limbo_name('new-1')
2829
foo_content = osutils.readlink(abspath)
2830
self.assertEqual(u'bar\u1234', foo_content)
2832
def make_destruction_preview(self):
2833
tree = self.make_branch_and_tree('.')
2834
self.build_tree([u'foo\u1234', 'bar'])
2835
tree.add([u'foo\u1234', 'bar'], ['foo-id', 'bar-id'])
2836
return self.get_preview(tree)
2838
def destruction_records(self):
2839
attribs = self.default_attribs()
2840
attribs['_id_number'] = 3
2841
attribs['_removed_id'] = ['new-1']
2842
attribs['_removed_contents'] = ['new-2']
2843
attribs['_tree_path_ids'] = {
2845
u'foo\u1234'.encode('utf-8'): 'new-1',
2848
return self.make_records(attribs, [])
2850
def test_serialize_destruction(self):
2851
tt = self.make_destruction_preview()
2852
foo_trans_id = tt.trans_id_tree_file_id('foo-id')
2853
tt.unversion_file(foo_trans_id)
2854
bar_trans_id = tt.trans_id_tree_file_id('bar-id')
2855
tt.delete_contents(bar_trans_id)
2856
self.assertSerializesTo(self.destruction_records(), tt)
2858
def test_deserialize_destruction(self):
2859
tt = self.make_destruction_preview()
2860
tt.deserialize(iter(self.destruction_records()))
2861
self.assertEqual({u'foo\u1234': 'new-1',
2863
'': tt.root}, tt._tree_path_ids)
2864
self.assertEqual({'new-1': u'foo\u1234',
2866
tt.root: ''}, tt._tree_id_paths)
2867
self.assertEqual(set(['new-1']), tt._removed_id)
2868
self.assertEqual(set(['new-2']), tt._removed_contents)
2870
def missing_records(self):
2871
attribs = self.default_attribs()
2872
attribs['_id_number'] = 2
2873
attribs['_non_present_ids'] = {
2875
return self.make_records(attribs, [])
2877
def test_serialize_missing(self):
2878
tt = self.get_preview()
2879
boo_trans_id = tt.trans_id_file_id('boo')
2880
self.assertSerializesTo(self.missing_records(), tt)
2882
def test_deserialize_missing(self):
2883
tt = self.get_preview()
2884
tt.deserialize(iter(self.missing_records()))
2885
self.assertEqual({'boo': 'new-1'}, tt._non_present_ids)
2887
def make_modification_preview(self):
2888
LINES_ONE = 'aa\nbb\ncc\ndd\n'
2889
LINES_TWO = 'z\nbb\nx\ndd\n'
2890
tree = self.make_branch_and_tree('tree')
2891
self.build_tree_contents([('tree/file', LINES_ONE)])
2892
tree.add('file', 'file-id')
2893
return self.get_preview(tree), LINES_TWO
2895
def modification_records(self):
2896
attribs = self.default_attribs()
2897
attribs['_id_number'] = 2
2898
attribs['_tree_path_ids'] = {
2901
attribs['_removed_contents'] = ['new-1']
2902
contents = [('new-1', 'file',
2903
'i 1\nz\n\nc 0 1 1 1\ni 1\nx\n\nc 0 3 3 1\n')]
2904
return self.make_records(attribs, contents)
2906
def test_serialize_modification(self):
2907
tt, LINES = self.make_modification_preview()
2908
trans_id = tt.trans_id_file_id('file-id')
2909
tt.delete_contents(trans_id)
2910
tt.create_file(LINES, trans_id)
2911
self.assertSerializesTo(self.modification_records(), tt)
2913
def test_deserialize_modification(self):
2914
tt, LINES = self.make_modification_preview()
2915
tt.deserialize(iter(self.modification_records()))
2916
self.assertFileEqual(LINES, tt._limbo_name('new-1'))
2918
def make_kind_change_preview(self):
2919
LINES = 'a\nb\nc\nd\n'
2920
tree = self.make_branch_and_tree('tree')
2921
self.build_tree(['tree/foo/'])
2922
tree.add('foo', 'foo-id')
2923
return self.get_preview(tree), LINES
2925
def kind_change_records(self):
2926
attribs = self.default_attribs()
2927
attribs['_id_number'] = 2
2928
attribs['_tree_path_ids'] = {
2931
attribs['_removed_contents'] = ['new-1']
2932
contents = [('new-1', 'file',
2933
'i 4\na\nb\nc\nd\n\n')]
2934
return self.make_records(attribs, contents)
2936
def test_serialize_kind_change(self):
2937
tt, LINES = self.make_kind_change_preview()
2938
trans_id = tt.trans_id_file_id('foo-id')
2939
tt.delete_contents(trans_id)
2940
tt.create_file(LINES, trans_id)
2941
self.assertSerializesTo(self.kind_change_records(), tt)
2943
def test_deserialize_kind_change(self):
2944
tt, LINES = self.make_kind_change_preview()
2945
tt.deserialize(iter(self.kind_change_records()))
2946
self.assertFileEqual(LINES, tt._limbo_name('new-1'))
2948
def make_add_contents_preview(self):
2949
LINES = 'a\nb\nc\nd\n'
2950
tree = self.make_branch_and_tree('tree')
2951
self.build_tree(['tree/foo'])
2953
os.unlink('tree/foo')
2954
return self.get_preview(tree), LINES
2956
def add_contents_records(self):
2957
attribs = self.default_attribs()
2958
attribs['_id_number'] = 2
2959
attribs['_tree_path_ids'] = {
2962
contents = [('new-1', 'file',
2963
'i 4\na\nb\nc\nd\n\n')]
2964
return self.make_records(attribs, contents)
2966
def test_serialize_add_contents(self):
2967
tt, LINES = self.make_add_contents_preview()
2968
trans_id = tt.trans_id_tree_path('foo')
2969
tt.create_file(LINES, trans_id)
2970
self.assertSerializesTo(self.add_contents_records(), tt)
2972
def test_deserialize_add_contents(self):
2973
tt, LINES = self.make_add_contents_preview()
2974
tt.deserialize(iter(self.add_contents_records()))
2975
self.assertFileEqual(LINES, tt._limbo_name('new-1'))
2977
def test_get_parents_lines(self):
2978
LINES_ONE = 'aa\nbb\ncc\ndd\n'
2979
LINES_TWO = 'z\nbb\nx\ndd\n'
2980
tree = self.make_branch_and_tree('tree')
2981
self.build_tree_contents([('tree/file', LINES_ONE)])
2982
tree.add('file', 'file-id')
2983
tt = self.get_preview(tree)
2984
trans_id = tt.trans_id_tree_path('file')
2985
self.assertEqual((['aa\n', 'bb\n', 'cc\n', 'dd\n'],),
2986
tt._get_parents_lines(trans_id))
2988
def test_get_parents_texts(self):
2989
LINES_ONE = 'aa\nbb\ncc\ndd\n'
2990
LINES_TWO = 'z\nbb\nx\ndd\n'
2991
tree = self.make_branch_and_tree('tree')
2992
self.build_tree_contents([('tree/file', LINES_ONE)])
2993
tree.add('file', 'file-id')
2994
tt = self.get_preview(tree)
2995
trans_id = tt.trans_id_tree_path('file')
2996
self.assertEqual((LINES_ONE,),
2997
tt._get_parents_texts(trans_id))
added
removed
bzrlib/tests/test_transform.py
