2 from __future__ import print_function
3 from collections import namedtuple
5 from os import environ, symlink
6 from stat import S_IFDIR, S_IFREG, S_ISDIR, S_ISREG
8 from time import localtime, strftime
12 from bup import git, metadata, vfs2 as vfs
13 from bup.git import BUP_CHUNKED
14 from bup.helpers import exc, exo, shstr
15 from bup.metadata import Metadata
16 from bup.repo import LocalRepo
17 from buptest import no_lingering_errors, test_tempdir
20 bup_tmp = os.path.realpath('../../../t/tmp')
21 bup_path = top_dir + '/bup'
22 start_dir = os.getcwd()
24 def ex(cmd, **kwargs):
25 print(shstr(cmd), file=stderr)
26 return exc(cmd, **kwargs)
28 TreeDictValue = namedtuple('TreeDictValue', ('name', 'oid', 'meta'))
30 def tree_items(repo, oid):
31 """Yield (name, entry_oid, meta) for each entry in oid. meta will be
32 a Metadata object for any non-directories and for '.', otherwise
36 # This is a simpler approach than the one in the vfs, used to
37 # cross-check its behavior.
38 tree_data, bupm_oid = vfs.tree_data_and_bupm(repo, oid)
39 bupm = vfs._FileReader(repo, bupm_oid) if bupm_oid else None
41 maybe_meta = lambda : Metadata.read(bupm) if bupm else None
45 yield TreeDictValue(name='.', oid=oid, meta=m)
46 tree_ents = vfs.ordered_tree_entries(tree_data, bupm=True)
47 for name, mangled_name, kind, gitmode, sub_oid in tree_ents:
48 if mangled_name == '.bupm':
52 if kind == BUP_CHUNKED:
53 yield TreeDictValue(name=name, oid=sub_oid,
56 yield TreeDictValue(name=name, oid=sub_oid,
57 meta=vfs.default_dir_mode)
59 yield TreeDictValue(name=name, oid=sub_oid, meta=maybe_meta())
64 def tree_dict(repo, oid):
65 return dict((x.name, x) for x in tree_items(repo, oid))
67 def run_augment_item_meta_tests(repo,
69 link_path, link_target):
70 _, file_item = vfs.resolve(repo, file_path)[-1]
71 _, link_item = vfs.lresolve(repo, link_path)[-1]
72 wvpass(isinstance(file_item.meta, Metadata))
73 wvpass(isinstance(link_item.meta, Metadata))
74 # Note: normally, modifying item.meta values is forbidden
75 file_item.meta.size = file_item.meta.size or vfs.item_size(repo, file_item)
76 link_item.meta.size = link_item.meta.size or vfs.item_size(repo, link_item)
78 ## Ensure a fully populated item is left alone
79 augmented = vfs.augment_item_meta(repo, file_item)
80 wvpass(augmented is file_item)
81 wvpass(augmented.meta is file_item.meta)
82 augmented = vfs.augment_item_meta(repo, file_item, include_size=True)
83 wvpass(augmented is file_item)
84 wvpass(augmented.meta is file_item.meta)
86 ## Ensure a missing size is handled poperly
87 file_item.meta.size = None
88 augmented = vfs.augment_item_meta(repo, file_item)
89 wvpass(augmented is file_item)
90 wvpass(augmented.meta is file_item.meta)
91 augmented = vfs.augment_item_meta(repo, file_item, include_size=True)
92 wvpass(augmented is not file_item)
93 wvpasseq(file_size, augmented.meta.size)
95 ## Ensure a meta mode is handled properly
96 mode_item = file_item._replace(meta=vfs.default_file_mode)
97 augmented = vfs.augment_item_meta(repo, mode_item)
98 augmented_w_size = vfs.augment_item_meta(repo, mode_item, include_size=True)
99 for item in (augmented, augmented_w_size):
101 wvpass(item is not file_item)
102 wvpass(isinstance(meta, Metadata))
103 wvpasseq(vfs.default_file_mode, meta.mode)
104 wvpasseq((0, 0, 0, 0, 0),
105 (meta.uid, meta.gid, meta.atime, meta.mtime, meta.ctime))
106 wvpass(augmented.meta.size is None)
107 wvpasseq(file_size, augmented_w_size.meta.size)
109 ## Ensure symlinks are handled properly
110 mode_item = link_item._replace(meta=vfs.default_symlink_mode)
111 augmented = vfs.augment_item_meta(repo, mode_item)
112 wvpass(augmented is not mode_item)
113 wvpass(isinstance(augmented.meta, Metadata))
114 wvpasseq(link_target, augmented.meta.symlink_target)
115 wvpasseq(len(link_target), augmented.meta.size)
116 augmented = vfs.augment_item_meta(repo, mode_item, include_size=True)
117 wvpass(augmented is not mode_item)
118 wvpass(isinstance(augmented.meta, Metadata))
119 wvpasseq(link_target, augmented.meta.symlink_target)
120 wvpasseq(len(link_target), augmented.meta.size)
124 def test_item_mode():
125 with no_lingering_errors():
126 mode = S_IFDIR | 0o755
127 meta = metadata.from_path('.')
129 wvpasseq(mode, vfs.item_mode(vfs.Item(oid=oid, meta=mode)))
130 wvpasseq(meta.mode, vfs.item_mode(vfs.Item(oid=oid, meta=meta)))
133 def test_reverse_suffix_duplicates():
134 suffix = lambda x: tuple(vfs._reverse_suffix_duplicates(x))
135 wvpasseq(('x',), suffix(('x',)))
136 wvpasseq(('x', 'y'), suffix(('x', 'y')))
137 wvpasseq(('x-1', 'x-0'), suffix(('x',) * 2))
138 wvpasseq(['x-%02d' % n for n in reversed(range(11))],
139 list(suffix(('x',) * 11)))
140 wvpasseq(('x-1', 'x-0', 'y'), suffix(('x', 'x', 'y')))
141 wvpasseq(('x', 'y-1', 'y-0'), suffix(('x', 'y', 'y')))
142 wvpasseq(('x', 'y-1', 'y-0', 'z'), suffix(('x', 'y', 'y', 'z')))
146 with no_lingering_errors():
147 with test_tempdir('bup-tvfs-') as tmpdir:
148 bup_dir = tmpdir + '/bup'
149 environ['GIT_DIR'] = bup_dir
150 environ['BUP_DIR'] = bup_dir
151 git.repodir = bup_dir
152 data_path = tmpdir + '/src'
154 with open(data_path + '/file', 'w+') as tmpfile:
155 tmpfile.write(b'canary\n')
156 symlink('file', data_path + '/symlink')
157 ex((bup_path, 'init'))
158 ex((bup_path, 'index', '-v', data_path))
159 ex((bup_path, 'save', '-d', '100000', '-tvvn', 'test', '--strip',
164 ls_tree = exo(('git', 'ls-tree', 'test', 'symlink'))
165 mode, typ, oidx, name = ls_tree[0].strip().split(None, 3)
166 assert name == 'symlink'
167 link_item = vfs.Item(oid=oidx.decode('hex'), meta=int(mode, 8))
168 wvpasseq('file', vfs.readlink(repo, link_item))
170 ls_tree = exo(('git', 'ls-tree', 'test', 'file'))
171 mode, typ, oidx, name = ls_tree[0].strip().split(None, 3)
172 assert name == 'file'
173 file_item = vfs.Item(oid=oidx.decode('hex'), meta=int(mode, 8))
174 wvexcept(Exception, vfs.readlink, repo, file_item)
177 wvpasseq(4, vfs.item_size(repo, link_item))
178 wvpasseq(7, vfs.item_size(repo, file_item))
179 meta = metadata.from_path(__file__)
181 fake_item = file_item._replace(meta=meta)
182 wvpasseq(42, vfs.item_size(repo, fake_item))
184 wvstart('augment_item_meta')
185 run_augment_item_meta_tests(repo,
186 '/test/latest/file', 7,
187 '/test/latest/symlink', 'file')
190 # FIXME: this caused StopIteration
191 #_, file_item = vfs.resolve(repo, '/file')[-1]
192 _, file_item = vfs.resolve(repo, '/test/latest/file')[-1]
193 file_copy = vfs.copy_item(file_item)
194 wvpass(file_copy is not file_item)
195 wvpass(file_copy.meta is not file_item.meta)
196 wvpass(isinstance(file_copy, tuple))
197 wvpass(file_item.meta.user)
198 wvpass(file_copy.meta.user)
199 file_copy.meta.user = None
200 wvpass(file_item.meta.user)
204 with no_lingering_errors():
205 with test_tempdir('bup-tvfs-') as tmpdir:
206 resolve = vfs.resolve
207 lresolve = vfs.lresolve
208 bup_dir = tmpdir + '/bup'
209 environ['GIT_DIR'] = bup_dir
210 environ['BUP_DIR'] = bup_dir
211 git.repodir = bup_dir
212 data_path = tmpdir + '/src'
214 save_time_str = strftime('%Y-%m-%d-%H%M%S', localtime(save_time))
216 with open(data_path + '/file', 'w+') as tmpfile:
217 print('canary', file=tmpfile)
218 symlink('file', data_path + '/symlink')
219 ex((bup_path, 'init'))
220 ex((bup_path, 'index', '-v', data_path))
221 ex((bup_path, 'save', '-d', str(save_time), '-tvvn', 'test',
222 '--strip', data_path))
223 ex((bup_path, 'tag', 'test-tag', 'test'))
226 tip_hash = exo(('git', 'show-ref', 'refs/heads/test'))[0]
227 tip_oidx = tip_hash.strip().split()[0]
228 tip_oid = tip_oidx.decode('hex')
229 tip_meta = Metadata()
230 tip_meta.mode = S_IFDIR | 0o755
231 tip_meta.uid = tip_meta.gid = tip_meta.size = 0
232 tip_meta.atime = tip_meta.mtime = tip_meta.ctime = save_time * 10**9
233 test_revlist = vfs.RevList(meta=tip_meta, oid=tip_oid)
234 tip_tree_oidx = exo(('git', 'log', '--pretty=%T', '-n1',
235 tip_oidx))[0].strip()
236 tip_tree_oid = tip_tree_oidx.decode('hex')
237 tip_tree = tree_dict(repo, tip_tree_oid)
239 wvstart('resolve: /')
240 res = resolve(repo, '/')
241 wvpasseq(1, len(res))
242 wvpasseq((('', vfs._root),), res)
243 ignore, root_item = res[0]
244 root_content = frozenset(vfs.contents(repo, root_item))
245 wvpasseq(frozenset([('.', root_item),
247 ('test', test_revlist)]),
250 wvstart('resolve: /.tag')
251 res = resolve(repo, '/.tag')
252 wvpasseq(2, len(res))
253 wvpasseq((('', vfs._root), ('.tag', vfs._tags)),
255 ignore, tag_item = res[1]
256 tag_content = frozenset(vfs.contents(repo, tag_item))
257 wvpasseq(frozenset([('.', tag_item),
258 ('test-tag', test_revlist)]),
261 wvstart('resolve: /test')
262 res = resolve(repo, '/test')
263 wvpasseq(2, len(res))
264 wvpasseq((('', vfs._root), ('test', test_revlist)), res)
265 ignore, test_item = res[1]
266 test_content = frozenset(vfs.contents(repo, test_item))
267 expected_latest_item = vfs.Commit(meta=S_IFDIR | 0o755,
270 wvpasseq(frozenset([('.', test_revlist),
271 (save_time_str, expected_latest_item),
272 ('latest', expected_latest_item)]),
275 wvstart('resolve: /test/latest')
276 res = resolve(repo, '/test/latest')
277 wvpasseq(3, len(res))
278 expected_latest_item_w_meta = vfs.Commit(meta=tip_tree['.'].meta,
281 expected = (('', vfs._root),
282 ('test', test_revlist),
283 ('latest', expected_latest_item_w_meta))
284 wvpasseq(expected, res)
285 ignore, latest_item = res[2]
286 latest_content = frozenset(vfs.contents(repo, latest_item))
287 expected = frozenset((x.name, vfs.Item(oid=x.oid, meta=x.meta))
288 for x in (tip_tree[name]
289 for name in ('.', 'file',
291 wvpasseq(expected, latest_content)
293 wvstart('resolve: /test/latest/foo')
294 res = resolve(repo, '/test/latest/file')
295 wvpasseq(4, len(res))
296 expected_file_item_w_meta = vfs.Item(meta=tip_tree['file'].meta,
297 oid=tip_tree['file'].oid)
298 expected = (('', vfs._root),
299 ('test', test_revlist),
300 ('latest', expected_latest_item_w_meta),
301 ('file', expected_file_item_w_meta))
302 wvpasseq(expected, res)
304 wvstart('resolve: /test/latest/symlink')
305 res = resolve(repo, '/test/latest/symlink')
306 wvpasseq(4, len(res))
307 expected = (('', vfs._root),
308 ('test', test_revlist),
309 ('latest', expected_latest_item_w_meta),
310 ('file', expected_file_item_w_meta))
311 wvpasseq(expected, res)
313 wvstart('lresolve: /test/latest/symlink')
314 res = lresolve(repo, '/test/latest/symlink')
315 wvpasseq(4, len(res))
316 symlink_value = tip_tree['symlink']
317 expected_symlink_item_w_meta = vfs.Item(meta=symlink_value.meta,
318 oid=symlink_value.oid)
319 expected = (('', vfs._root),
320 ('test', test_revlist),
321 ('latest', expected_latest_item_w_meta),
322 ('symlink', expected_symlink_item_w_meta))
323 wvpasseq(expected, res)
325 wvstart('resolve: /test/latest/missing')
326 res = resolve(repo, '/test/latest/missing')
327 wvpasseq(4, len(res))
329 wvpasseq('missing', name)
333 def test_resolve_loop():
334 with no_lingering_errors():
335 with test_tempdir('bup-tvfs-resloop-') as tmpdir:
336 resolve = vfs.resolve
337 lresolve = vfs.lresolve
338 bup_dir = tmpdir + '/bup'
339 environ['GIT_DIR'] = bup_dir
340 environ['BUP_DIR'] = bup_dir
341 git.repodir = bup_dir
343 data_path = tmpdir + '/src'
345 symlink('loop', data_path + '/loop')
346 ex((bup_path, 'init'))
347 ex((bup_path, 'index', '-v', data_path))
348 ex((bup_path, 'save', '-d', '100000', '-tvvn', 'test', '--strip',
350 wvexcept(vfs.Loop, resolve, repo, '/test/latest/loop')
353 def test_contents_with_mismatched_bupm_git_ordering():
354 with no_lingering_errors():
355 with test_tempdir('bup-tvfs-') as tmpdir:
356 bup_dir = tmpdir + '/bup'
357 environ['GIT_DIR'] = bup_dir
358 environ['BUP_DIR'] = bup_dir
359 git.repodir = bup_dir
360 data_path = tmpdir + '/src'
362 os.mkdir(data_path + '/foo')
363 with open(data_path + '/foo.', 'w+') as tmpfile:
364 tmpfile.write(b'canary\n')
365 ex((bup_path, 'init'))
366 ex((bup_path, 'index', '-v', data_path))
367 ex((bup_path, 'save', '-tvvn', 'test', '--strip',
370 tip_sref = exo(('git', 'show-ref', 'refs/heads/test'))[0]
371 tip_oidx = tip_sref.strip().split()[0]
372 tip_tree_oidx = exo(('git', 'log', '--pretty=%T', '-n1',
373 tip_oidx))[0].strip()
374 tip_tree_oid = tip_tree_oidx.decode('hex')
375 tip_tree = tree_dict(repo, tip_tree_oid)
377 name, item = vfs.resolve(repo, '/test/latest')[2]
378 wvpasseq('latest', name)
379 expected = frozenset((x.name, vfs.Item(oid=x.oid, meta=x.meta))
380 for x in (tip_tree[name]
381 for name in ('.', 'foo', 'foo.')))
382 contents = tuple(vfs.contents(repo, item))
383 wvpasseq(expected, frozenset(contents))
384 # Spot check, in case tree_dict shares too much code with the vfs
385 name, item = next(((n, i) for n, i in contents if n == 'foo'))
386 wvpass(S_ISDIR(item.meta))
387 name, item = next(((n, i) for n, i in contents if n == 'foo.'))
388 wvpass(S_ISREG(item.meta.mode))
391 def test_duplicate_save_dates():
392 with no_lingering_errors():
393 with test_tempdir('bup-tvfs-') as tmpdir:
394 bup_dir = tmpdir + '/bup'
395 environ['GIT_DIR'] = bup_dir
396 environ['BUP_DIR'] = bup_dir
397 environ['TZ'] = 'UTC'
398 git.repodir = bup_dir
399 data_path = tmpdir + '/src'
401 with open(data_path + '/file', 'w+') as tmpfile:
402 tmpfile.write(b'canary\n')
403 ex((bup_path, 'init'))
404 ex((bup_path, 'index', '-v', data_path))
406 ex((bup_path, 'save', '-d', '100000', '-n', 'test', data_path))
408 res = vfs.resolve(repo, '/test')
409 wvpasseq(2, len(res))
410 name, revlist = res[-1]
411 wvpasseq('test', name)
413 '1970-01-02-034640-10',
414 '1970-01-02-034640-09',
415 '1970-01-02-034640-08',
416 '1970-01-02-034640-07',
417 '1970-01-02-034640-06',
418 '1970-01-02-034640-05',
419 '1970-01-02-034640-04',
420 '1970-01-02-034640-03',
421 '1970-01-02-034640-02',
422 '1970-01-02-034640-01',
423 '1970-01-02-034640-00',
425 tuple(x[0] for x in vfs.contents(repo, revlist)))
427 # FIXME: add tests for the want_meta=False cases.