~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

  • Committer: Patch Queue Manager
  • Date: 2016-01-31 13:36:59 UTC
  • mfrom: (6613.1.5 1538480-match-hostname)
  • Revision ID: pqm@pqm.ubuntu.com-20160131133659-ouy92ee2wlv9xz8m
(vila) Use ssl.match_hostname instead of our own. (Vincent Ladeuil)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
 
1
# Copyright (C) 2006, 2007, 2009-2012 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 'checkout' CLI command."""
18
18
 
19
 
from cStringIO import StringIO
20
19
import os
21
 
import re
22
 
import shutil
23
 
import sys
24
20
 
25
21
from bzrlib import (
26
22
    branch as _mod_branch,
27
23
    bzrdir,
 
24
    controldir,
28
25
    errors,
29
26
    workingtree,
30
27
    )
31
 
from bzrlib.tests.blackbox import ExternalBase
32
 
 
33
 
 
34
 
class TestCheckout(ExternalBase):
35
 
    
 
28
from bzrlib.tests import (
 
29
    TestCaseWithTransport,
 
30
    )
 
31
from bzrlib.tests.matchers import ContainsNoVfsCalls
 
32
from bzrlib.tests.features import (
 
33
    HardlinkFeature,
 
34
    )
 
35
 
 
36
 
 
37
class TestCheckout(TestCaseWithTransport):
 
38
 
36
39
    def setUp(self):
37
40
        super(TestCheckout, self).setUp()
38
 
        tree = bzrdir.BzrDir.create_standalone_workingtree('branch')
 
41
        tree = controldir.ControlDir.create_standalone_workingtree('branch')
39
42
        tree.commit('1', rev_id='1', allow_pointless=True)
40
43
        self.build_tree(['branch/added_in_2'])
41
44
        tree.add('added_in_2')
44
47
    def test_checkout_makes_bound_branch(self):
45
48
        self.run_bzr('checkout branch checkout')
46
49
        # if we have a checkout, the branch base should be 'branch'
47
 
        source = bzrdir.BzrDir.open('branch')
48
 
        result = bzrdir.BzrDir.open('checkout')
 
50
        source = controldir.ControlDir.open('branch')
 
51
        result = controldir.ControlDir.open('checkout')
49
52
        self.assertEqual(source.open_branch().bzrdir.root_transport.base,
50
53
                         result.open_branch().get_bound_location())
51
54
 
52
55
    def test_checkout_light_makes_checkout(self):
53
56
        self.run_bzr('checkout --lightweight branch checkout')
54
57
        # if we have a checkout, the branch base should be 'branch'
55
 
        source = bzrdir.BzrDir.open('branch')
56
 
        result = bzrdir.BzrDir.open('checkout')
 
58
        source = controldir.ControlDir.open('branch')
 
59
        result = controldir.ControlDir.open('checkout')
57
60
        self.assertEqual(source.open_branch().bzrdir.root_transport.base,
58
61
                         result.open_branch().bzrdir.root_transport.base)
59
62
 
60
63
    def test_checkout_dash_r(self):
61
 
        self.run_bzr('checkout -r -2 branch checkout')
 
64
        out, err = self.run_bzr(['checkout', '-r', '-2', 'branch', 'checkout'])
62
65
        # the working tree should now be at revision '1' with the content
63
66
        # from 1.
64
 
        result = bzrdir.BzrDir.open('checkout')
 
67
        result = controldir.ControlDir.open('checkout')
65
68
        self.assertEqual(['1'], result.open_workingtree().get_parent_ids())
66
 
        self.failIfExists('checkout/added_in_2')
 
69
        self.assertPathDoesNotExist('checkout/added_in_2')
67
70
 
68
71
    def test_checkout_light_dash_r(self):
69
 
        self.run_bzr('checkout --lightweight -r -2 branch checkout')
 
72
        out, err = self.run_bzr(['checkout','--lightweight', '-r', '-2',
 
73
            'branch', 'checkout'])
70
74
        # the working tree should now be at revision '1' with the content
71
75
        # from 1.
72
 
        result = bzrdir.BzrDir.open('checkout')
 
76
        result = controldir.ControlDir.open('checkout')
73
77
        self.assertEqual(['1'], result.open_workingtree().get_parent_ids())
74
 
        self.failIfExists('checkout/added_in_2')
 
78
        self.assertPathDoesNotExist('checkout/added_in_2')
 
79
 
 
80
    def test_checkout_into_empty_dir(self):
 
81
        self.make_bzrdir('checkout')
 
82
        out, err = self.run_bzr(['checkout', 'branch', 'checkout'])
 
83
        result = controldir.ControlDir.open('checkout')
 
84
        tree = result.open_workingtree()
 
85
        branch = result.open_branch()
75
86
 
76
87
    def test_checkout_reconstitutes_working_trees(self):
77
88
        # doing a 'bzr checkout' in the directory of a branch with no tree
78
89
        # or a 'bzr checkout path' with path the name of a directory with
79
90
        # a branch with no tree will reconsistute the tree.
80
91
        os.mkdir('treeless-branch')
