← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_branchbuilder.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2011-08-17 18:13:57 UTC
  • mfrom: (5268.7.29 transport-segments)
  • Revision ID: pqm@pqm.ubuntu.com-20110817181357-y5q5eth1hk8bl3om
(jelmer) Allow specifying the colocated branch to use in the branch URL,
 and retrieving the branch name using ControlDir._get_selected_branch.
 (Jelmer Vernooij)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_branchbuilder.py
1
 
# Copyright (C) 2007 Canonical Ltd
 
1
# Copyright (C) 2007-2011 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
"""Tests for the BranchBuilder class."""
18
18
 
19
19
from bzrlib import (
20
20
    branch as _mod_branch,
21
 
    errors,
22
21
    revision as _mod_revision,
23
22
    tests,
24
23
    )
26
25
 
27
26
 
28
27
class TestBranchBuilder(tests.TestCaseWithMemoryTransport):
29
 
    
 
28
 
30
29
    def test_create(self):
31
30
        """Test the constructor api."""
32
31
        builder = BranchBuilder(self.get_transport().clone('foo'))
59
58
            'commit 1',
60
59
            branch.repository.get_revision(branch.last_revision()).message)
61
60
 
 
61
    def test_build_commit_timestamp(self):
 
62
        """You can set a date when committing."""
 
63
        builder = self.make_branch_builder('foo')
 
64
        rev_id = builder.build_commit(timestamp=1236043340)
 
65
        branch = builder.get_branch()
 
66
        self.assertEqual((1, rev_id), branch.last_revision_info())
 
67
        rev = branch.repository.get_revision(branch.last_revision())
 
68
        self.assertEqual(
 
69
            'commit 1',
 
70
            rev.message)
 
71
        self.assertEqual(
 
72
            1236043340,
 
73
            int(rev.timestamp))
 
74
 
62
75
    def test_build_two_commits(self):
63
76
        """The second commit has the right parents and message."""
64
77
        builder = BranchBuilder(self.get_transport().clone('foo'))
130
143
                              (u'b', 'b-id', 'directory'),
131
144
                             ], rev_tree)
132
145
 
 
146
    def test_commit_timestamp(self):
 
147
        builder = self.make_branch_builder('foo')
 
148
        rev_id = builder.build_snapshot(None, None,
 
149
            [('add', (u'', None, 'directory', None))],
 
150
            timestamp=1234567890)
 
151
        rev = builder.get_branch().repository.get_revision(rev_id)
 
152
        self.assertEqual(
 
153
            1234567890,
 
154
            int(rev.timestamp))
 
155
 
133
156
    def test_commit_message_default(self):
134
157
        builder = BranchBuilder(self.get_transport().clone('foo'))
135
158
        rev_id = builder.build_snapshot(None, None,
147
170
        rev = branch.repository.get_revision(rev_id)
148
171
        self.assertEqual(u'Foo', rev.message)
149
172
 
 
173
    def test_commit_message_callback(self):
 
174
        builder = BranchBuilder(self.get_transport().clone('foo'))
 
175
        rev_id = builder.build_snapshot(None, None,
 
176
            [('add', (u'', None, 'directory', None))],
 
177
            message_callback=lambda x:u'Foo')
 
178
        branch = builder.get_branch()
 
179
        rev = branch.repository.get_revision(rev_id)
 
180
        self.assertEqual(u'Foo', rev.message)
 
181
 
150
182
    def test_modify_file(self):
151
183
        builder = self.build_a_rev()
152
184
        rev_id2 = builder.build_snapshot('B-id', None,
215
247
                              (u'dir', 'dir-id', 'directory'),
216
248
                              (u'dir/a', 'a-id', 'file')], rev_tree)
217
249
 
 
250
    def test_rename_out_of_unversioned_subdir(self):
 
251
        builder = self.build_a_rev()
 
252
        builder.build_snapshot('B-id', None,
 
253
            [('add', ('dir', 'dir-id', 'directory', None)),
 
254
             ('rename', ('a', 'dir/a'))])
 
255
        builder.build_snapshot('C-id', None,
 
256
            [('rename', ('dir/a', 'a')),
 
257
             ('unversion', 'dir-id')])
 
258
        rev_tree = builder.get_branch().repository.revision_tree('C-id')
 
259
        self.assertTreeShape([(u'', 'a-root-id', 'directory'),
 
260
                              (u'a', 'a-id', 'file')], rev_tree)
 
261
 
218
262
    def test_set_parent(self):
219
263
        builder = self.build_a_rev()
220
264
        builder.start_series()
221
265
        self.addCleanup(builder.finish_series)
222
266
        builder.build_snapshot('B-id', ['A-id'],
223
267
            [('modify', ('a-id', 'new\ncontent\n'))])
224
 
        builder.build_snapshot('C-id', ['A-id'], 
 
268
        builder.build_snapshot('C-id', ['A-id'],
225
269
            [('add', ('c', 'c-id', 'file', 'alt\ncontent\n'))])
226
270
        # We should now have a graph:
227
271
        #   A
289
333
                             ], d_tree)
290
334
        # Because we copied the exact text into *this* tree, the 'c' file
291
335
        # should look like it was not modified in the merge
292
 
        self.assertEqual('C-id', d_tree.inventory['c-id'].revision)
