← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/per_workingtree/test_commit.py

  • Committer: Andrew Bennetts
  • Date: 2010-01-12 03:53:21 UTC
  • mfrom: (4948 +trunk)
  • mto: This revision was merged to the branch mainline in revision 4964.
  • Revision ID: andrew.bennetts@canonical.com-20100112035321-hofpz5p10224ryj3
Merge lp:bzr, resolving conflicts.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/per_workingtree/test_commit.py
13
13
#
14
14
# You should have received a copy of the GNU General Public License
15
15
# along with this program; if not, write to the Free Software
16
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
17
 
18
18
from cStringIO import StringIO
19
19
import os
30
30
    uncommit,
31
31
    workingtree,
32
32
    )
33
 
from bzrlib.errors import (NotBranchError, NotVersionedError, 
 
33
from bzrlib.errors import (NotBranchError, NotVersionedError,
34
34
                           UnsupportedOperation)
35
35
from bzrlib.osutils import pathjoin, getcwd
36
36
from bzrlib.tests import TestCase
37
 
from bzrlib.tests.workingtree_implementations import TestCaseWithWorkingTree
 
37
from bzrlib.tests.per_workingtree import TestCaseWithWorkingTree
38
38
from bzrlib.trace import mutter
39
39
from bzrlib.workingtree import (TreeEntry, TreeDirectory, TreeFile, TreeLink,
40
40
                                WorkingTree)
49
49
        self.depth = 0
50
50
 
51
51
    def clear(self):
52
 
        """See progress.ProgressBar.clear()."""
 
52
        """See progress.ProgressTask.clear()."""
53
53
 
54
54
    def clear_term(self):
55
 
        """See progress.ProgressBar.clear_term()."""
 
55
        """See progress.ProgressTask.clear_term()."""
56
56
 
57
57
    def finished(self):
58
 
        """See progress.ProgressBar.finished()."""
 
58
        """See progress.ProgressTask.finished()."""
59
59
        self.depth -= 1
60
60
 
61
61
    def note(self, fmt_string, *args, **kwargs):
62
 
        """See progress.ProgressBar.note()."""
 
62
        """See progress.ProgressTask.note()."""
63
63
 
64
64
    def progress_bar(self):
65
65
        return self
66
 
    
 
66
 
67
67
    def nested_progress_bar(self):
68
68
        self.depth += 1
69
69
        return self
70
70
 
71
71
    def update(self, message, count=None, total=None):
72
 
        """See progress.ProgressBar.update()."""
 
72
        """See progress.ProgressTask.update()."""
73
73
        if self.depth == 1:
74
74
            self._calls.append(("update", count, total, message))
75
75
 
255
255
        self.assertNotEqual(None, committed_id)
256
256
 
257
257
    def test_commit_local_unbound(self):
258
 
        # using the library api to do a local commit on unbound branches is 
 
258
        # using the library api to do a local commit on unbound branches is
259
259
        # also an error
260
260
        tree = self.make_branch_and_tree('tree')
261
261
        self.assertRaises(errors.LocalRequiresBoundBranch,
280
280
        wt2.merge_from_branch(wt.branch)
281
281
        wt2.commit('merged kind change')
282
282
 
 
283
    def test_commit_aborted_does_not_apply_automatic_changes_bug_282402(self):
 
284
        wt = self.make_branch_and_tree('.')
 
285
        wt.add(['a'], ['a-id'], ['file'])
 
286
        def fail_message(obj):
 
287
            raise errors.BzrCommandError("empty commit message")
 
288
        self.assertRaises(errors.BzrCommandError, wt.commit,
 
289
            message_callback=fail_message)
 
290
        self.assertEqual('a', wt.id2path('a-id'))
 
291
 
283
292
    def test_local_commit_ignores_master(self):
284
293
        # a --local commit does not require access to the master branch
285
294
        # at all, or even for it to exist.
299
308
                          bzrdir.BzrDir.open,
300
309
                          'master')
301
310
        tree.commit('foo', rev_id='foo', local=True)
302
 
 
 
311
 
303
312
    def test_local_commit_does_not_push_to_master(self):
304
313
        # a --local commit does not require access to the master branch
305
314
        # at all, or even for it to exist.
444
453
            basis.get_reference_revision(basis.path2id('subtree')))
445
454
        # the outer tree must have have changed too.
446
455
        self.assertNotEqual(None, rev_id)
447
 
        
 
456
 
448
457
    def test_nested_commit_second_commit_detects_changes(self):
449
458
        """Commit with a nested tree picks up the correct child revid."""
450
459
        tree = self.make_branch_and_tree('.')
456
465
        self.build_tree(['subtree/file'])
457
466
        subtree.add(['file'], ['file-id'])
458
467
        rev_id = tree.commit('added reference', allow_pointless=False)
 
468
        tree.get_reference_revision(tree.path2id('subtree'))
459
469
        child_revid = subtree.last_revision()
460
470
        # now change the child tree
461
471
        self.build_tree_contents([('subtree/file', 'new-content')])
494
504
 
495
505
 
496
506
class TestCommitProgress(TestCaseWithWorkingTree):
497
 
    
 
507
 
498
508
    def restoreDefaults(self):
499
509
        ui.ui_factory = self.old_ui_factory
500
510
 
501
511
    def test_commit_progress_steps(self):
502
 
        # during commit we one progress update for every entry in the 
 
512
        # during commit we one progress update for every entry in the
503
513
        # inventory, and then one for the inventory, and one for the
504
514
        # inventory, and one for the revision insertions.
505
 
        # first we need a test commit to do. Lets setup a branch with 
 
