~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: 2014-02-12 18:22:22 UTC
  • mfrom: (6589.2.1 trunk)
  • Revision ID: pqm@pqm.ubuntu.com-20140212182222-beouo25gaf1cny76
(vila) The XDG Base Directory Specification uses the XDG_CACHE_HOME,
 not XDG_CACHE_DIR. (Andrew Starr-Bochicchio)

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
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
 
from bzrlib.tests import HardlinkFeature
33
 
 
34
 
 
35
 
class TestCheckout(ExternalBase):
 
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):
36
38
 
37
39
    def setUp(self):
38
40
        super(TestCheckout, self).setUp()
39
 
        tree = bzrdir.BzrDir.create_standalone_workingtree('branch')
 
41
        tree = controldir.ControlDir.create_standalone_workingtree('branch')
40
42
        tree.commit('1', rev_id='1', allow_pointless=True)
41
43
        self.build_tree(['branch/added_in_2'])
42
44
        tree.add('added_in_2')
45
47
    def test_checkout_makes_bound_branch(self):
46
48
        self.run_bzr('checkout branch checkout')
47
49
        # if we have a checkout, the branch base should be 'branch'
48
 
        source = bzrdir.BzrDir.open('branch')
49
 
        result = bzrdir.BzrDir.open('checkout')
 
50
        source = controldir.ControlDir.open('branch')
 
51
        result = controldir.ControlDir.open('checkout')
50
52
        self.assertEqual(source.open_branch().bzrdir.root_transport.base,
51
53
                         result.open_branch().get_bound_location())
52
54
 
53
55
    def test_checkout_light_makes_checkout(self):
54
56
        self.run_bzr('checkout --lightweight branch checkout')
55
57
        # if we have a checkout, the branch base should be 'branch'
56
 
        source = bzrdir.BzrDir.open('branch')
57
 
        result = bzrdir.BzrDir.open('checkout')
 
58
        source = controldir.ControlDir.open('branch')
 
59
        result = controldir.ControlDir.open('checkout')
58
60
        self.assertEqual(source.open_branch().bzrdir.root_transport.base,
59
61
                         result.open_branch().bzrdir.root_transport.base)
60
62
 
61
63
    def test_checkout_dash_r(self):
62
 
        self.run_bzr('checkout -r -2 branch checkout')
 
64
        out, err = self.run_bzr(['checkout', '-r', '-2', 'branch', 'checkout'])
63
65
        # the working tree should now be at revision '1' with the content
64
66
        # from 1.
65
 
        result = bzrdir.BzrDir.open('checkout')
 
67
        result = controldir.ControlDir.open('checkout')
66
68
        self.assertEqual(['1'], result.open_workingtree().get_parent_ids())
67
 
        self.failIfExists('checkout/added_in_2')
 
69
        self.assertPathDoesNotExist('checkout/added_in_2')
68
70
 
69
71
    def test_checkout_light_dash_r(self):
70
 
        self.run_bzr('checkout --lightweight -r -2 branch checkout')
 
72
        out, err = self.run_bzr(['checkout','--lightweight', '-r', '-2',
 
73
            'branch', 'checkout'])
71
74
        # the working tree should now be at revision '1' with the content
72
75
        # from 1.
73
 
        result = bzrdir.BzrDir.open('checkout')
 
76
        result = controldir.ControlDir.open('checkout')
74
77
        self.assertEqual(['1'], result.open_workingtree().get_parent_ids())
75
 
        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()
76
86
 
77
87
    def test_checkout_reconstitutes_working_trees(self):
78
88
        # doing a 'bzr checkout' in the directory of a branch with no tree
79
89
        # or a 'bzr checkout path' with path the name of a directory with
80
90
        # a branch with no tree will reconsistute the tree.
81
91
        os.mkdir('treeless-branch')
82
 
        branch = bzrdir.BzrDir.create_branch_convenience(
 
92
        branch = controldir.ControlDir.create_branch_convenience(
83
93
            'treeless-branch',
84
94
            force_new_tree=False,
85
95
            format=bzrdir.BzrDirMetaFormat1())
92
102
        out, err = self.run_bzr('diff treeless-branch')
93
103
 
94
104
        # now test with no parameters
95
 
        branch = bzrdir.BzrDir.create_branch_convenience(
 
105
        branch = controldir.ControlDir.create_branch_convenience(
96
106
            '.',
97
107
            force_new_tree=False,
98
108
            format=bzrdir.BzrDirMetaFormat1())
118
128
            cmd.append('--lightweight')
119
129
        self.run_bzr('checkout source target')
120
130
        # files with unique content should be moved
121
 
        self.failUnlessExists('target/file2.moved')
 
131
        self.assertPathExists('target/file2.moved')
122
132
        # files with content matching tree should not be moved
123
 
        self.failIfExists('target/file1.moved')
 
133
        self.assertPathDoesNotExist('target/file1.moved')
124
134
 
125
135
    def test_checkout_existing_dir_heavy(self):
126
136
        self._test_checkout_existing_dir(False)
131
141
    def test_checkout_in_branch_with_r(self):
132
142
        branch = _mod_branch.Branch.open('branch')
133
143
        branch.bzrdir.destroy_workingtree()
134
 
        os.chdir('branch')
135
 
        self.run_bzr('checkout -r 1')
136
 
        tree = workingtree.WorkingTree.open('.')
 
144
        self.run_bzr('checkout -r 1', working_dir='branch')
 
145
        tree = workingtree.WorkingTree.open('branch')
137
146
        self.assertEqual('1', tree.last_revision())
138
147
        branch.bzrdir.destroy_workingtree()
139
 
        self.run_bzr('checkout -r 0')
 
148
        self.run_bzr('checkout -r 0', working_dir='branch')
140
149
        self.assertEqual('null:', tree.last_revision())
141
150
 
142
151
    def test_checkout_files_from(self):
150
159
        self.build_tree(['source/file1'])
151
160
        source.add('file1')
152
161
        source.commit('added file')
153
 
        self.run_bzr(['checkout', 'source', 'target', '--files-from', 'source',
154
 
                      '--hardlink'])
 
162
        out, err = self.run_bzr('checkout source target --hardlink')
155
163
        source_stat = os.stat('source/file1')
156
164
        target_stat = os.stat('target/file1')
157
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)