~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to testbzr

  • Committer: Martin Pool
  • Date: 2005-06-27 01:39:26 UTC
  • Revision ID: mbp@sourcefrog.net-20050627013926-49413d5928809350
- import effbot.org http client

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
1
#! /usr/bin/python
 
2
# -*- coding: utf-8 -*-
2
3
 
3
4
# Copyright (C) 2005 Canonical Ltd
4
5
 
16
17
# along with this program; if not, write to the Free Software
17
18
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
19
 
19
 
 
20
 
"""External black-box test for bzr.
21
 
 
22
 
This always runs bzr as an external process to try to catch bugs
23
 
related to argument processing, startup, etc.
24
 
 
25
 
This replaces the previous test.sh which was not very portable."""
26
 
 
27
 
import sys, os, traceback
28
 
from os import mkdir
29
 
from os.path import exists
30
 
 
31
 
TESTDIR = "testbzr.tmp"
32
 
 
33
 
LOGFILENAME = 'testbzr.log'
34
 
 
35
 
try:
36
 
    import shutil
37
 
    from subprocess import call, Popen, PIPE
38
 
except ImportError, e:
39
 
    sys.stderr.write("testbzr: sorry, this test suite requires modules from python2.4\n"
40
 
                     + '    ' + str(e))
41
 
    sys.exit(1)
42
 
 
43
 
 
44
 
class CommandFailed(Exception):
45
 
    pass
46
 
 
47
 
 
48
 
def formcmd(cmd):
49
 
    if isinstance(cmd, basestring):
50
 
        logfile.write('$ %s\n' % cmd)
51
 
        cmd = cmd.split()
52
 
    else:
53
 
        logfile.write('$ %r\n' % cmd)
54
 
 
55
 
    if cmd[0] == 'bzr':
56
 
        cmd[0] = BZRPATH
57
 
 
58
 
    return cmd
59
 
 
60
 
 
61
 
def runcmd(cmd, retcode=0):
62
 
    """Run one command and check the return code.
63
 
 
64
 
    Returns a tuple of (stdout,stderr) strings.
65
 
 
66
 
    If a single string is based, it is split into words.
67
 
    For commands that are not simple space-separated words, please
68
 
    pass a list instead."""
69
 
    cmd = formcmd(cmd)
70
 
    log_linenumber()
71
 
    
72
 
    actual_retcode = call(cmd, stdout=logfile, stderr=logfile)
73
 
    
74
 
    if retcode != actual_retcode:
75
 
        raise CommandFailed("test failed: %r returned %d, expected %d"
76
 
                            % (cmd, actual_retcode, retcode))
77
 
 
78
 
 
79
 
 
80
 
def backtick(cmd, retcode=0):
81
 
    cmd = formcmd(cmd)
82
 
    log_linenumber()
83
 
    child = Popen(cmd, stdout=PIPE, stderr=logfile)
84
 
    outd, errd = child.communicate()
85
 
    logfile.write(outd)
86
 
    actual_retcode = child.wait()
87
 
 
88
 
    outd = outd.replace('\r', '')
89
 
    
90
 
    if retcode != actual_retcode:
91
 
        raise CommandFailed("test failed: %r returned %d, expected %d"
92
 
                            % (cmd, actual_retcode, retcode))
93
 
 
94
 
    return outd
95
 
 
96
 
 
97
 
 
98
 
def progress(msg):
99
 
    print '* ' + msg
100
 
    logfile.write('* '+ msg + '\n')
101
 
    log_linenumber()
102
 
 
103
 
 
104
 
def cd(dirname):
105
 
    logfile.write('$ cd %s\n' % dirname)
106
 
    os.chdir(dirname)
107
 
 
108
 
 
109
 
 
110
 
def log_linenumber():
111
 
    """Log the stack frame location two things up."""
112
 
    stack = traceback.extract_stack()[-3]
113
 
    logfile.write('   at %s:%d\n' % stack[:2])
114
 
 
115
 
 
116
 
 
117
 
# prepare an empty scratch directory
118
 
if os.path.exists(TESTDIR):
119
 
    shutil.rmtree(TESTDIR)
120
 
 
121
 
 
122
 
logfile = open(LOGFILENAME, 'wt', buffering=1)
123
 
 
124
 
 
125
 
try:
126
 
    mypath = os.path.abspath(sys.argv[0])
127
 
    print '%-30s %s' % ('running tests from', mypath)
128
 
 
129
 
    global BZRPATH