81
 
        branch = bzrdir.BzrDir.create_branch_convenience(
 
92
        branch = controldir.ControlDir.create_branch_convenience(
82
93
            'treeless-branch',
83
94
            force_new_tree=False,
84
95
            format=bzrdir.BzrDirMetaFormat1())
91
102
        out, err = self.run_bzr('diff treeless-branch')
92
103
 
93
104
        # now test with no parameters
94
 
        branch = bzrdir.BzrDir.create_branch_convenience(
 
105
        branch = controldir.ControlDir.create_branch_convenience(
95
106
            '.',
96
107
            force_new_tree=False,
97
108
            format=bzrdir.BzrDirMetaFormat1())
117
128
            cmd.append('--lightweight')
118
129
        self.run_bzr('checkout source target')
119
130
        # files with unique content should be moved
120
 
        self.failUnlessExists('target/file2.moved')
 
131
        self.assertPathExists('target/file2.moved')
121
132
        # files with content matching tree should not be moved
122
 
        self.failIfExists('target/file1.moved')
 
133
        self.assertPathDoesNotExist('target/file1.moved')
123
134
 
124
135
    def test_checkout_existing_dir_heavy(self):
125
136
        self._test_checkout_existing_dir(False)
130
141
    def test_checkout_in_branch_with_r(self):
131
142
        branch = _mod_branch.Branch.open('branch')
132
143
        branch.bzrdir.destroy_workingtree()
133
 
        os.chdir('branch')
134
 
        self.run_bzr('checkout -r 1')
135
 
        tree = workingtree.WorkingTree.open('.')
 
144
        self.run_bzr('checkout -r 1', working_dir='branch')
 
145
        tree = workingtree.WorkingTree.open('branch')
136
146
        self.assertEqual('1', tree.last_revision())
137
147
        branch.bzrdir.destroy_workingtree()
138
 
        self.run_bzr('checkout -r 0')
 
148
        self.run_bzr('checkout -r 0', working_dir='branch')
139
149
        self.assertEqual('null:', tree.last_revision())
 
150
 
 
151
    def test_checkout_files_from(self):
 
152
        branch = _mod_branch.Branch.open('branch')
 
153
        self.run_bzr(['checkout', 'branch', 'branch2', '--files-from',
 
154
                      'branch'])
 
155
 
 
156
    def test_checkout_hardlink(self):
 
157
        self.requireFeature(HardlinkFeature)
 
158
        source = self.make_branch_and_tree('source')
 
159
        self.build_tree(['source/file1'])
 
160
        source.add('file1')
 
161
        source.commit('added file')
 
162
        out, err = self.run_bzr('checkout source target --hardlink')
 
163
        source_stat = os.stat('source/file1')
 
164
        target_stat = os.stat('target/file1')
 
165
        self.assertEqual(source_stat, target_stat)
 
166
 
 
167
    def test_checkout_hardlink_files_from(self):
 
168
        self.requireFeature(HardlinkFeature)
 
169
        source = self.make_branch_and_tree('source')
 
170
        self.build_tree(['source/file1'])
 
171
        source.add('file1')
 
172
        source.commit('added file')
 
173
        source.bzrdir.sprout('second')
 
174
        out, err = self.run_bzr('checkout source target --hardlink'
 
175
                                ' --files-from second')
 
176
        second_stat = os.stat('second/file1')
 
177
        target_stat = os.stat('target/file1')
 
178
        self.assertEqual(second_stat, target_stat)
 
179
 
 
180
    def test_colo_checkout(self):
 
181
        source = self.make_branch_and_tree('source', format='development-colo')
 
182
        self.build_tree(['source/file1'])
 
183
        source.add('file1')
 
184
        source.commit('added file')
 
185
        target = source.bzrdir.sprout('file:second,branch=somebranch',
 
186
            create_tree_if_local=False)
 
187
        out, err = self.run_bzr('checkout file:,branch=somebranch .',
 
188
            working_dir='second')
 
189
        # We should always be creating a lighweight checkout for colocated
 
190
        # branches.
 
191
        self.assertEquals(
 
192
            target.open_branch(name='somebranch').base,
 
193
            target.get_branch_reference(name=""))
 
194
 
 
195
 
 
196
class TestSmartServerCheckout(TestCaseWithTransport):
 
197
 
 
198
    def test_heavyweight_checkout(self):
 
199
        self.setup_smart_server_with_call_log()
 
200
        t = self.make_branch_and_tree('from')
 
201
        for count in range(9):
 
202
            t.commit(message='commit %d' % count)
 
203
        self.reset_smart_call_log()
 
204
        out, err = self.run_bzr(['checkout', self.get_url('from'), 'target'])
 
205
        # This figure represent the amount of work to perform this use case. It
 
206
        # is entirely ok to reduce this number if a test fails due to rpc_count
 
207
        # being too low. If rpc_count increases, more network roundtrips have
 
208
        # become necessary for this use case. Please do not adjust this number
 
209
        # upwards without agreement from bzr's network support maintainers.
 
210
        self.assertLength(10, self.hpss_calls)
 
211
        self.assertLength(1, self.hpss_connections)
 
212
        self.assertThat(self.hpss_calls, ContainsNoVfsCalls)
 
213
 
 
214
    def test_lightweight_checkout(self):
 
215
        self.setup_smart_server_with_call_log()
 
216
        t = self.make_branch_and_tree('from')
 
217
        for count in range(9):
 
218
            t.commit(message='commit %d' % count)
 
219
        self.reset_smart_call_log()
 
220
        out, err = self.run_bzr(['checkout', '--lightweight', self.get_url('from'),
 
221
            'target'])
 
222
        # This figure represent the amount of work to perform this use case. It
 
223
        # is entirely ok to reduce this number if a test fails due to rpc_count
 
224
        # being too low. If rpc_count increases, more network roundtrips have
 
225
        # become necessary for this use case. Please do not adjust this number
 
226
        # upwards without agreement from bzr's network support maintainers.
 
227
        self.assertLength(13, self.hpss_calls)
 
228
        self.assertThat(self.hpss_calls, ContainsNoVfsCalls)