~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/versioning.py

  • Committer: Martin Pool
  • Date: 2005-09-12 09:50:44 UTC
  • Revision ID: mbp@sourcefrog.net-20050912095044-6acfdb5611729987
- no tests in bzrlib.fetch anymore

Show diffs side-by-side

added added

removed removed

Lines of Context:
17
17
 
18
18
"""Tests of simple versioning operations"""
19
19
 
 
20
# TODO: test adding a file whose directory is not versioned
20
21
# TODO: test trying to commit within a directory that is not yet added
21
22
 
22
23
 
23
24
import os
24
25
 
25
 
from bzrlib.tests import BzrTestBase, TestCaseInTempDir
 
26
from bzrlib.selftest import BzrTestBase, TestCaseInTempDir
26
27
from bzrlib.branch import Branch
27
 
from bzrlib.trace import mutter
28
 
from bzrlib.osutils import pathjoin
29
 
from bzrlib.workingtree import WorkingTree
 
28
 
 
29
import logging
 
30
logger = logging.getLogger('bzr.test.versioning')
 
31
debug = logger.debug
30
32
 
31
33
 
32
34
class TestVersioning(TestCaseInTempDir):
33
35
    
34
36
    def test_mkdir(self): 
35
37
        """Basic 'bzr mkdir' operation"""
 
38
        from bzrlib.commands import run_bzr
36
39
 
37
 
        self.run_bzr('init')
38
 
        self.run_bzr('mkdir', 'foo')
 
40
        run_bzr(['init'])
 
41
        run_bzr(['mkdir', 'foo'])
39
42
        self.assert_(os.path.isdir('foo'))
40
43
 
41
 
        self.run_bzr('mkdir', 'foo', retcode=3)
 
44
        self.assertRaises(OSError, run_bzr, ['mkdir', 'foo'])
42
45
 
43
46
        from bzrlib.diff import compare_trees
44
 
        wt = WorkingTree.open('.')
 
47
        from bzrlib.branch import Branch
 
48
        b = Branch('.')
45
49
        
46
 
        delta = compare_trees(wt.basis_tree(), wt)
 
50
        delta = compare_trees(b.basis_tree(), b.working_tree())
47
51
 
48
52
        self.log('delta.added = %r' % delta.added)
49
53
 
51
55
        self.assertEquals(delta.added[0][0], 'foo')
52
56
        self.failIf(delta.modified)
53
57
 
54
 
    def test_mkdir_in_subdir(self):
55
 
        """'bzr mkdir' operation in subdirectory"""
56
 
 
57
 
        self.run_bzr('init')
58
 
        self.run_bzr('mkdir', 'dir')
59
 
        self.assert_(os.path.isdir('dir'))
60
 
 
61
 
        os.chdir('dir')
62
 
        self.log('Run mkdir in subdir')
63
 
        self.run_bzr('mkdir', 'subdir')
64
 
        self.assert_(os.path.isdir('subdir'))
65
 
        os.chdir('..')
66
 
 
67
 
        from bzrlib.diff import compare_trees
68
 
        wt = WorkingTree.open('.')