130
 
 
131
 
    if len(sys.argv) > 1:
132
 
        BZRPATH = sys.argv[1]
133
 
    else:
134
 
        BZRPATH = os.path.join(os.path.split(mypath)[0], 'bzr')
135
 
 
136
 
    print '%-30s %s' % ('against bzr', BZRPATH)
137
 
    print '%-30s %s' % ('in directory', os.getcwd())
138
 
    print
139
 
    print backtick([BZRPATH, 'version'])
140
 
    
141
 
    runcmd(['mkdir', TESTDIR])
142
 
    cd(TESTDIR)
143
 
    test_root = os.getcwd()
144
 
 
145
 
    progress("introductory commands")
146
 
    runcmd("bzr version")
147
 
    runcmd("bzr --version")
148
 
    runcmd("bzr help")
149
 
    runcmd("bzr --help")
150
 
 
151
 
    progress("internal tests")
152
 
    runcmd("bzr selftest")
153
 
 
154
 
    progress("user identity")
155
 
    # this should always identify something, if only "john@localhost"
156
 
    runcmd("bzr whoami")
157
 
    runcmd("bzr whoami --email")
158
 
    assert backtick("bzr whoami --email").count('@') == 1
159
 
 
160
 
    progress("invalid commands")
161
 
    runcmd("bzr pants", retcode=1)
162
 
    runcmd("bzr --pants off", retcode=1)
163
 
    runcmd("bzr diff --message foo", retcode=1)
164
 
 
165
 
    progress("basic branch creation")
166
 
    runcmd(['mkdir', 'branch1'])
167
 
    cd('branch1')
168
 
    runcmd('bzr init')
169
 
 
170
 
    assert backtick('bzr root')[:-1] == os.path.join(test_root, 'branch1')
171
 
 
172
 
    progress("status of new file")
173
 
    
174
 
    f = file('test.txt', 'wt')
175
 
    f.write('hello world!\n')
176
 
    f.close()
177
 
 
178
 
    out = backtick("bzr unknowns")
179
 
    assert out == 'test.txt\n'
180
 
 
181
 
    out = backtick("bzr status")
182
 
    assert out == '''?       test.txt\n'''
183
 
 
184
 
    out = backtick("bzr status --all")
185
 
    assert out == "?       test.txt\n"
186
 
 
187
 
    out = backtick("bzr status test.txt --all")
188
 
    assert out == "?       test.txt\n"
189
 
 
190
 
    f = file('test2.txt', 'wt')
191
 
    f.write('goodbye cruel world...\n')
192
 
    f.close()
193
 
 
194
 
    out = backtick("bzr status test.txt")
195
 
    assert out == "?       test.txt\n"
196
 
 
197
 
    out = backtick("bzr status")
198
 
    assert out == "?       test.txt\n" \
199
 
                + "?       test2.txt\n"
200
 
 
201
 
    os.unlink('test2.txt')
202
 
 
203
 
    progress("command aliases")
204
 
    out = backtick("bzr st --all")
205
 
    assert out == "?       test.txt\n"
206
 
    out = backtick("bzr stat")
207
 
    assert out == "?       test.txt\n"
208
 
 
209
 
    progress("command help")
210
 
    runcmd("bzr help st")
211
 
    runcmd("bzr help")
212
 
    runcmd("bzr help commands")
213
 
    runcmd("bzr help slartibartfast", 1)
214
 
 
215
 
    out = backtick("bzr help ci")
216
 
    out.index('aliases: ')
217
 
 
218
 
    progress("can't rename unversioned file")
219
 
    runcmd("bzr rename test.txt new-test.txt", 1)
220
 
 
221
 
    progress("adding a file")
222
 
 
223
 
    runcmd("bzr add test.txt")
224
 
    assert backtick("bzr unknowns") == ''
225
 
    assert backtick("bzr status --all") == "A       test.txt\n"
226
 
 
227
 
    progress("rename newly-added file")
228
 
    runcmd("bzr rename test.txt hello.txt")
229
 
    assert os.path.exists("hello.txt")
230
 
    assert not os.path.exists("test.txt")
231
 
 
232
 
    assert backtick("bzr revno") == '0\n'
233
 
 
234
 
    progress("add first revision")
235
 
    runcmd(["bzr", "commit", "-m", 'add first revision'])
236
 
 
237
 
    progress("more complex renames")
238
 
    os.mkdir("sub1")
239
 
    runcmd("bzr rename hello.txt sub1", 1)