293
 
 
 
336
        self.assertEqual('C-id', d_tree.get_file_revision('c-id'))
 
337
 
 
338
    def test_set_parent_to_null(self):
 
339
        builder = self.build_a_rev()
 
340
        builder.start_series()
 
341
        self.addCleanup(builder.finish_series)
 
342
        builder.build_snapshot('B-id', [],
 
343
            [('add', ('', None, 'directory', None))])
 
344
        # We should now have a graph:
 
345
        #   A B
 
346
        # And not A => B
 
347
        repo = builder.get_branch().repository
 
348
        self.assertEqual({'A-id': (_mod_revision.NULL_REVISION,),
 
349
                          'B-id': (_mod_revision.NULL_REVISION,),},
 
350
                         repo.get_parent_map(['A-id', 'B-id']))
 
351
 
 
352
    
294
353
    def test_start_finish_series(self):
295
354
        builder = BranchBuilder(self.get_transport().clone('foo'))
296
355
        builder.start_series()
302
361
            builder.finish_series()
303
362
        self.assertIs(None, builder._tree)
304
363
        self.assertFalse(builder._branch.is_locked())
 
364
 
 
365
    def test_ghost_mainline_history(self):
 
366
        builder = BranchBuilder(self.get_transport().clone('foo'))
 
367
        builder.start_series()
 
368
        try:
 
369
            builder.build_snapshot('tip', ['ghost'],
 
370
                [('add', ('', 'ROOT_ID', 'directory', ''))],
 
371
                allow_leftmost_as_ghost=True)
 
372
        finally:
 
373
            builder.finish_series()
 
374
        b = builder.get_branch()
 
375
        b.lock_read()
 
376
        self.addCleanup(b.unlock)
 
377
        self.assertEqual(('ghost',),
 
378
            b.repository.get_graph().get_parent_map(['tip'])['tip'])
 
379
 
 
380
    def test_unversion_root_add_new_root(self):
 
381
        builder = BranchBuilder(self.get_transport().clone('foo'))
 
382
        builder.start_series()
 
383
        builder.build_snapshot('rev-1', None,
 
384
            [('add', ('', 'TREE_ROOT', 'directory', ''))])
 
385
        builder.build_snapshot('rev-2', None,
 
386
            [('unversion', 'TREE_ROOT'),
 
387
             ('add', ('', 'my-root', 'directory', ''))])
 
388
        builder.finish_series()
 
389
        rev_tree = builder.get_branch().repository.revision_tree('rev-2')
 
390
        self.assertTreeShape([(u'', 'my-root', 'directory')], rev_tree)
 
391
 
 
392
    def test_empty_flush(self):
 
393
        """A flush with no actions before it is a no-op."""
 
394
        builder = BranchBuilder(self.get_transport().clone('foo'))
 
395
        builder.start_series()
 
396
        builder.build_snapshot('rev-1', None,
 
397
            [('add', ('', 'TREE_ROOT', 'directory', ''))])
 
398
        builder.build_snapshot('rev-2', None, [('flush', None)])
 
399
        builder.finish_series()
 
400
        rev_tree = builder.get_branch().repository.revision_tree('rev-2')
 
401
        self.assertTreeShape([(u'', 'TREE_ROOT', 'directory')], rev_tree)
 
402
 
 
403
    def test_kind_change(self):
 
404
        """It's possible to change the kind of an entry in a single snapshot
 
405
        with a bit of help from the 'flush' action.
 
406
        """
 
407
        builder = BranchBuilder(self.get_transport().clone('foo'))
 
408
        builder.start_series()
 
409
        builder.build_snapshot('A-id', None,
 
410
            [('add', (u'', 'a-root-id', 'directory', None)),
 
411
             ('add', (u'a', 'a-id', 'file', 'content\n'))])
 
412
        builder.build_snapshot('B-id', None,
 
413
            [('unversion', 'a-id'),
 
414
             ('flush', None),
 
415
             ('add', (u'a', 'a-id', 'directory', None))])
 
416
        builder.finish_series()
 
417
        rev_tree = builder.get_branch().repository.revision_tree('B-id')
 
418
        self.assertTreeShape(
 
419
            [(u'', 'a-root-id', 'directory'), (u'a', 'a-id', 'directory')],
 
420
            rev_tree)
 
421
 
 
422
    def test_pivot_root(self):
 
423
        """It's possible (albeit awkward) to move an existing dir to the root
 
424
        in a single snapshot by using unversion then flush then add.
 
425
        """
 
426
        builder = BranchBuilder(self.get_transport().clone('foo'))
 
427
        builder.start_series()
 
428
        builder.build_snapshot('A-id', None,
 
429
            [('add', (u'', 'orig-root', 'directory', None)),
 
430
             ('add', (u'dir', 'dir-id', 'directory', None))])
 
431
        builder.build_snapshot('B-id', None,
 
432
            [('unversion', 'orig-root'),  # implicitly unversions all children
 
433
             ('flush', None),
 
434
             ('add', (u'', 'dir-id', 'directory', None))])
 
435
        builder.finish_series()
 
436
        rev_tree = builder.get_branch().repository.revision_tree('B-id')
 
437
        self.assertTreeShape([(u'', 'dir-id', 'directory')], rev_tree)
 
438