← 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

[merge] John, sftp and others

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_versioning.py
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.tests 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): 
25
33
        """Basic 'bzr mkdir' operation"""
26
 
        from bzrlib.commands import run_bzr
27
 
        import os
28
34
 
29
 
        run_bzr(['bzr', 'init'])
30
 
        run_bzr(['bzr', 'mkdir', 'foo'])
 
35
        self.run_bzr('init')
 
36
        self.run_bzr('mkdir', 'foo')
31
37
        self.assert_(os.path.isdir('foo'))
32
38
 
33
 
        self.assertRaises(OSError, run_bzr, ['bzr', 'mkdir', 'foo'])
 
39
        self.run_bzr('mkdir', 'foo', retcode=3)
34
40
 
35
 
        from bzrlib.diff import compare_trees, TreeDelta
 
41
        from bzrlib.diff import compare_trees
36
42
        from bzrlib.branch import Branch
37
 
        b = Branch('.')
 
43
        b = Branch.open(u'.')
38
44
        
39
45
        delta = compare_trees(b.basis_tree(), b.working_tree())
40
46
 
 
47
        self.log('delta.added = %r' % delta.added)
 
48
 
41
49
        self.assertEquals(len(delta.added), 1)
42
50
        self.assertEquals(delta.added[0][0], 'foo')
43
51
        self.failIf(delta.modified)
44
52
 
45
 
 
46
 
 
47
 
class AddInUnversioned(InTempDir):
48
 
    def runTest(self):
 
53
    def test_mkdir_in_subdir(self):
 
54
        """'bzr mkdir' operation in subdirectory"""
 
55
 
 
56
        self.run_bzr('init')
 
57
        self.run_bzr('mkdir', 'dir')
 
58
        self.assert_(os.path.isdir('dir'))
 
59
 
 
60
        os.chdir('dir')
 
61
        self.log('Run mkdir in subdir')
 
62
        self.run_bzr('mkdir', 'subdir')
 
63
        self.assert_(os.path.isdir('subdir'))
 
64
        os.chdir('..')
 
65
 
 
66
        from bzrlib.diff import compare_trees
 
67
        from bzrlib.branch import Branch
 
68
        b = Branch.open(u'.')
 
69
        
 
70
        delta = compare_trees(b.basis_tree(), b.working_tree())
 
71
 
 
72
        self.log('delta.added = %r' % delta.added)
 
73
 
 
74
        self.assertEquals(len(delta.added), 2)
 
75
        self.assertEquals(delta.added[0][0], 'dir')
 
76
        self.assertEquals(delta.added[1][0], os.path.join('dir','subdir'))
 
77
        self.failIf(delta.modified)
 
78
 
 
79
    def test_mkdir_w_nested_trees(self):
 
80
        """'bzr mkdir' with nested trees"""
 
81
 
 
82
        self.run_bzr('init')
 
83
        os.mkdir('a')
 
84
        os.chdir('a')
 
85
        self.run_bzr('init')
 
86
        os.mkdir('b')
 
87
        os.chdir('b')
 
88
        self.run_bzr('init')
 
89
        os.chdir('../..')
 
90
 
 
91
        self.run_bzr('mkdir', 'dir', 'a/dir', 'a/b/dir')
 
92
        self.failUnless(os.path.isdir('dir'))
 
93
        self.failUnless(os.path.isdir('a/dir'))
 
94
        self.failUnless(os.path.isdir('a/b/dir'))
 
95
 
 
96
        from bzrlib.diff import compare_trees
 
97
        b = Branch.open(u'.')
 
98
        b_a = Branch.open('a')
 
99
        b_b = Branch.open('a/b')
 
100
        
 
101
        delta = compare_trees(b.basis_tree(), b.working_tree())
 
102
        self.assertEquals(len(delta.added), 1)
 
103
        self.assertEquals(delta.added[0][0], 'dir')
 
104
        self.failIf(delta.modified)
 
105
 
 
106
        delta = compare_trees(b_a.basis_tree(), b_a.working_tree())
 
107
        self.assertEquals(len(delta.added), 1)
 
108
        self.assertEquals(delta.added[0][0], 'dir')
 
109
        self.failIf(delta.modified)
 
110
 
 
111
        delta = compare_trees(b_b.basis_tree(), b_b.working_tree())
 
