← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

(jelmer) Add RepositoryFormat.is_deprecated(). (Jelmer Vernooij)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_checkout.py
 
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
from cStringIO import StringIO
 
20
import os
 
21
import re
 
22
import shutil
 
23
import sys
 
24
 
 
25
from bzrlib import (
 
26
    branch as _mod_branch,
 
27
    bzrdir,
 
28
    errors,
 
29
    workingtree,
 
30
    )
 
31
from bzrlib.tests import (
 
32
    TestCaseWithTransport,
 
33
    )
 
34
from bzrlib.tests import (
 
35
    HardlinkFeature,
 
36
    KnownFailure,
 
37
    )
 
38
 
 
39
 
 
40
class TestCheckout(TestCaseWithTransport):
 
41
 
 
42
    def setUp(self):
 
43
        super(TestCheckout, self).setUp()
 
44
        tree = bzrdir.BzrDir.create_standalone_workingtree('branch')
 
45
        tree.commit('1', rev_id='1', allow_pointless=True)
 
46
        self.build_tree(['branch/added_in_2'])
 
47
        tree.add('added_in_2')
 
48
        tree.commit('2', rev_id='2')
 
49
 
 
50
    def test_checkout_makes_bound_branch(self):
 
51
        self.run_bzr('checkout branch checkout')
 
52
        # if we have a checkout, the branch base should be 'branch'
 
53
        source = bzrdir.BzrDir.open('branch')
 
54
        result = bzrdir.BzrDir.open('checkout')
 
55
        self.assertEqual(source.open_branch().bzrdir.root_transport.base,
 
56
                         result.open_branch().get_bound_location())
 
57
 
 
58
    def test_checkout_light_makes_checkout(self):
 
59
        self.run_bzr('checkout --lightweight branch checkout')
 
60
        # if we have a checkout, the branch base should be 'branch'
 
61
        source = bzrdir.BzrDir.open('branch')
 
62
        result = bzrdir.BzrDir.open('checkout')
 
63
        self.assertEqual(source.open_branch().bzrdir.root_transport.base,
 
64
                         result.open_branch().bzrdir.root_transport.base)
 
65
 
 
66
    def test_checkout_dash_r(self):
 
67
        out, err = self.run_bzr(['checkout', '-r', '-2', 'branch', 'checkout'])
 
68
        # the working tree should now be at revision '1' with the content
 
69
        # from 1.
 
70
        result = bzrdir.BzrDir.open('checkout')
 
71
        self.assertEqual(['1'], result.open_workingtree().get_parent_ids())
 
72
        self.failIfExists('checkout/added_in_2')
 
73
 
 
74
    def test_checkout_light_dash_r(self):
 
75
        out, err = self.run_bzr(['checkout','--lightweight', '-r', '-2',
 
76
            'branch', 'checkout'])
 
77
        # the working tree should now be at revision '1' with the content
 
78
        # from 1.
 
79
        result = bzrdir.BzrDir.open('checkout')
 
80
        self.assertEqual(['1'], result.open_workingtree().get_parent_ids())
 
81
        self.failIfExists('checkout/added_in_2')
 
82
 
 
83
    def test_checkout_reconstitutes_working_trees(self):
 
84
        # doing a 'bzr checkout' in the directory of a branch with no tree
 
85
        # or a 'bzr checkout path' with path the name of a directory with
 
86
        # a branch with no tree will reconsistute the tree.
 
87
        os.mkdir('treeless-branch')
 
88
        branch = bzrdir.BzrDir.create_branch_convenience(
 
89
            'treeless-branch',
 
90
            force_new_tree=False,
 
91
            format=bzrdir.BzrDirMetaFormat1())
 
92
        # check no tree was created
 
93
        self.assertRaises(errors.NoWorkingTree, branch.bzrdir.open_workingtree)
 
94
        out, err = self.run_bzr('checkout treeless-branch')
 
95
        # we should have a tree now
 
