~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

  • Committer: Aaron Bentley
  • Date: 2005-07-26 14:06:11 UTC
  • mto: (1092.1.41) (1185.3.4) (974.1.47)
  • mto: This revision was merged to the branch mainline in revision 982.
  • Revision ID: abentley@panoramicfeedback.com-20050726140611-403e366f3c79c1f1
Fixed python invocation

Show diffs side-by-side

added added

removed removed

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