~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/whitebox.py

  • Committer: Robert Collins
  • Date: 2005-08-23 06:52:09 UTC
  • mto: (974.1.50) (1185.1.10) (1092.3.1)
  • mto: This revision was merged to the branch mainline in revision 1139.
  • Revision ID: robertc@robertcollins.net-20050823065209-81cd5962c401751b
move io redirection into each test case from the global runner

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
#! /usr/bin/python
 
2
 
 
3
import os
 
4
import unittest
 
5
 
 
6
from bzrlib.selftest import InTempDir, TestCase
 
7
from bzrlib.branch import ScratchBranch, Branch
 
8
from bzrlib.errors import NotBranchError, NotVersionedError
 
9
 
 
10
 
 
11
class Unknowns(InTempDir):
 
12
    def runTest(self):
 
13
        b = Branch('.', init=True)
 
14
 
 
15
        self.build_tree(['hello.txt',
 
16
                         'hello.txt~'])
 
17
 
 
18
        self.assertEquals(list(b.unknowns()),
 
19
                          ['hello.txt'])
 
20
 
 
21
 
 
22
 
 
23
class NoChanges(InTempDir):
 
24
    def runTest(self):
 
25
        from bzrlib.errors import PointlessCommit
 
26
        
 
27
        b = Branch('.', init=True)
 
28
 
 
29
        self.build_tree(['hello.txt'])
 
30
 
 
31
        self.assertRaises(PointlessCommit,
 
32
                          b.commit,
 
33
                          'commit without adding',
 
34
                          allow_pointless=False)
 
35
 
 
36
        b.commit('commit pointless tree',
 
37
                 allow_pointless=True)
 
38
 
 
39
        b.add('hello.txt')
 
40
        
 
41
        b.commit('commit first added file',
 
42
                 allow_pointless=False)
 
43
        
 
44
        self.assertRaises(PointlessCommit,
 
45
                          b.commit,
 
46
                          'commit after adding file',
 
47
                          allow_pointless=False)
 
48
        
 
49
        b.commit('commit pointless revision with one file',
 
50
                 allow_pointless=True)
 
51
 
 
52
        b.add_pending_merge('mbp@892739123-2005-123123')
 
53
        b.commit('commit new merge with no text changes',
 
54
                 allow_pointless=False)
 
55
        
 
56
 
 
57
 
 
58
class ValidateRevisionId(TestCase):
 
59
    def runTest(self):
 
60
        from bzrlib.revision import validate_revision_id
 
61
        validate_revision_id('mbp@sourcefrog.net-20050311061123-96a255005c7c9dbe')
 
62
        
 
63
        self.assertRaises(ValueError,
 
64
                          validate_revision_id,
 
65
                          ' asdkjas')
 
66
 
 
67
 
 
68
        self.assertRaises(ValueError,
 
69
                          validate_revision_id,
 
70
                          'mbp@sourcefrog.net-20050311061123-96a255005c7c9dbe\n')
 
71
 
 
72
 
 
73
        self.assertRaises(ValueError,
 
74
                          validate_revision_id,
 
75
                          ' mbp@sourcefrog.net-20050311061123-96a255005c7c9dbe')
 
76
 
 
77
        self.assertRaises(ValueError,
 
78
                          validate_revision_id,
 
79
                          'Martin Pool <mbp@sourcefrog.net>-20050311061123-96a255005c7c9dbe')
 
80
 
 
81
 
 
82
 
 
83
class PendingMerges(InTempDir):
 
84
    """Tracking pending-merged revisions."""
 
85
    def runTest(self):
 
86
        b = Branch('.', init=True)
 
87
 
 
88
        self.assertEquals(b.pending_merges(), [])
 
89
        
 
90
        b.add_pending_merge('foo@azkhazan-123123-abcabc')
 
91
        
 
92
        self.assertEquals(b.pending_merges(), ['foo@azkhazan-123123-abcabc'])
 
93
    
 
94
        b.add_pending_merge('foo@azkhazan-123123-abcabc')
 
95
        
 
96
        self.assertEquals(b.pending_merges(), ['foo@azkhazan-123123-abcabc'])
 
97
 
 
98
        b.add_pending_merge('wibble@fofof--20050401--1928390812')
 
99
        self.assertEquals(b.pending_merges(),
 
100
                          ['foo@azkhazan-123123-abcabc',
 
101
                           'wibble@fofof--20050401--1928390812'])
 
102
 
 
103
        b.commit("commit from base with two merges")
 
104
 
 
105
        rev = b.get_revision(b.revision_history()[0])
 
106
        self.assertEquals(len(rev.parents), 2)
 
107
        self.assertEquals(rev.parents[0].revision_id,
 
108
                          'foo@azkhazan-123123-abcabc')
 
109
        self.assertEquals(rev.parents[1].revision_id,
 
110
                           'wibble@fofof--20050401--1928390812')
 
111
 
 
112
        # list should be cleared when we do a commit
 
113
        self.assertEquals(b.pending_merges(), [])
 
114
        
 
115
        
 
116
        
 
117
 
 
118
class Revert(InTempDir):
 
119
    """Test selected-file revert"""
 
120
    def runTest(self):
 
121
        b = Branch('.', init=True)
 
122
 
 
123
        self.build_tree(['hello.txt'])
 