96
        branch.bzrdir.open_workingtree()
 
97
        # with no diff
 
98
        out, err = self.run_bzr('diff treeless-branch')
 
99
 
 
100
        # now test with no parameters
 
101
        branch = bzrdir.BzrDir.create_branch_convenience(
 
102
            '.',
 
103
            force_new_tree=False,
 
104
            format=bzrdir.BzrDirMetaFormat1())
 
105
        # check no tree was created
 
106
        self.assertRaises(errors.NoWorkingTree, branch.bzrdir.open_workingtree)
 
107
        out, err = self.run_bzr('checkout')
 
108
        # we should have a tree now
 
109
        branch.bzrdir.open_workingtree()
 
110
        # with no diff
 
111
        out, err = self.run_bzr('diff')
 
112
 
 
113
    def _test_checkout_existing_dir(self, lightweight):
 
114
        source = self.make_branch_and_tree('source')
 
115
        self.build_tree_contents([('source/file1', 'content1'),
 
116
                                  ('source/file2', 'content2'),])
 
117
        source.add(['file1', 'file2'])
 
118
        source.commit('added files')
 
119
        self.build_tree_contents([('target/', ''),
 
120
                                  ('target/file1', 'content1'),
 
121
                                  ('target/file2', 'content3'),])
 
122
        cmd = ['checkout', 'source', 'target']
 
123
        if lightweight:
 
124
            cmd.append('--lightweight')
 
125
        self.run_bzr('checkout source target')
 
126
        # files with unique content should be moved
 
127
        self.failUnlessExists('target/file2.moved')
 
128
        # files with content matching tree should not be moved
 
129
        self.failIfExists('target/file1.moved')
 
130
 
 
131
    def test_checkout_existing_dir_heavy(self):
 
132
        self._test_checkout_existing_dir(False)
 
133
 
 
134
    def test_checkout_existing_dir_lightweight(self):
 
135
        self._test_checkout_existing_dir(True)
 
136
 
 
137
    def test_checkout_in_branch_with_r(self):
 
138
        branch = _mod_branch.Branch.open('branch')
 
139
        branch.bzrdir.destroy_workingtree()
 
140
        os.chdir('branch')
 
141
        self.run_bzr('checkout -r 1')
 
142
        tree = workingtree.WorkingTree.open('.')
 
143
        self.assertEqual('1', tree.last_revision())
 
144
        branch.bzrdir.destroy_workingtree()
 
145
        self.run_bzr('checkout -r 0')
 
146
        self.assertEqual('null:', tree.last_revision())
 
147
 
 
148
    def test_checkout_files_from(self):
 
149
        branch = _mod_branch.Branch.open('branch')
 
150
        self.run_bzr(['checkout', 'branch', 'branch2', '--files-from',
 
151
                      'branch'])
 
152
 
 
153
    def test_checkout_hardlink(self):
 
154
        self.requireFeature(HardlinkFeature)
 
155
        source = self.make_branch_and_tree('source')
 
156
        self.build_tree(['source/file1'])
 
157
        source.add('file1')
 
158
        source.commit('added file')
 
159
        out, err = self.run_bzr('checkout source target --hardlink')
 
160
        source_stat = os.stat('source/file1')
 
161
        target_stat = os.stat('target/file1')
 
162
        self.assertEqual(source_stat, target_stat)
 
163
 
 
164
    def test_checkout_hardlink_files_from(self):
 
165
        self.requireFeature(HardlinkFeature)
 
166
        source = self.make_branch_and_tree('source')
 
167
        self.build_tree(['source/file1'])
 
168
        source.add('file1')
 
169
        source.commit('added file')
 
170
        source.bzrdir.sprout('second')
 
171
        out, err = self.run_bzr('checkout source target --hardlink'
 
172
                                ' --files-from second')
 
173
        second_stat = os.stat('second/file1')
 
174
        target_stat = os.stat('target/file1')
 
175
        self.assertEqual(second_stat, target_stat)