515
        # first we need a test commit to do. Lets setup a branch with
506
516
        # 3 files, and alter one in a selected-file commit. This exercises
507
 
        # a number of cases quickly. We should also test things like 
 
517
        # a number of cases quickly. We should also test things like
508
518
        # selective commits which excludes newly added files.
509
519
        tree = self.make_branch_and_tree('.')
510
520
        self.build_tree(['a', 'b', 'c'])
513
523
        f = file('b', 'wt')
514
524
        f.write('new content')
515
525
        f.close()
516
 
        # set a progress bar that captures the calls so we can see what is 
 
526
        # set a progress bar that captures the calls so we can see what is
517
527
        # emitted
518
528
        self.old_ui_factory = ui.ui_factory
519
529
        self.addCleanup(self.restoreDefaults)
525
535
        tree.commit('second post', specific_files=['b'])
526
536
        # 5 steps, the first of which is reported 2 times, once per dir
527
537
        self.assertEqual(
528
 
            [('update', 1, 5, 'Collecting changes [Directory 0] - Stage'),
529
 
             ('update', 1, 5, 'Collecting changes [Directory 1] - Stage'),
 
538
            [('update', 1, 5, 'Collecting changes [0] - Stage'),
 
539
             ('update', 1, 5, 'Collecting changes [1] - Stage'),
530
540
             ('update', 2, 5, 'Saving data locally - Stage'),
531
541
             ('update', 3, 5, 'Running pre_commit hooks - Stage'),
532
542
             ('update', 4, 5, 'Updating the working tree - Stage'),
536
546
 
537
547
    def test_commit_progress_shows_post_hook_names(self):
538
548
        tree = self.make_branch_and_tree('.')
539
 
        # set a progress bar that captures the calls so we can see what is 
 
549
        # set a progress bar that captures the calls so we can see what is
540
550
        # emitted
541
551
        self.old_ui_factory = ui.ui_factory
542
552
        self.addCleanup(self.restoreDefaults)
548
558
                                               'hook name')
549
559
        tree.commit('first post')
550
560
        self.assertEqual(
551
 
            [('update', 1, 5, 'Collecting changes [Directory 0] - Stage'),
552
 
             ('update', 1, 5, 'Collecting changes [Directory 1] - Stage'),
 
561
            [('update', 1, 5, 'Collecting changes [0] - Stage'),
 
562
             ('update', 1, 5, 'Collecting changes [1] - Stage'),
553
563
             ('update', 2, 5, 'Saving data locally - Stage'),
554
564
             ('update', 3, 5, 'Running pre_commit hooks - Stage'),
555
565
             ('update', 4, 5, 'Updating the working tree - Stage'),
561
571
 
562
572
    def test_commit_progress_shows_pre_hook_names(self):
563
573
        tree = self.make_branch_and_tree('.')
564
 
        # set a progress bar that captures the calls so we can see what is 
 
574
        # set a progress bar that captures the calls so we can see what is
565
575
        # emitted
566
576
        self.old_ui_factory = ui.ui_factory
567
577
        self.addCleanup(self.restoreDefaults)
573
583
                                               'hook name')
574
584
        tree.commit('first post')
575
585
        self.assertEqual(
576
 
            [('update', 1, 5, 'Collecting changes [Directory 0] - Stage'),
577
 
             ('update', 1, 5, 'Collecting changes [Directory 1] - Stage'),
 
586
            [('update', 1, 5, 'Collecting changes [0] - Stage'),
 
587
             ('update', 1, 5, 'Collecting changes [1] - Stage'),
578
588
             ('update', 2, 5, 'Saving data locally - Stage'),
579
589
             ('update', 3, 5, 'Running pre_commit hooks - Stage'),
580
590
             ('update', 3, 5, 'Running pre_commit hooks [hook name] - Stage'),
585
595
           )
586
596
 
587
597
    def test_start_commit_hook(self):
588
 
        """Make sure a start commit hook can modify the tree that is 
 
598
        """Make sure a start commit hook can modify the tree that is
589
599
        committed."""
590
600
        def start_commit_hook_adds_file(tree):
591
601
            open(tree.abspath("newfile"), 'w').write("data")
601
611
        revid = tree.commit('first post')
602
612
        committed_tree = tree.basis_tree()
603
613
        self.assertTrue(committed_tree.has_filename("newfile"))
 
614
 
 
615
    def test_post_commit_hook(self):
 
616
        """Make sure a post_commit hook is called after a commit."""
 
617
        def post_commit_hook_test_params(params):
 
618
            self.assertTrue(isinstance(params,
 
619
                mutabletree.PostCommitHookParams))
 
620
            self.assertTrue(isinstance(params.mutable_tree,
 
621
                mutabletree.MutableTree))
 
622
            open(tree.abspath("newfile"), 'w').write("data")
 
623
            params.mutable_tree.add(["newfile"])
 
624
        def restoreDefaults():
 
625
            mutabletree.MutableTree.hooks['post_commit'] = []
 
626
        self.addCleanup(restoreDefaults)
 
627
        tree = self.make_branch_and_tree('.')
 
628
        mutabletree.MutableTree.hooks.install_named_hook(
 
629
            'post_commit',
 
630
            post_commit_hook_test_params,
 
631
            None)
 
632
        self.assertFalse(tree.has_filename("newfile"))
 
633
        revid = tree.commit('first post')
 
634
        self.assertTrue(tree.has_filename("newfile"))
 
635
        committed_tree = tree.basis_tree()
 
636
        self.assertFalse(committed_tree.has_filename("newfile"))