124
        file('hello.txt', 'w').write('initial hello')
 
125
 
 
126
        self.assertRaises(NotVersionedError,
 
127
                          b.revert, ['hello.txt'])
 
128
        
 
129
        b.add(['hello.txt'])
 
130
        b.commit('create initial hello.txt')
 
131
 
 
132
        self.check_file_contents('hello.txt', 'initial hello')
 
133
        file('hello.txt', 'w').write('new hello')
 
134
        self.check_file_contents('hello.txt', 'new hello')
 
135
 
 
136
        # revert file modified since last revision
 
137
        b.revert(['hello.txt'])
 
138
        self.check_file_contents('hello.txt', 'initial hello')
 
139
        self.check_file_contents('hello.txt~', 'new hello')
 
140
 
 
141
        # reverting again clobbers the backup
 
142
        b.revert(['hello.txt'])
 
143
        self.check_file_contents('hello.txt', 'initial hello')
 
144
        self.check_file_contents('hello.txt~', 'initial hello')
 
145
 
 
146
 
 
147
 
 
148
class RenameDirs(InTempDir):
 
149
    """Test renaming directories and the files within them."""
 
150
    def runTest(self):
 
151
        b = Branch('.', init=True)
 
152
        self.build_tree(['dir/', 'dir/sub/', 'dir/sub/file'])
 
153
        b.add(['dir', 'dir/sub', 'dir/sub/file'])
 
154
 
 
155
        b.commit('create initial state')
 
156
 
 
157
        # TODO: lift out to a test helper that checks the shape of
 
158
        # an inventory
 
159
        
 
160
        revid = b.revision_history()[0]
 
161
        self.log('first revision_id is {%s}' % revid)
 
162
        
 
163
        inv = b.get_revision_inventory(revid)
 
164
        self.log('contents of inventory: %r' % inv.entries())
 
165
 
 
166
        self.check_inventory_shape(inv,
 
167
                                   ['dir', 'dir/sub', 'dir/sub/file'])
 
168
 
 
169
        b.rename_one('dir', 'newdir')
 
170
 
 
171
        self.check_inventory_shape(b.inventory,
 
172
                                   ['newdir', 'newdir/sub', 'newdir/sub/file'])
 
173
 
 
174
        b.rename_one('newdir/sub', 'newdir/newsub')
 
175
        self.check_inventory_shape(b.inventory,
 
176
                                   ['newdir', 'newdir/newsub',
 
177
                                    'newdir/newsub/file'])
 
178
 
 
179
        
 
180
 
 
181
 
 
182
class BranchPathTestCase(TestCase):
 
183
    """test for branch path lookups
 
184
 
 
185
    Branch.relpath and bzrlib.branch._relpath do a simple but subtle
 
186
    job: given a path (either relative to cwd or absolute), work out
 
187
    if it is inside a branch and return the path relative to the base.
 
188
    """
 
189
 
 
190
    def runTest(self):
 
191
        from bzrlib.branch import _relpath
 
192
        import tempfile, shutil
 
193
        
 
194
        savedir = os.getcwdu()
 
195
        dtmp = tempfile.mkdtemp()
 
196
        # On Mac OSX, /tmp actually expands to /private/tmp
 
197
        dtmp = os.path.realpath(dtmp)
 
198
 
 
199
        def rp(p):
 
200
            return _relpath(dtmp, p)
 
201
        
 
202
        try:
 
203
            # check paths inside dtmp while standing outside it
 
204
            self.assertEqual(rp(os.path.join(dtmp, 'foo')), 'foo')
 
205
 
 
206
            # root = nothing
 
207
            self.assertEqual(rp(dtmp), '')
 
208
 
 
209
            self.assertRaises(NotBranchError,
 
210
                              rp,
 
211
                              '/etc')
 
212
 
 
213
            # now some near-miss operations -- note that
 
214
            # os.path.commonprefix gets these wrong!
 
215
            self.assertRaises(NotBranchError,
 
216
                              rp,
 
217
                              dtmp.rstrip('\\/') + '2')
 
218
 
 
219
            self.assertRaises(NotBranchError,
 
220
                              rp,
 
221
                              dtmp.rstrip('\\/') + '2/foo')
 
222
 
 
223
            # now operations based on relpath of files in current
 
224
            # directory, or nearby
 
225
            os.chdir(dtmp)
 
226
 
 
227
            self.assertEqual(rp('foo/bar/quux'), 'foo/bar/quux')
 
228
 
 
229
            self.assertEqual(rp('foo'), 'foo')
 
230
 
 
231
            self.assertEqual(rp('./foo'), 'foo')
 
232
 
 
233
            self.assertEqual(rp(os.path.abspath('foo')), 'foo')
 
234
 
 
235
            self.assertRaises(NotBranchError,
 
236
                              rp, '../foo')
 
237
 
 
238
        finally:
 
239
            os.chdir(savedir)
 
240
            shutil.rmtree(dtmp)
 
241
 
 
242
 
 
243
 
 
244
 
 
245
TEST_CLASSES = [Unknowns,
 
246
                ValidateRevisionId,
 
247
                PendingMerges,
 
248
                Revert,
 
249
                RenameDirs,
 
250
                BranchPathTestCase,
 
251
                ]