~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

  • Committer: Vincent Ladeuil
  • Date: 2008-01-29 15:16:31 UTC
  • mto: (3206.1.1 trunk)
  • mto: This revision was merged to the branch mainline in revision 3207.
  • Revision ID: v.ladeuil+lp@free.fr-20080129151631-vqjd13tb405mobx6
Fix two more leaking tmp dirs, by reworking TransformPreview lock handling.

* bzrlib/tests/test_transform.py:
(TestTransformMerge): Revert previous patch and cleanly call
preview.finalize now that we can.

* bzrlib/tests/test_merge.py:
(TestMerge.test_make_preview_transform): Catch TransformPreview
leak.

* bzrlib/builtins.py:
(cmd_merge._do_preview): Finalize the TransformPreview or the
limbodir will stay in /tmp.

* bzrlib/transform.py:
(TreeTransformBase.__init__): Create the _deletiondir since it's
reffered to by finalize.
(TreeTransformBase.finalize): Delete the dir only if _deletiondir
is set.
(TreeTransform.__init__): Use a temp var for deletiondir and set
the attribute after the base class __init__ has been called.
(TransformPreview.__init__): Read locks the tree since finalize
wants to unlock it (as suggested by Aaron).

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006 by Canonical Ltd
2
 
 
 
1
# Copyright (C) 2005, 2006 Canonical Ltd
 
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
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
 
 
7
#
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
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
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
22
22
import shutil
23
23
import sys
24
24
 
25
 
import bzrlib.bzrdir as bzrdir
26
 
import bzrlib.errors as errors
 
25
from bzrlib import (
 
26
    branch as _mod_branch,
 
27
    bzrdir,
 
28
    errors,
 
29
    workingtree,
 
30
    )
27
31
from bzrlib.tests.blackbox import ExternalBase
28
32
 
29
33
 
38
42
        tree.commit('2', rev_id='2')
39
43
 
40
44
    def test_checkout_makes_bound_branch(self):
41
 
        self.runbzr('checkout branch checkout')
 
45
        self.run_bzr('checkout branch checkout')
42
46
        # if we have a checkout, the branch base should be 'branch'
43
47
        source = bzrdir.BzrDir.open('branch')
44
48
        result = bzrdir.BzrDir.open('checkout')
46
50
                         result.open_branch().get_bound_location())
47
51
 
48
52
    def test_checkout_light_makes_checkout(self):
49
 
        self.runbzr('checkout --lightweight branch checkout')
 
53
        self.run_bzr('checkout --lightweight branch checkout')
50
54
        # if we have a checkout, the branch base should be 'branch'
51
55
        source = bzrdir.BzrDir.open('branch')
52
56
        result = bzrdir.BzrDir.open('checkout')
54
58
                         result.open_branch().bzrdir.root_transport.base)
55
59
 
56
60
    def test_checkout_dash_r(self):
57
 
        self.runbzr('checkout -r -2 branch checkout')
 
61
        self.run_bzr('checkout -r -2 branch checkout')
58
62
        # the working tree should now be at revision '1' with the content
59
63
        # from 1.
60
64
        result = bzrdir.BzrDir.open('checkout')
61
 
        self.assertEqual('1', result.open_workingtree().last_revision())
 
65
        self.assertEqual(['1'], result.open_workingtree().get_parent_ids())
62
66
        self.failIfExists('checkout/added_in_2')
63
67
 
64
68
    def test_checkout_light_dash_r(self):
65
 
        self.runbzr('checkout --lightweight -r -2 branch checkout')
 
69
        self.run_bzr('checkout --lightweight -r -2 branch checkout')
66
70
        # the working tree should now be at revision '1' with the content
67
71
        # from 1.
68
72
        result = bzrdir.BzrDir.open('checkout')
69
 
        self.assertEqual('1', result.open_workingtree().last_revision())
 
73
        self.assertEqual(['1'], result.open_workingtree().get_parent_ids())
70
74
        self.failIfExists('checkout/added_in_2')
71
75
 
72
76
    def test_checkout_reconstitutes_working_trees(self):
80
84
            format=bzrdir.BzrDirMetaFormat1())
81
85
        # check no tree was created
82
86
        self.assertRaises(errors.NoWorkingTree, branch.bzrdir.open_workingtree)
83
 
        out, err = self.run_bzr('checkout', 'treeless-branch')
 
87
        out, err = self.run_bzr('checkout treeless-branch')
84
88
        # we should have a tree now
85
89
        branch.bzrdir.open_workingtree()
86
90
        # with no diff
87
 
        out, err = self.run_bzr('diff', 'treeless-branch')
 
91
        out, err = self.run_bzr('diff treeless-branch')
88
92
 
89
93
        # now test with no parameters
90
94
        branch = bzrdir.BzrDir.create_branch_convenience(
98
102
        branch.bzrdir.open_workingtree()
99
103
        # with no diff
100
104
        out, err = self.run_bzr('diff')
 
105
 
 
106
    def _test_checkout_existing_dir(self, lightweight):
 
107
        source = self.make_branch_and_tree('source')
 
108
        self.build_tree_contents([('source/file1', 'content1'),
 
109
                                  ('source/file2', 'content2'),])
 
110
        source.add(['file1', 'file2'])
 
111
        source.commit('added files')
 
112
        self.build_tree_contents([('target/', ''),
 
113
                                  ('target/file1', 'content1'),
 
114
                                  ('target/file2', 'content3'),])
 
115
        cmd = ['checkout', 'source', 'target']
 
116
        if lightweight:
 
117
            cmd.append('--lightweight')
 
118
        self.run_bzr('checkout source target')
 
119
        # files with unique content should be moved
 
120
        self.failUnlessExists('target/file2.moved')
 
121
        # files with content matching tree should not be moved
 
122
        self.failIfExists('target/file1.moved')
 
123
 
 
124
    def test_checkout_existing_dir_heavy(self):
 
125
        self._test_checkout_existing_dir(False)
 
126
 
 
127
    def test_checkout_existing_dir_lightweight(self):
 
128
        self._test_checkout_existing_dir(True)
 
129
 
 
130
    def test_checkout_in_branch_with_r(self):
 
131
        branch = _mod_branch.Branch.open('branch')
 
132
        branch.bzrdir.destroy_workingtree()
 
133
        os.chdir('branch')
 
134
        self.run_bzr('checkout -r 1')
 
135
        tree = workingtree.WorkingTree.open('.')
 
136
        self.assertEqual('1', tree.last_revision())
 
137
        branch.bzrdir.destroy_workingtree()
 
138
        self.run_bzr('checkout -r 0')
 
139
        self.assertEqual('null:', tree.last_revision())
 
140
 
 
141
    def test_checkout_files_from(self):
 
142
        branch = _mod_branch.Branch.open('branch')
 
143
        self.run_bzr(['checkout', 'branch', 'branch2', '--files-from',
 
144
                      'branch'])