← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

cleanup of globster code.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_versioning.py
1
 
# Copyright (C) 2005 by Canonical Ltd
2
 
 
 
1
# Copyright (C) 2005, 2006, 2007, 2009, 2010 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
 
# 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
 
18
18
"""Tests of simple versioning operations"""
19
19
 
20
 
 
21
 
from bzrlib.selftest import InTempDir
22
 
 
23
 
class Mkdir(InTempDir):
24
 
    def runTest(self): 
 
20
# TODO: test trying to commit within a directory that is not yet added
 
21
 
 
22
 
 
23
import os
 
24
 
 
25
from bzrlib.branch import Branch
 
26
from bzrlib.osutils import pathjoin
 
27
from bzrlib.tests import TestCaseInTempDir, TestCaseWithTransport
 
28
from bzrlib.trace import mutter
 
29
from bzrlib.workingtree import WorkingTree
 
30
 
 
31
 
 
32
class TestMkdir(TestCaseWithTransport):
 
33
 
 
34
    def test_mkdir_fails_cleanly(self):
 
35
        """'mkdir' fails cleanly when no working tree is available.
 
36
        https://bugs.launchpad.net/bzr/+bug/138600
 
37
        """
 
38
        # Since there is a safety working tree above us, we create a bare repo
 
39
        # here locally.
 
40
        shared_repo = self.make_repository('.')
 
41
        self.run_bzr(['mkdir', 'abc'], retcode=3)
 
42
        self.failIfExists('abc')
 
43
 
 
44
 
 
45
class TestVersioning(TestCaseInTempDir):
 
46
 
 
47
    def test_mkdir(self):
25
48
        """Basic 'bzr mkdir' operation"""
26
 
        from bzrlib.commands import run_bzr
27
 
        import os
28
49
 
29
 
        run_bzr(['init'])
30
 
        run_bzr(['mkdir', 'foo'])
 
50
        self.run_bzr('init')
 
51
        self.run_bzr(['mkdir', 'foo'])
31
52
        self.assert_(os.path.isdir('foo'))
32
53
 
33
 
        self.assertRaises(OSError, run_bzr, ['mkdir', 'foo'])
34
 
 
35
 
        from bzrlib.diff import compare_trees, TreeDelta
36
 
        from bzrlib.branch import Branch
37
 
        b = Branch('.')
38
 
        
39
 
        delta = compare_trees(b.basis_tree(), b.working_tree())
 
54
        self.run_bzr(['mkdir', 'foo'], retcode=3)
 
55
 
 
56
        wt = WorkingTree.open('.')
 
57
 
 
58
        delta = wt.changes_from(wt.basis_tree())
 
59
 
 
60
        self.log('delta.added = %r' % delta.added)
40
61
 
41
62
        self.assertEquals(len(delta.added), 1)
42
63
        self.assertEquals(delta.added[0][0], 'foo')
43
64
        self.failIf(delta.modified)
44
65
 
45
 
 
46
 
 
47
 
class AddInUnversioned(InTempDir):
48
 
    def runTest(self):
49
 
        """Try to add a file in an unversioned directory.
50
 
 
51
 
        smart_add may eventually add the parent as necessary, but simple
52
 
        branch add doesn't do that.
53
 
        """
54
 
        from bzrlib.branch import Branch
55
 
        import os
56
 
        from bzrlib.errors import NotVersionedError
57
 
 
58
 
        b = Branch('.', init=True)
59
 
 
60
 
        self.build_tree(['foo/',
61
 
                         'foo/hello'])
62
 
 
63
 
        self.assertRaises(NotVersionedError,
64
 
                          b.add,
65
 
                          'foo/hello')
66
 
        
67
 
        
68
 
class SubdirCommit(InTempDir):
69
 
    def runTest(self):
70
 
        """Various commits from a subdirectory"""
71
 
        
72
 
        from bzrlib.branch import Branch
73
 
        from bzrlib.commands import run_bzr
74
 
        import os
75
 
        
76
 
        b = Branch('.', init=True)
77
 
        self.build_tree(['src/', 'src/foo.c', 'README'])
78
 
        
79
 
        self.assertEqual(sorted(b.unknowns()),
80
 
                         ['README', 'src'])
81
 
        
82
 
        self.assertEqual(run_bzr(['version']), 0)
83
 
        
84
 
        self.assertEqual(run_bzr(['add']), 0)
85
 
        
86
 
        
87
 
 
88
 
 
89
 
TEST_CLASSES = [
90
 
    Mkdir,
91
 
    AddInUnversioned,
92
 
    ]
 
66
    def test_mkdir_in_subdir(self):
 
67
        """'bzr mkdir' operation in subdirectory"""
 
68
 
 
69
        self.run_bzr('init')
 
70
        self.run_bzr(['mkdir', 'dir'])
 
71
        self.assert_(os.path.isdir('dir'))
 
72
 
 
73
        os.chdir('dir')
 
74
        self.log('Run mkdir in subdir')
 
75
        self.run_bzr(['mkdir', 'subdir'])
 
76
        self.assert_(os.path.isdir('subdir'))
 
77
        os.chdir('..')
 