112
        self.assertEquals(len(delta.added), 1)
 
113
        self.assertEquals(delta.added[0][0], 'dir')
 
114
        self.failIf(delta.modified)
 
115
 
 
116
    def test_working_tree_add_in_unversioned(self):
49
117
        """Try to add a file in an unversioned directory.
50
118
 
51
 
        smart_add may eventually add the parent as necessary, but simple
52
 
        branch add doesn't do that.
 
119
        "bzr add" adds the parent as necessary, but simple working tree add
 
120
        doesn't do that.
53
121
        """
54
122
        from bzrlib.branch import Branch
55
 
        import os
56
123
        from bzrlib.errors import NotVersionedError
 
124
        from bzrlib.workingtree import WorkingTree
57
125
 
58
 
        b = Branch('.', init=True)
 
126
        b = Branch.initialize(u'.')
59
127
 
60
128
        self.build_tree(['foo/',
61
129
                         'foo/hello'])
62
130
 
63
131
        self.assertRaises(NotVersionedError,
64
 
                          b.add,
 
132
                          WorkingTree(b.base, b).add,
65
133
                          'foo/hello')
66
134
        
67
 
 
68
 
 
69
 
TEST_CLASSES = [
70
 
    Mkdir,
71
 
    AddInUnversioned,
72
 
    ]
 
135
        self.check_branch()
 
136
 
 
137
    def check_branch(self):
 
138
        """After all the above changes, run the check and upgrade commands.
 
139
 
 
140
        The upgrade should be a no-op."""
 
141
        b = Branch.open(u'.')
 
142
        mutter('branch has %d revisions', b.revno())
 
143
        
 
144
        mutter('check branch...')
 
145
        from bzrlib.check import check
 
146
        check(b, False)
 
147
 
 
148
 
 
149
class SubdirCommit(TestCaseInTempDir):
 
150
 
 
151
    def test_subdir_commit(self):
 
152
        """Test committing a subdirectory, and committing within a directory."""
 
153
        run_bzr = self.run_bzr
 
154
        eq = self.assertEqual
 
155
 
 
156
        self.build_tree(['a/', 'b/'])
 
157
        
 
158
        run_bzr('init')
 
159
        b = Branch.open(u'.')
 
160
        
 
161
        for fn in ('a/one', 'b/two', 'top'):
 
162
            file(fn, 'w').write('old contents')
 
163
            
 
164
        run_bzr('add')
 
165
        run_bzr('commit', '-m', 'first revision')
 
166
        
 
167
        for fn in ('a/one', 'b/two', 'top'):
 
168
            file(fn, 'w').write('new contents')
 
169
            
 
170
        mutter('start selective subdir commit')
 
171
        run_bzr('commit', 'a', '-m', 'commit a only')
 
172
        
 
173
        old = b.revision_tree(b.get_rev_id(1))
 
174
        new = b.revision_tree(b.get_rev_id(2))
 
175
        
 
176
        eq(new.get_file_by_path('b/two').read(), 'old contents')
 
177
        eq(new.get_file_by_path('top').read(), 'old contents')
 
178
        eq(new.get_file_by_path('a/one').read(), 'new contents')
 
179
        
 
180
        os.chdir('a')
 
181
        # commit from here should do nothing
 
182
        run_bzr('commit', '.', '-m', 'commit subdir only', '--unchanged')
 
183
        v3 = b.revision_tree(b.get_rev_id(3))
 
184
        eq(v3.get_file_by_path('b/two').read(), 'old contents')
 
185
        eq(v3.get_file_by_path('top').read(), 'old contents')
 
186
        eq(v3.get_file_by_path('a/one').read(), 'new contents')
 
187
                
 
188
        # commit in subdirectory commits whole tree
 
189
        run_bzr('commit', '-m', 'commit whole tree from subdir')
 
190
        v4 = b.revision_tree(b.get_rev_id(4))
 
191
        eq(v4.get_file_by_path('b/two').read(), 'new contents')        
 
192
        eq(v4.get_file_by_path('top').read(), 'new contents')
 
193
        
 
194
        # TODO: factor out some kind of assert_tree_state() method
 
195
        
 
196
 
 
197
if __name__ == '__main__':
 
198
    import unittest
 
199
    unittest.main()
 
200