/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/selftest/versioning.py

MergeĀ fromĀ mainline

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2005 by 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
 
 
18
"""Tests of simple versioning operations"""
 
19
 
 
20
# TODO: test trying to commit within a directory that is not yet added
 
21
 
 
22
 
 
23
import os
 
24
 
 
25
from bzrlib.selftest import BzrTestBase, TestCaseInTempDir
 
26
from bzrlib.branch import Branch
 
27
from bzrlib.trace import mutter
 
28
 
 
29
 
 
30
class TestVersioning(TestCaseInTempDir):
 
31
    
 
32
    def test_mkdir(self): 
 
33
        """Basic 'bzr mkdir' operation"""
 
34
 
 
35
        self.run_bzr('init')
 
36
        self.run_bzr('mkdir', 'foo')
 
37
        self.assert_(os.path.isdir('foo'))
 
38
 
 
39
        self.run_bzr('mkdir', 'foo', retcode=3)
 
40
 
 
41
        from bzrlib.diff import compare_trees
 
42
        from bzrlib.branch import Branch
 
43
        b = Branch.open('.')
 
44
        
 
45
        delta = compare_trees(b.basis_tree(), b.working_tree())
 
46
 
 
47
        self.log('delta.added = %r' % delta.added)
 
48
 
 
49
        self.assertEquals(len(delta.added), 1)
 
50
        self.assertEquals(delta.added[0][0], 'foo')
 
51
        self.failIf(delta.modified)
 
52
 
 
53
    def test_branch_add_in_unversioned(self):
 
54
        """Try to add a file in an unversioned directory.
 
55
 
 
56
        "bzr add" adds the parent as necessary, but simple branch add
 
57
        doesn't do that.
 
58
        """
 
59
        from bzrlib.branch import Branch
 
60
        from bzrlib.errors import NotVersionedError
 
61
 
 
62
        b = Branch.initialize('.')
 
63
 
 
64
        self.build_tree(['foo/',
 
65
                         'foo/hello'])
 
66
 
 
67
        self.assertRaises(NotVersionedError,
 
68
                          b.add,
 
69
                          'foo/hello')
 
70
        
 
71
        self.check_branch()
 
72
 
 
73
    def test_add_in_unversioned(self):
 
74
        """Try to add a file in an unversioned directory.
 
75
 
 
76
        "bzr add" should add the parent(s) as necessary.
 
77
        """
 
78
        from bzrlib.branch import Branch
 
79
        eq = self.assertEqual
 
80
 
 
81
        b = Branch.initialize('.')
 
82
 
 
83
        self.build_tree(['inertiatic/', 'inertiatic/esp'])
 
84
        eq(list(b.unknowns()), ['inertiatic'])
 
85
        self.run_bzr('add', 'inertiatic/esp')
 
86
        eq(list(b.unknowns()), [])
 
87
 
 
88
        # Multiple unversioned parents
 
89
        self.build_tree(['veil/', 'veil/cerpin/', 'veil/cerpin/taxt'])
 
90
        eq(list(b.unknowns()), ['veil'])
 
91
        self.run_bzr('add', 'veil/cerpin/taxt')
 
92
        eq(list(b.unknowns()), [])
 
93
 
 
94
        # Check whacky paths work
 
95
        self.build_tree(['cicatriz/', 'cicatriz/esp'])
 
96
        eq(list(b.unknowns()), ['cicatriz'])
 
97
        self.run_bzr('add', 'inertiatic/../cicatriz/esp')
 
98
        eq(list(b.unknowns()), [])
 
99
 
 
100
    def test_add_in_versioned(self):
 
101
        """Try to add a file in a versioned directory.
 
102
 
 
103
        "bzr add" should do this happily.
 
104
        """
 
105
        from bzrlib.branch import Branch
 
106
        eq = self.assertEqual
 
107
 
 
108
        b = Branch.initialize('.')
 
109
 
 
110
        self.build_tree(['inertiatic/', 'inertiatic/esp'])
 
111
        eq(list(b.unknowns()), ['inertiatic'])
 
112
        self.run_bzr('add', '--no-recurse', 'inertiatic')
 
113
        eq(list(b.unknowns()), ['inertiatic'+os.sep+'esp'])
 