78
 
 
79
        wt = WorkingTree.open('.')
 
80
 
 
81
        delta = wt.changes_from(wt.basis_tree())
 
82
 
 
83
        self.log('delta.added = %r' % delta.added)
 
84
 
 
85
        self.assertEquals(len(delta.added), 2)
 
86
        self.assertEquals(delta.added[0][0], 'dir')
 
87
        self.assertEquals(delta.added[1][0], pathjoin('dir','subdir'))
 
88
        self.failIf(delta.modified)
 
89
 
 
90
    def test_mkdir_w_nested_trees(self):
 
91
        """'bzr mkdir' with nested trees"""
 
92
 
 
93
        self.run_bzr('init')
 
94
        os.mkdir('a')
 
95
        os.chdir('a')
 
96
        self.run_bzr('init')
 
97
        os.mkdir('b')
 
98
        os.chdir('b')
 
99
        self.run_bzr('init')
 
100
        os.chdir('../..')
 
101
 
 
102
        self.run_bzr(['mkdir', 'dir', 'a/dir', 'a/b/dir'])
 
103
        self.failUnless(os.path.isdir('dir'))
 
104
        self.failUnless(os.path.isdir('a/dir'))
 
105
        self.failUnless(os.path.isdir('a/b/dir'))
 
106
 
 
107
        wt = WorkingTree.open('.')
 
108
        wt_a = WorkingTree.open('a')
 
109
        wt_b = WorkingTree.open('a/b')
 
110
 
 
111
        delta = wt.changes_from(wt.basis_tree())
 
112
        self.assertEquals(len(delta.added), 1)
 
113
        self.assertEquals(delta.added[0][0], 'dir')
 
114
        self.failIf(delta.modified)
 
115
 
 
116
        delta = wt_a.changes_from(wt_a.basis_tree())
 
117
        self.assertEquals(len(delta.added), 1)
 
118
        self.assertEquals(delta.added[0][0], 'dir')
 
119
        self.failIf(delta.modified)
 
120
 
 
121
        delta = wt_b.changes_from(wt_b.basis_tree())
 
122
        self.assertEquals(len(delta.added), 1)
 
123
        self.assertEquals(delta.added[0][0], 'dir')
 
124
        self.failIf(delta.modified)
 
125
 
 
126
    def check_branch(self):
 
127
        """After all the above changes, run the check and upgrade commands.
 
128
 
 
129
        The upgrade should be a no-op."""
 
130
        b = Branch.open(u'.')
 
131
        mutter('branch has %d revisions', b.revno())
 
132
 
 
133
        mutter('check branch...')
 
134
        from bzrlib.check import check
 
135
        check(b, False)
 
136
 
 
137
 
 
138
class SubdirCommit(TestCaseWithTransport):
 
139
 
 
140
    def test_subdir_commit(self):
 
141
        """Test committing a subdirectory, and committing a directory."""
 
142
        tree = self.make_branch_and_tree('.')
 
143
        b = tree.branch
 
144
        self.build_tree(['a/', 'b/'])
 
145
        def set_contents(contents):
 
146
            self.build_tree_contents([
 
147
                ('a/one', contents),
 
148
                ('b/two', contents),
 
149
                ('top', contents),
 
150
                ])
 
151
        set_contents('old contents')
 
152
        tree.smart_add(['.'])
 
153
        tree.commit('first revision')
 
154
        set_contents('new contents')
 
155
 
 
156
        mutter('start selective subdir commit')
 
157
        self.run_bzr(['commit', 'a', '-m', 'commit a only'])
 
158
 
 
159
        new = b.repository.revision_tree(b.get_rev_id(2))
 
160
        new.lock_read()
 
161
 
 
162
        self.assertEqual(new.get_file_by_path('b/two').read(), 'old contents')
 
163
        self.assertEqual(new.get_file_by_path('top').read(), 'old contents')
 
164
        self.assertEqual(new.get_file_by_path('a/one').read(), 'new contents')
 
165
        new.unlock()
 
166
 
 
167
        os.chdir('a')
 
168
        # commit from here should do nothing
 
169
        self.run_bzr(['commit', '.', '-m', 'commit subdir only', '--unchanged'])
 
170
        v3 = b.repository.revision_tree(b.get_rev_id(3))
 
171
        v3.lock_read()
 
172
        self.assertEqual(v3.get_file_by_path('b/two').read(), 'old contents')
 
173
        self.assertEqual(v3.get_file_by_path('top').read(), 'old contents')
 
174
        self.assertEqual(v3.get_file_by_path('a/one').read(), 'new contents')
 
175
        v3.unlock()
 
176
 
 
177
        # commit in subdirectory commits whole tree
 
178
        self.run_bzr(['commit', '-m', 'commit whole tree from subdir'])
 
179
        v4 = b.repository.revision_tree(b.get_rev_id(4))
 
180
        v4.lock_read()
 
181
        self.assertEqual(v4.get_file_by_path('b/two').read(), 'new contents')
 
182
        self.assertEqual(v4.get_file_by_path('top').read(), 'new contents')
 
183
        v4.unlock()
 
184
 
 
185
        # TODO: factor out some kind of assert_tree_state() method