69
 
        
70
 
        delta = compare_trees(wt.basis_tree(), wt)
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], pathjoin('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
 
        wt = WorkingTree.open('.')
98
 
        wt_a = WorkingTree.open('a')
99
 
        wt_b = WorkingTree.open('a/b')
100
 
        
101
 
        delta = compare_trees(wt.basis_tree(), wt)
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(wt_a.basis_tree(), wt_a)
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(wt_b.basis_tree(), wt_b)
112
 
        self.assertEquals(len(delta.added), 1)
113
 
        self.assertEquals(delta.added[0][0], 'dir')
114
 
        self.failIf(delta.modified)
115
 
 
116
 
    def check_branch(self):
 
58
    def test_add_in_unversioned(self):
 
59
        """Try to add a file in an unversioned directory.
 
60
 
 
61
        smart_add may eventually add the parent as necessary, but simple
 
62
        branch add doesn't do that.
 
63
        """
 
64
        from bzrlib.branch import Branch
 
65
        from bzrlib.errors import NotVersionedError
 
66
 
 
67
        b = Branch('.', init=True)
 
68
 
 
69
        self.build_tree(['foo/',
 
70
                         'foo/hello'])
 
71
 
 
72
        self.assertRaises(NotVersionedError,
 
73
                          b.add,
 
74
                          'foo/hello')
 
75
        
 
76
        self.check_and_upgrade()
 
77
 
 
78
        
 
79
    def test_subdir_add(self):
 
80
        """Add in subdirectory should add only things from there down"""
 
81
        
 
82
        from bzrlib.branch import Branch
 
83
        from bzrlib.commands import run_bzr
 
84
        
 
85
        eq = self.assertEqual
 
86
        ass = self.assert_
 
87
        chdir = os.chdir
 
88
        
 
89
        b = Branch('.', init=True)
 
90
        self.build_tree(['src/', 'README'])
 
91
        
 
92
        eq(sorted(b.unknowns()),
 
93
           ['README', 'src'])
 
94
        
 
95
        eq(run_bzr(['add', 'src']), 0)
 
96
        
 
97
        self.build_tree(['src/foo.c'])
 
98
        
 
99
        chdir('src')
 
100
        eq(run_bzr(['add']), 0)
 
101
        
 
102
        eq(sorted(b.unknowns()), 
 
103
           ['README'])
 
104
        eq(len(b.inventory), 3)
 
105
                
 
106
        chdir('..')
 
107
        eq(run_bzr(['add']), 0)
 
108
        eq(list(b.unknowns()), [])
 
109
 
 
110
        self.check_and_upgrade()
 
111
 
 
112
 
 
113
    def check_and_upgrade(self):
117
114
        """After all the above changes, run the check and upgrade commands.
118
115
 
119
116
        The upgrade should be a no-op."""
120
 
        b = Branch.open(u'.')
121
 
        mutter('branch has %d revisions', b.revno())
122
 
        
123
 
        mutter('check branch...')
124
 
        from bzrlib.check import check
125
 
        check(b, False)
126
 
 
127
 
 
 
117
        b = Branch('.')
 
118
        debug('branch has %d revisions', b.revno())
 
119
        
 
120
        debug('check branch...')
 
121
        from bzrlib.check import check
 
122
        check(b)
 
123
        
 
124
        debug('upgrade branch...')
 
125
        from bzrlib.upgrade import upgrade
 
126
        upgrade(b)
 
127
        
 
128
        debug('check branch...')
 
129
        from bzrlib.check import check
 
130
        check(b)
 
131
        
 
132
 
 
133
        
 
134
        
128
135
class SubdirCommit(TestCaseInTempDir):
129
136
 
130
137
    def test_subdir_commit(self):
135
142
        self.build_tree(['a/', 'b/'])
136
143
        
137
144
        run_bzr('init')
138
 
        b = Branch.open(u'.')
 
145
        b = Branch('.')
139
146
        
140
147
        for fn in ('a/one', 'b/two', 'top'):
141
148
            file(fn, 'w').write('old contents')
146
153
        for fn in ('a/one', 'b/two', 'top'):
147
154
            file(fn, 'w').write('new contents')
148
155
            
149
 
        mutter('start selective subdir commit')
150
156
        run_bzr('commit', 'a', '-m', 'commit a only')
151
157
        
152
 
        old = b.repository.revision_tree(b.get_rev_id(1))
153
 
        new = b.repository.revision_tree(b.get_rev_id(2))
 
158
        old = b.revision_tree(b.lookup_revision(1))
 
159
        new = b.revision_tree(b.lookup_revision(2))
154
160
        
155
161
        eq(new.get_file_by_path('b/two').read(), 'old contents')
156
162
        eq(new.get_file_by_path('top').read(), 'old contents')
159
165
        os.chdir('a')
160
166
        # commit from here should do nothing
161
167
        run_bzr('commit', '.', '-m', 'commit subdir only', '--unchanged')
162
 
        v3 = b.repository.revision_tree(b.get_rev_id(3))
 
168
        v3 = b.revision_tree(b.lookup_revision(3))
163
169
        eq(v3.get_file_by_path('b/two').read(), 'old contents')
164
170
        eq(v3.get_file_by_path('top').read(), 'old contents')
165
171
        eq(v3.get_file_by_path('a/one').read(), 'new contents')
166
172
                
167
173
        # commit in subdirectory commits whole tree
168
174
        run_bzr('commit', '-m', 'commit whole tree from subdir')
169
 
        v4 = b.repository.revision_tree(b.get_rev_id(4))
 
175
        v4 = b.revision_tree(b.lookup_revision(4))
170
176
        eq(v4.get_file_by_path('b/two').read(), 'new contents')        
171
177
        eq(v4.get_file_by_path('top').read(), 'new contents')
172
178
        
173
179
        # TODO: factor out some kind of assert_tree_state() method
174
180
        
175
 
 
176
 
if __name__ == '__main__':
177
 
    import unittest
178
 
    unittest.main()
179