114
        self.run_bzr('add', 'inertiatic/esp')
 
115
        eq(list(b.unknowns()), [])
 
116
 
 
117
    def test_subdir_add(self):
 
118
        """Add in subdirectory should add only things from there down"""
 
119
        
 
120
        from bzrlib.branch import Branch
 
121
        
 
122
        eq = self.assertEqual
 
123
        ass = self.assert_
 
124
        chdir = os.chdir
 
125
        
 
126
        b = Branch.initialize('.')
 
127
        t = b.working_tree()
 
128
        self.build_tree(['src/', 'README'])
 
129
        
 
130
        eq(sorted(b.unknowns()),
 
131
           ['README', 'src'])
 
132
        
 
133
        self.run_bzr('add', 'src')
 
134
        
 
135
        self.build_tree(['src/foo.c'])
 
136
        
 
137
        chdir('src')
 
138
        self.run_bzr('add')
 
139
        
 
140
        eq(sorted(b.unknowns()), 
 
141
           ['README'])
 
142
        eq(len(t.read_working_inventory()), 3)
 
143
                
 
144
        chdir('..')
 
145
        self.run_bzr('add')
 
146
        eq(list(b.unknowns()), [])
 
147
 
 
148
        self.check_branch()
 
149
 
 
150
    def check_branch(self):
 
151
        """After all the above changes, run the check and upgrade commands.
 
152
 
 
153
        The upgrade should be a no-op."""
 
154
        b = Branch.open('.')
 
155
        mutter('branch has %d revisions', b.revno())
 
156
        
 
157
        mutter('check branch...')
 
158
        from bzrlib.check import check
 
159
        check(b, False)
 
160
        
 
161
 
 
162
        
 
163
class SubdirCommit(TestCaseInTempDir):
 
164
 
 
165
    def test_subdir_commit(self):
 
166
        """Test committing a subdirectory, and committing within a directory."""
 
167
        run_bzr = self.run_bzr
 
168
        eq = self.assertEqual
 
169
 
 
170
        self.build_tree(['a/', 'b/'])
 
171
        
 
172
        run_bzr('init')
 
173
        b = Branch.open('.')
 
174
        
 
175
        for fn in ('a/one', 'b/two', 'top'):
 
176
            file(fn, 'w').write('old contents')
 
177
            
 
178
        run_bzr('add')
 
179
        run_bzr('commit', '-m', 'first revision')
 
180
        
 
181
        for fn in ('a/one', 'b/two', 'top'):
 
182
            file(fn, 'w').write('new contents')
 
183
            
 
184
        mutter('start selective subdir commit')
 
185
        run_bzr('commit', 'a', '-m', 'commit a only')
 
186
        
 
187
        old = b.revision_tree(b.get_rev_id(1))
 
188
        new = b.revision_tree(b.get_rev_id(2))
 
189
        
 
190
        eq(new.get_file_by_path('b/two').read(), 'old contents')
 
191
        eq(new.get_file_by_path('top').read(), 'old contents')
 
192
        eq(new.get_file_by_path('a/one').read(), 'new contents')
 
193
        
 
194
        os.chdir('a')
 
195
        # commit from here should do nothing
 
196
        run_bzr('commit', '.', '-m', 'commit subdir only', '--unchanged')
 
197
        v3 = b.revision_tree(b.get_rev_id(3))
 
198
        eq(v3.get_file_by_path('b/two').read(), 'old contents')
 
199
        eq(v3.get_file_by_path('top').read(), 'old contents')
 
200
        eq(v3.get_file_by_path('a/one').read(), 'new contents')
 
201
                
 
202
        # commit in subdirectory commits whole tree
 
203
        run_bzr('commit', '-m', 'commit whole tree from subdir')
 
204
        v4 = b.revision_tree(b.get_rev_id(4))
 
205
        eq(v4.get_file_by_path('b/two').read(), 'new contents')        
 
206
        eq(v4.get_file_by_path('top').read(), 'new contents')
 
207
        
 
208
        # TODO: factor out some kind of assert_tree_state() method
 
209
        
 
210
 
 
211
if __name__ == '__main__':
 
212
    import unittest
 
213
    unittest.main()
 
214