103
117
def build_a_rev(self):
104
118
builder = BranchBuilder(self.get_transport().clone('foo'))
105
rev_id1 = builder.build_snapshot('A-id', None,
106
[('add', ('', 'a-root-id', 'directory', None)),
107
('add', ('a', 'a-id', 'file', 'contents'))])
108
self.assertEqual('A-id', rev_id1)
119
rev_id1 = builder.build_snapshot(None,
120
[('add', ('', b'a-root-id', 'directory', None)),
121
('add', ('a', b'a-id', 'file', b'contents'))],
123
self.assertEqual(b'A-id', rev_id1)
111
126
def test_add_one_file(self):
112
127
builder = self.build_a_rev()
113
128
branch = builder.get_branch()
114
self.assertEqual((1, 'A-id'), branch.last_revision_info())
115
rev_tree = branch.repository.revision_tree('A-id')
129
self.assertEqual((1, b'A-id'), branch.last_revision_info())
130
rev_tree = branch.repository.revision_tree(b'A-id')
116
131
rev_tree.lock_read()
117
132
self.addCleanup(rev_tree.unlock)
118
self.assertTreeShape([(u'', 'a-root-id', 'directory'),
119
(u'a', 'a-id', 'file')], rev_tree)
120
self.assertEqual('contents', rev_tree.get_file_text('a-id'))
133
self.assertTreeShape([(u'', b'a-root-id', 'directory'),
134
(u'a', b'a-id', 'file')], rev_tree)
135
self.assertEqual(b'contents', rev_tree.get_file_text('a'))
122
137
def test_add_second_file(self):
123
138
builder = self.build_a_rev()
124
rev_id2 = builder.build_snapshot('B-id', None,
125
[('add', ('b', 'b-id', 'file', 'content_b'))])
126
self.assertEqual('B-id', rev_id2)
139
rev_id2 = builder.build_snapshot(None,
140
[('add', ('b', b'b-id', 'file', b'content_b'))],
142
self.assertEqual(b'B-id', rev_id2)
127
143
branch = builder.get_branch()
128
144
self.assertEqual((2, rev_id2), branch.last_revision_info())
129
145
rev_tree = branch.repository.revision_tree(rev_id2)
130
146
rev_tree.lock_read()
131
147
self.addCleanup(rev_tree.unlock)
132
self.assertTreeShape([(u'', 'a-root-id', 'directory'),
133
(u'a', 'a-id', 'file'),
134
(u'b', 'b-id', 'file')], rev_tree)
135
self.assertEqual('content_b', rev_tree.get_file_text('b-id'))
148
self.assertTreeShape([(u'', b'a-root-id', 'directory'),
149
(u'a', b'a-id', 'file'),
150
(u'b', b'b-id', 'file')], rev_tree)
151
self.assertEqual(b'content_b', rev_tree.get_file_text('b'))
137
153
def test_add_empty_dir(self):
138
154
builder = self.build_a_rev()
139
rev_id2 = builder.build_snapshot('B-id', None,
140
[('add', ('b', 'b-id', 'directory', None))])
141
rev_tree = builder.get_branch().repository.revision_tree('B-id')
142
self.assertTreeShape([(u'', 'a-root-id', 'directory'),
143
(u'a', 'a-id', 'file'),
144
(u'b', 'b-id', 'directory'),
155
rev_id2 = builder.build_snapshot(None,
156
[('add', ('b', b'b-id', 'directory', None))],
158
rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
159
self.assertTreeShape([(u'', b'a-root-id', 'directory'),
160
(u'a', b'a-id', 'file'),
161
(u'b', b'b-id', 'directory'),
147
164
def test_commit_timestamp(self):
148
165
builder = self.make_branch_builder('foo')
149
rev_id = builder.build_snapshot(None, None,
166
rev_id = builder.build_snapshot(None,
150
167
[('add', (u'', None, 'directory', None))],
151
168
timestamp=1234567890)
152
169
rev = builder.get_branch().repository.get_revision(rev_id)
183
200
def test_modify_file(self):
184
201
builder = self.build_a_rev()
185
rev_id2 = builder.build_snapshot('B-id', None,
186
[('modify', ('a-id', 'new\ncontent\n'))])
187
self.assertEqual('B-id', rev_id2)
202
rev_id2 = builder.build_snapshot(None,
203
[('modify', ('a', b'new\ncontent\n'))],
205
self.assertEqual(b'B-id', rev_id2)
188
206
branch = builder.get_branch()
189
207
rev_tree = branch.repository.revision_tree(rev_id2)
190
208
rev_tree.lock_read()
191
209
self.addCleanup(rev_tree.unlock)
192
self.assertEqual('new\ncontent\n', rev_tree.get_file_text('a-id'))
210
self.assertEqual(b'new\ncontent\n',
211
rev_tree.get_file_text(rev_tree.id2path(b'a-id')))
194
213
def test_delete_file(self):
195
214
builder = self.build_a_rev()
196
rev_id2 = builder.build_snapshot('B-id', None,
197
[('unversion', 'a-id')])
198
self.assertEqual('B-id', rev_id2)
215
rev_id2 = builder.build_snapshot(None,
216
[('unversion', 'a')], revision_id=b'B-id')
217
self.assertEqual(b'B-id', rev_id2)
199
218
branch = builder.get_branch()
200
219
rev_tree = branch.repository.revision_tree(rev_id2)
201
220
rev_tree.lock_read()
202
221
self.addCleanup(rev_tree.unlock)
203
self.assertTreeShape([(u'', 'a-root-id', 'directory')], rev_tree)
222
self.assertTreeShape([(u'', b'a-root-id', 'directory')], rev_tree)
205
224
def test_delete_directory(self):
206
225
builder = self.build_a_rev()
207
rev_id2 = builder.build_snapshot('B-id', None,
208
[('add', ('b', 'b-id', 'directory', None)),
209
('add', ('b/c', 'c-id', 'file', 'foo\n')),
210
('add', ('b/d', 'd-id', 'directory', None)),
211
('add', ('b/d/e', 'e-id', 'file', 'eff\n')),
213
rev_tree = builder.get_branch().repository.revision_tree('B-id')
214
self.assertTreeShape([(u'', 'a-root-id', 'directory'),
215
(u'a', 'a-id', 'file'),
216
(u'b', 'b-id', 'directory'),
217
(u'b/c', 'c-id', 'file'),
218
(u'b/d', 'd-id', 'directory'),
219
(u'b/d/e', 'e-id', 'file')], rev_tree)
226
rev_id2 = builder.build_snapshot(None,
227
[('add', ('b', b'b-id', 'directory', None)),
228
('add', ('b/c', b'c-id', 'file', b'foo\n')),
229
('add', ('b/d', b'd-id', 'directory', None)),
230
('add', ('b/d/e', b'e-id', 'file', b'eff\n')),
231
], revision_id=b'B-id')
232
rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
233
self.assertTreeShape([(u'', b'a-root-id', 'directory'),
234
(u'a', b'a-id', 'file'),
235
(u'b', b'b-id', 'directory'),
236
(u'b/c', b'c-id', 'file'),
237
(u'b/d', b'd-id', 'directory'),
238
(u'b/d/e', b'e-id', 'file')], rev_tree)
220
239
# Removing a directory removes all child dirs
221
builder.build_snapshot('C-id', None, [('unversion', 'b-id')])
222
rev_tree = builder.get_branch().repository.revision_tree('C-id')
223
self.assertTreeShape([(u'', 'a-root-id', 'directory'),
224
(u'a', 'a-id', 'file'),
240
builder.build_snapshot(
241
None, [('unversion', 'b')],
243
rev_tree = builder.get_branch().repository.revision_tree(b'C-id')
244
self.assertTreeShape([(u'', b'a-root-id', 'directory'),
245
(u'a', b'a-id', 'file'),
227
248
def test_unknown_action(self):
228
249
builder = self.build_a_rev()
229
250
e = self.assertRaises(ValueError,
230
builder.build_snapshot, 'B-id', None, [('weirdo', ('foo',))])
251
builder.build_snapshot, None, [('weirdo', ('foo',))],
231
253
self.assertEqual('Unknown build action: "weirdo"', str(e))
233
255
def test_rename(self):
234
256
builder = self.build_a_rev()
235
builder.build_snapshot('B-id', None,
236
[('rename', ('a', 'b'))])
237
rev_tree = builder.get_branch().repository.revision_tree('B-id')
238
self.assertTreeShape([(u'', 'a-root-id', 'directory'),
239
(u'b', 'a-id', 'file')], rev_tree)
257
builder.build_snapshot(None,
258
[('rename', ('a', 'b'))], revision_id=b'B-id')
259
rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
260
self.assertTreeShape([(u'', b'a-root-id', 'directory'),
261
(u'b', b'a-id', 'file')], rev_tree)
241
263
def test_rename_into_subdir(self):
242
264
builder = self.build_a_rev()
243
builder.build_snapshot('B-id', None,
244
[('add', ('dir', 'dir-id', 'directory', None)),
245
('rename', ('a', 'dir/a'))])
246
rev_tree = builder.get_branch().repository.revision_tree('B-id')
247
self.assertTreeShape([(u'', 'a-root-id', 'directory'),
248
(u'dir', 'dir-id', 'directory'),
249
(u'dir/a', 'a-id', 'file')], rev_tree)
265
builder.build_snapshot(None,
266
[('add', ('dir', b'dir-id', 'directory', None)),
267
('rename', ('a', 'dir/a'))], revision_id=b'B-id')
268
rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
269
self.assertTreeShape([(u'', b'a-root-id', 'directory'),
270
(u'dir', b'dir-id', 'directory'),
271
(u'dir/a', b'a-id', 'file')], rev_tree)
273
def test_rename_out_of_unversioned_subdir(self):
274
builder = self.build_a_rev()
275
builder.build_snapshot(None,
276
[('add', ('dir', b'dir-id', 'directory', None)),
277
('rename', ('a', 'dir/a'))],
279
builder.build_snapshot(None,
280
[('rename', ('dir/a', 'a')),
281
('unversion', 'dir')], revision_id=b'C-id')
282
rev_tree = builder.get_branch().repository.revision_tree(b'C-id')
283
self.assertTreeShape([(u'', b'a-root-id', 'directory'),
284
(u'a', b'a-id', 'file')], rev_tree)
251
286
def test_set_parent(self):
252
287
builder = self.build_a_rev()
253
288
builder.start_series()
254
289
self.addCleanup(builder.finish_series)
255
builder.build_snapshot('B-id', ['A-id'],
256
[('modify', ('a-id', 'new\ncontent\n'))])
257
builder.build_snapshot('C-id', ['A-id'],
258
[('add', ('c', 'c-id', 'file', 'alt\ncontent\n'))])
290
builder.build_snapshot([b'A-id'],
291
[('modify', ('a', b'new\ncontent\n'))],
293
builder.build_snapshot([b'A-id'],
294
[('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
259
296
# We should now have a graph:
263
300
# And not A => B => C
264
301
repo = builder.get_branch().repository
265
self.assertEqual({'B-id': ('A-id',), 'C-id': ('A-id',)},
266
repo.get_parent_map(['B-id', 'C-id']))
267
b_tree = repo.revision_tree('B-id')
268
self.assertTreeShape([(u'', 'a-root-id', 'directory'),
269
(u'a', 'a-id', 'file'),
302
self.assertEqual({b'B-id': (b'A-id',), b'C-id': (b'A-id',)},
303
repo.get_parent_map([b'B-id', b'C-id']))
304
b_tree = repo.revision_tree(b'B-id')
305
self.assertTreeShape([(u'', b'a-root-id', 'directory'),
306
(u'a', b'a-id', 'file'),
271
self.assertEqual('new\ncontent\n', b_tree.get_file_text('a-id'))
308
self.assertEqual(b'new\ncontent\n', b_tree.get_file_text('a'))
273
310
# We should still be using the content from A in C, not from B
274
c_tree = repo.revision_tree('C-id')
275
self.assertTreeShape([(u'', 'a-root-id', 'directory'),
276
(u'a', 'a-id', 'file'),
277
(u'c', 'c-id', 'file'),
311
c_tree = repo.revision_tree(b'C-id')
312
self.assertTreeShape([(u'', b'a-root-id', 'directory'),
313
(u'a', b'a-id', 'file'),
314
(u'c', b'c-id', 'file'),
279
self.assertEqual('contents', c_tree.get_file_text('a-id'))
280
self.assertEqual('alt\ncontent\n', c_tree.get_file_text('c-id'))
316
self.assertEqual(b'contents', c_tree.get_file_text('a'))
317
self.assertEqual(b'alt\ncontent\n', c_tree.get_file_text('c'))
282
319
def test_set_merge_parent(self):
283
320
builder = self.build_a_rev()
284
321
builder.start_series()
285
322
self.addCleanup(builder.finish_series)
286
builder.build_snapshot('B-id', ['A-id'],
287
[('add', ('b', 'b-id', 'file', 'b\ncontent\n'))])
288
builder.build_snapshot('C-id', ['A-id'],
289
[('add', ('c', 'c-id', 'file', 'alt\ncontent\n'))])
290
builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
323
builder.build_snapshot([b'A-id'],
324
[('add', ('b', b'b-id', 'file', b'b\ncontent\n'))],
326
builder.build_snapshot([b'A-id'],
327
[('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
329
builder.build_snapshot([b'B-id', b'C-id'], [], revision_id=b'D-id')
291
330
repo = builder.get_branch().repository
292
self.assertEqual({'B-id': ('A-id',), 'C-id': ('A-id',),
293
'D-id': ('B-id', 'C-id')},
294
repo.get_parent_map(['B-id', 'C-id', 'D-id']))
295
d_tree = repo.revision_tree('D-id')
331
self.assertEqual({b'B-id': (b'A-id',), b'C-id': (b'A-id',),
332
b'D-id': (b'B-id', b'C-id')},
333
repo.get_parent_map([b'B-id', b'C-id', b'D-id']))
334
d_tree = repo.revision_tree(b'D-id')
296
335
# Note: by default a merge node does *not* pull in the changes from the
297
336
# merged tree, you have to supply it yourself.
298
self.assertTreeShape([(u'', 'a-root-id', 'directory'),
299
(u'a', 'a-id', 'file'),
300
(u'b', 'b-id', 'file'),
337
self.assertTreeShape([(u'', b'a-root-id', 'directory'),
338
(u'a', b'a-id', 'file'),
339
(u'b', b'b-id', 'file'),
303
342
def test_set_merge_parent_and_contents(self):
304
343
builder = self.build_a_rev()
305
344
builder.start_series()
306
345
self.addCleanup(builder.finish_series)
307
builder.build_snapshot('B-id', ['A-id'],
308
[('add', ('b', 'b-id', 'file', 'b\ncontent\n'))])
309
builder.build_snapshot('C-id', ['A-id'],
310
[('add', ('c', 'c-id', 'file', 'alt\ncontent\n'))])
311
builder.build_snapshot('D-id', ['B-id', 'C-id'],
312
[('add', ('c', 'c-id', 'file', 'alt\ncontent\n'))])
346
builder.build_snapshot([b'A-id'],
347
[('add', ('b', b'b-id', 'file', b'b\ncontent\n'))],
349
builder.build_snapshot([b'A-id'],
350
[('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
352
builder.build_snapshot([b'B-id', b'C-id'],
353
[('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
313
355
repo = builder.get_branch().repository
314
self.assertEqual({'B-id': ('A-id',), 'C-id': ('A-id',),
315
'D-id': ('B-id', 'C-id')},
316
repo.get_parent_map(['B-id', 'C-id', 'D-id']))
317
d_tree = repo.revision_tree('D-id')
318
self.assertTreeShape([(u'', 'a-root-id', 'directory'),
319
(u'a', 'a-id', 'file'),
320
(u'b', 'b-id', 'file'),
321
(u'c', 'c-id', 'file'),
356
self.assertEqual({b'B-id': (b'A-id',), b'C-id': (b'A-id',),
357
b'D-id': (b'B-id', b'C-id')},
358
repo.get_parent_map([b'B-id', b'C-id', b'D-id']))
359
d_tree = repo.revision_tree(b'D-id')
360
self.assertTreeShape([(u'', b'a-root-id', 'directory'),
361
(u'a', b'a-id', 'file'),
362
(u'b', b'b-id', 'file'),
363
(u'c', b'c-id', 'file'),
323
365
# Because we copied the exact text into *this* tree, the 'c' file
324
366
# should look like it was not modified in the merge
325
self.assertEqual('C-id', d_tree.inventory['c-id'].revision)
367
self.assertEqual(b'C-id', d_tree.get_file_revision('c'))
369
def test_set_parent_to_null(self):
370
builder = self.build_a_rev()
371
builder.start_series()
372
self.addCleanup(builder.finish_series)
373
builder.build_snapshot([],
374
[('add', ('', None, 'directory', None))],
376
# We should now have a graph:
379
repo = builder.get_branch().repository
380
self.assertEqual({b'A-id': (_mod_revision.NULL_REVISION,),
381
b'B-id': (_mod_revision.NULL_REVISION,),},
382
repo.get_parent_map([b'A-id', b'B-id']))
327
384
def test_start_finish_series(self):
328
385
builder = BranchBuilder(self.get_transport().clone('foo'))
340
397
builder = BranchBuilder(self.get_transport().clone('foo'))
341
398
builder.start_series()
343
builder.build_snapshot('tip', ['ghost'],
344
[('add', ('', 'ROOT_ID', 'directory', ''))],
345
allow_leftmost_as_ghost=True)
400
builder.build_snapshot([b'ghost'],
401
[('add', ('', b'ROOT_ID', 'directory', ''))],
402
allow_leftmost_as_ghost=True, revision_id=b'tip')
347
404
builder.finish_series()
348
405
b = builder.get_branch()
350
407
self.addCleanup(b.unlock)
351
self.assertEqual(('ghost',),
352
b.repository.get_graph().get_parent_map(['tip'])['tip'])
408
self.assertEqual((b'ghost',),
409
b.repository.get_graph().get_parent_map([b'tip'])[b'tip'])
411
def test_unversion_root_add_new_root(self):
412
builder = BranchBuilder(self.get_transport().clone('foo'))
413
builder.start_series()
414
builder.build_snapshot(None,
415
[('add', ('', b'TREE_ROOT', 'directory', ''))],
416
revision_id=b'rev-1')
417
builder.build_snapshot(None,
419
('add', ('', b'my-root', 'directory', ''))],
420
revision_id=b'rev-2')
421
builder.finish_series()
422
rev_tree = builder.get_branch().repository.revision_tree(b'rev-2')
423
self.assertTreeShape([(u'', b'my-root', 'directory')], rev_tree)
425
def test_empty_flush(self):
426
"""A flush with no actions before it is a no-op."""
427
builder = BranchBuilder(self.get_transport().clone('foo'))
428
builder.start_series()
429
builder.build_snapshot(None,
430
[('add', ('', b'TREE_ROOT', 'directory', ''))],
431
revision_id=b'rev-1')
432
builder.build_snapshot(None, [('flush', None)], revision_id=b'rev-2')
433
builder.finish_series()
434
rev_tree = builder.get_branch().repository.revision_tree(b'rev-2')
435
self.assertTreeShape([(u'', b'TREE_ROOT', 'directory')], rev_tree)
437
def test_kind_change(self):
438
"""It's possible to change the kind of an entry in a single snapshot
439
with a bit of help from the 'flush' action.
441
builder = BranchBuilder(self.get_transport().clone('foo'))
442
builder.start_series()
443
builder.build_snapshot(None,
444
[('add', (u'', b'a-root-id', 'directory', None)),
445
('add', (u'a', b'a-id', 'file', b'content\n'))],
447
builder.build_snapshot(None,
450
('add', (u'a', b'a-id', 'directory', None))],
452
builder.finish_series()
453
rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
454
self.assertTreeShape(
455
[(u'', b'a-root-id', 'directory'), (u'a', b'a-id', 'directory')],
458
def test_pivot_root(self):
459
"""It's possible (albeit awkward) to move an existing dir to the root
460
in a single snapshot by using unversion then flush then add.
462
builder = BranchBuilder(self.get_transport().clone('foo'))
463
builder.start_series()
464
builder.build_snapshot(None,
465
[('add', (u'', b'orig-root', 'directory', None)),
466
('add', (u'dir', b'dir-id', 'directory', None))],
468
builder.build_snapshot(None,
469
[('unversion', ''), # implicitly unversions all children
471
('add', (u'', b'dir-id', 'directory', None))],
473
builder.finish_series()
474
rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
475
self.assertTreeShape([(u'', b'dir-id', 'directory')], rev_tree)