~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/versioning.py

  • Committer: Martin Pool
  • Date: 2005-04-28 07:24:55 UTC
  • Revision ID: mbp@sourcefrog.net-20050428072453-7b99afa993a1e549
todo

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