240
 
    runcmd("bzr rename hello.txt sub1/hello.txt", 1)
241
 
    runcmd("bzr move hello.txt sub1", 1)
242
 
 
243
 
    runcmd("bzr add sub1")
244
 
    runcmd("bzr rename sub1 sub2")
245
 
    runcmd("bzr move hello.txt sub2")
246
 
    assert backtick("bzr relpath sub2/hello.txt") == "sub2/hello.txt\n"
247
 
 
248
 
    assert exists("sub2")
249
 
    assert exists("sub2/hello.txt")
250
 
    assert not exists("sub1")
251
 
    assert not exists("hello.txt")
252
 
 
253
 
    runcmd(['bzr', 'commit', '-m', 'commit with some things moved to subdirs'])
254
 
 
255
 
    mkdir("sub1")
256
 
    runcmd('bzr add sub1')
257
 
    runcmd('bzr move sub2/hello.txt sub1')
258
 
    assert not exists('sub2/hello.txt')
259
 
    assert exists('sub1/hello.txt')
260
 
    runcmd('bzr move sub2 sub1')
261
 
    assert not exists('sub2')
262
 
    assert exists('sub1/sub2')
263
 
 
264
 
    runcmd(['bzr', 'commit', '-m', 'rename nested subdirectories'])
265
 
 
266
 
    cd('sub1/sub2')
267
 
    assert backtick('bzr root')[:-1] == os.path.join(test_root, 'branch1')
268
 
    runcmd('bzr move ../hello.txt .')
269
 
    assert exists('./hello.txt')
270
 
    assert backtick('bzr relpath hello.txt') == 'sub1/sub2/hello.txt\n'
271
 
    assert backtick('bzr relpath ../../sub1/sub2/hello.txt') == 'sub1/sub2/hello.txt\n'
272
 
    runcmd(['bzr', 'commit', '-m', 'move to parent directory'])
273
 
    cd('..')
274
 
    assert backtick('bzr relpath sub2/hello.txt') == 'sub1/sub2/hello.txt\n'
275
 
 
276
 
    runcmd('bzr move sub2/hello.txt .')
277
 
    assert exists('hello.txt')
278
 
 
279
 
    f = file('hello.txt', 'wt')
280
 
    f.write('some nice new content\n')
281
 
    f.close()
282
 
 
283
 
    f = file('msg.tmp', 'wt')
284
 
    f.write('this is my new commit\n')
285
 
    f.close()
286
 
 
287
 
    runcmd('bzr commit -F msg.tmp')
288
 
 
289
 
    assert backtick('bzr revno') == '5\n'
290
 
    runcmd('bzr export -r 5 export-5.tmp')
291
 
    runcmd('bzr export export.tmp')
292
 
 
293
 
    runcmd('bzr log')
294
 
    runcmd('bzr log -v')
295
 
    
296
 
    cd('..')
297
 
    cd('..')
298
 
 
299
 
    progress('ignore patterns')
300
 
    mkdir('ignorebranch')
301
 
    cd('ignorebranch')
302
 
    runcmd('bzr init')
303
 
    assert backtick('bzr unknowns') == ''
304
 
 
305
 
    file('foo.tmp', 'wt').write('tmp files are ignored')
306
 
    assert backtick('bzr unknowns') == ''
307
 
 
308
 
    file('foo.c', 'wt').write('int main() {}')
309
 
    assert backtick('bzr unknowns') == 'foo.c\n'
310
 
    runcmd('bzr add foo.c')
311
 
    assert backtick('bzr unknowns') == ''
312
 
 
313
 
    file('foo.blah', 'wt').write('blah')
314
 
    assert backtick('bzr unknowns') == 'foo.blah\n'
315
 
    runcmd('bzr ignore *.blah')
316
 
    assert backtick('bzr unknowns') == ''
317
 
    assert file('.bzrignore', 'rt').read() == '*.blah\n'
318
 
 
319
 
 
320
 
    progress("all tests passed!")
321
 
except Exception, e:
322
 
    sys.stderr.write('*' * 50 + '\n'
323
 
                     + 'testbzr: tests failed\n'
324
 
                     + 'see ' + LOGFILENAME + ' for more information\n'
325
 
                     + '*' * 50 + '\n')
326
 
    logfile.write('tests failed!\n')
327
 
    traceback.print_exc(None, logfile)
328
 
    sys.exit(1)
 
20
print 'please use "bzr selftest" instead'
 
21
import sys
 
22
sys.exit(1)
 
23