← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/blackbox/test_dpush.py

  • Committer: Martin Pool
  • Date: 2010-01-29 10:36:23 UTC
  • mto: This revision was merged to the branch mainline in revision 4992.
  • Revision ID: mbp@sourcefrog.net-20100129103623-hywka5hymo5z13jw
Change url to canonical.com or wiki, plus some doc improvements in passing

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_dpush.py
 
1
# Copyright (C) 2005, 2007, 2008, 2009 Canonical Ltd
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
 
 
18
"""Black-box tests for bzr dpush."""
 
19
 
 
20
 
 
21
import os
 
22
 
 
23
from bzrlib import (
 
24
    branch,
 
25
    bzrdir,
 
26
    foreign,
 
27
    tests,
 
28
    workingtree,
 
29
    )
 
30
from bzrlib.tests import (
 
31
    blackbox,
 
32
    test_foreign,
 
33
    )
 
34
from bzrlib.tests.blackbox import test_push
 
35
 
 
36
 
 
37
def load_tests(standard_tests, module, loader):
 
38
    """Multiply tests for the dpush command."""
 
39
    result = loader.suiteClass()
 
40
 
 
41
    # one for each king of change
 
42
    changes_tests, remaining_tests = tests.split_suite_by_condition(
 
43
        standard_tests, tests.condition_isinstance((
 
44
                TestDpushStrictWithChanges,
 
45
                )))
 
46
    changes_scenarios = [
 
47
        ('uncommitted',
 
48
         dict(_changes_type= '_uncommitted_changes')),
 
49
        ('pending-merges',
 
50
         dict(_changes_type= '_pending_merges')),
 
51
        ('out-of-sync-trees',
 
52
         dict(_changes_type= '_out_of_sync_trees')),
 
53
        ]
 
54
    tests.multiply_tests(changes_tests, changes_scenarios, result)
 
55
    # No parametrization for the remaining tests
 
56
    result.addTests(remaining_tests)
 
57
 
 
58
    return result
 
59
 
 
60
 
 
61
class TestDpush(blackbox.ExternalBase):
 
62
 
 
63
    def setUp(self):
 
64
        super(TestDpush, self).setUp()
 
65
        test_foreign.register_dummy_foreign_for_test(self)
 
66
 
 
67
    def make_dummy_builder(self, relpath):
 
68
        builder = self.make_branch_builder(
 
69
            relpath, format=test_foreign.DummyForeignVcsDirFormat())
 
70
        builder.build_snapshot('revid', None,
 
71
            [('add', ('', 'TREE_ROOT', 'directory', None)),
 
72
             ('add', ('foo', 'fooid', 'file', 'bar'))])
 
73
        return builder
 
74
 
 
75
    def test_dpush_native(self):
 
76
        target_tree = self.make_branch_and_tree("dp")
 
77
        source_tree = self.make_branch_and_tree("dc")
 
78
        output, error = self.run_bzr("dpush -d dc dp", retcode=3)
 
79
        self.assertEquals("", output)
 
80
        self.assertContainsRe(error, 'in the same VCS, lossy push not necessary. Please use regular push.')
 
81
 
 
82
    def test_dpush(self):
 
83
        branch = self.make_dummy_builder('d').get_branch()
 
84
 
 
85
        dc = branch.bzrdir.sprout('dc', force_new_repo=True)
 
86
        self.build_tree(("dc/foo", "blaaaa"))
 
87
        dc.open_workingtree().commit('msg')
 
88
 
 
89
        output, error = self.run_bzr("dpush -d dc d")
 
90
        self.assertEquals(error, "Pushed up to revision 2.\n")
 
91
        self.check_output("", "status dc")
 
92
 
 
93
    def test_dpush_new(self):
 
94
        b = self.make_dummy_builder('d').get_branch()
 
95
 
 
96
        dc = b.bzrdir.sprout('dc', force_new_repo=True)
 
97
        self.build_tree_contents([("dc/foofile", "blaaaa")])
 
98
        dc_tree = dc.open_workingtree()
 
99
        dc_tree.add("foofile")
 
100
        dc_tree.commit("msg")
 
101
 
 
102
        self.check_output("", "dpush -d dc d")
 
103
        self.check_output("2\n", "revno dc")
 
104
        self.check_output("", "status dc")
 
105
 
 
106
    def test_dpush_wt_diff(self):
 
107
        b = self.make_dummy_builder('d').get_branch()
 
108
 
 
109
        dc = b.bzrdir.sprout('dc', force_new_repo=True)
 
110
        self.build_tree_contents([("dc/foofile", "blaaaa")])
 
111
        dc_tree = dc.open_workingtree()
 
112
        dc_tree.add("foofile")
 
113
        newrevid = dc_tree.commit('msg')
 
114
 
 
115
        self.build_tree_contents([("dc/foofile", "blaaaal")])
 
116
        self.check_output("", "dpush -d dc d --no-strict")
 
117
        self.assertFileEqual("blaaaal", "dc/foofile")
 
118
        # if the dummy vcs wasn't that dummy we could uncomment the line below
 
119
        # self.assertFileEqual("blaaaa", "d/foofile")
 
120
        self.check_output('modified:\n  foofile\n', "status dc")
 
121
 
 
122
    def test_diverged(self):
 
123
        builder = self.make_dummy_builder('d')
 
124
 
 
125
        b = builder.get_branch()
 
126
 
 
127
        dc = b.bzrdir.sprout('dc', force_new_repo=True)
 
128
        dc_tree = dc.open_workingtree()
 
129
 
 
130
        self.build_tree_contents([("dc/foo", "bar")])
 
131
        dc_tree.commit('msg1')
 
132
 
 
133
        builder.build_snapshot('revid2', None,
 
134
          [('modify', ('fooid', 'blie'))])
 
135
 
 
136
        output, error = self.run_bzr("dpush -d dc d", retcode=3)
 
137
        self.assertEquals(output, "")
 
138
        self.assertContainsRe(error, "have diverged")
 
139
 
 
140
 
 
141
class TestDpushStrictMixin(object):
 
142
 
 
143
    def setUp(self):
 
144
        test_foreign.register_dummy_foreign_for_test(self)
 
145
        # Create an empty branch where we will be able to push
 
146
        self.foreign = self.make_branch(
 
147
            'to', format=test_foreign.DummyForeignVcsDirFormat())
 
148
 
 
149
    def set_config_push_strict(self, value):
 
150
        # set config var (any of bazaar.conf, locations.conf, branch.conf
 
151
        # should do)
 
152
        conf = self.tree.branch.get_config()
 
153
        conf.set_user_option('dpush_strict', value)
 
154
 
 
155
    _default_command = ['dpush', '../to']
 
156
    _default_pushed_revid = False # Doesn't aplly for dpush
 
157
 
 
158
    def assertPushSucceeds(self, args, pushed_revid=None):
 
159
        self.run_bzr(self._default_command + args,
 
160
                     working_dir=self._default_wd)
 
161
        if pushed_revid is None:
 
162
            # dpush change the revids, so we need to get back to it
 
163
            branch_from = branch.Branch.open(self._default_wd)
 
164
            pushed_revid = branch_from.last_revision()
 
165
        branch_to = branch.Branch.open('to')
 
166
        repo_to = branch_to.repository
 
167
        self.assertTrue(repo_to.has_revision(pushed_revid))
 
168
        self.assertEqual(branch_to.last_revision(), pushed_revid)
 
169
 
 
170
 
 
171
 
 
172
class TestDpushStrictWithoutChanges(TestDpushStrictMixin,
 
173
                                    test_push.TestPushStrictWithoutChanges):
 
174
 
 
175
    def setUp(self):
 
176
        test_push.TestPushStrictWithoutChanges.setUp(self)
 
177
        TestDpushStrictMixin.setUp(self)
 
178
 
 
179
 
 
180
class TestDpushStrictWithChanges(TestDpushStrictMixin,
 
181
                                 test_push.TestPushStrictWithChanges):
 
182
 
 
183
    _changes_type = None # Set by load_tests
 
184
 
 
185
    def setUp(self):
 
186
        test_push.TestPushStrictWithChanges.setUp(self)
 
187
        TestDpushStrictMixin.setUp(self)
 
188
 
 
189
    def test_push_with_revision(self):
 
190
        raise tests.TestNotApplicable('dpush does not handle --revision')
 
191