~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to testbzr

  • Committer: Jelmer Vernooij
  • Date: 2006-06-21 13:54:14 UTC
  • mto: (1558.14.8 Aaron's integration)
  • mto: This revision was merged to the branch mainline in revision 1803.
  • Revision ID: jelmer@samba.org-20060621135414-11a3a70e53adbb99
Install benchmarks.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
#! /usr/bin/python
2
 
# -*- coding: utf-8 -*-
3
 
 
4
 
# Copyright (C) 2005 Canonical Ltd
5
 
 
6
 
# This program is free software; you can redistribute it and/or modify
7
 
# it under the terms of the GNU General Public License as published by
8
 
# the Free Software Foundation; either version 2 of the License, or
9
 
# (at your option) any later version.
10
 
 
11
 
# This program is distributed in the hope that it will be useful,
12
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
13
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14
 
# GNU General Public License for more details.
15
 
 
16
 
# You should have received a copy of the GNU General Public License
17
 
# along with this program; if not, write to the Free Software
18
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19
 
 
20
 
 
21
 
"""External black-box test for bzr.
22
 
 
23
 
This always runs bzr as an external process to try to catch bugs
24
 
related to argument processing, startup, etc.
25
 
 
26
 
usage:
27
 
 
28
 
    testbzr [-p PYTHON] [BZR]
29
 
 
30
 
By default this tests the copy of bzr found in the same directory as
31
 
testbzr, or the first one found on the $PATH.  A copy of bzr may be
32
 
given on the command line to override this, for example when applying
33
 
a new test suite to an old copy of bzr or vice versa.
34
 
 
35
 
testbzr normally invokes bzr using the same version of python as it
36
 
would normally use to run -- that is, the system default python,
37
 
unless that is older than 2.3.  The -p option allows specification of
38
 
a different Python interpreter, such as when testing that bzr still
39
 
works on python2.3.
40
 
 
41
 
This replaces the previous test.sh which was not very portable."""
42
 
 
43
 
import sys, os, traceback
44
 
from os import mkdir
45
 
from os.path import exists
46
 
 
47
 
TESTDIR = "testbzr.tmp"
48
 
 
49
 
 
50
 
# we always invoke bzr as 'python bzr' (or e.g. 'python2.3 bzr')
51
 
# partly so as to cope if the bzr binary is not marked executable
52
 
OVERRIDE_PYTHON = 'python'
53
 
 
54
 
LOGFILENAME = 'testbzr.log'
55
 
 
56
 
try:
57
 
    import shutil
58
 
    from subprocess import call, Popen, PIPE
59
 
except ImportError, e:
60
 
    sys.stderr.write("testbzr: sorry, this test suite requires modules from python2.4\n"
61
 
                     + '    ' + str(e))
62
 
    sys.exit(1)
63
 
 
64
 
 
65
 
class CommandFailed(Exception):
66
 
    pass
67
 
 
68
 
 
69
 
def formcmd(cmd):
70
 
    if isinstance(cmd, basestring):
71
 
        cmd = cmd.split()
72
 
 
73
 
    if cmd[0] == 'bzr':
74
 
        cmd[0] = BZRPATH
75
 
        if OVERRIDE_PYTHON:
76
 
            cmd.insert(0, OVERRIDE_PYTHON)
77
 
 
78
 
    logfile.write('$ %r\n' % cmd)
79
 
    
80
 
    return cmd
81
 
 
82
 
 
83
 
def runcmd(cmd, retcode=0):
84
 
    """Run one command and check the return code.
85
 
 
86
 
    Returns a tuple of (stdout,stderr) strings.
87
 
 
88
 
    If a single string is based, it is split into words.
89
 
    For commands that are not simple space-separated words, please
90
 
    pass a list instead."""
91
 
    cmd = formcmd(cmd)
92
 
    log_linenumber()
93
 
    
94
 
    actual_retcode = call(cmd, stdout=logfile, stderr=logfile)
95
 
    
96
 
    if retcode != actual_retcode:
97
 
        raise CommandFailed("test failed: %r returned %d, expected %d"
98
 
                            % (cmd, actual_retcode, retcode))
99
 
 
100
 
 
101
 
 
102
 
def backtick(cmd, retcode=0):
103
 
    cmd = formcmd(cmd)
104
 
    log_linenumber()
105
 
    child = Popen(cmd, stdout=PIPE, stderr=logfile)
106
 
    outd, errd = child.communicate()
107
 
    logfile.write(outd)
108
 
    actual_retcode = child.wait()
109
 
 
110
 
    outd = outd.replace('\r', '')
111
 
    
112
 
    if retcode != actual_retcode:
113
 
        raise CommandFailed("test failed: %r returned %d, expected %d"
114
 
                            % (cmd, actual_retcode, retcode))
115
 
 
116
 
    return outd
117
 
 
118
 
 
119
 
 
120
 
def progress(msg):
121
 
    print '* ' + msg
122
 
    logfile.write('* '+ msg + '\n')
123
 
    log_linenumber()
124
 
 
125
 
 
126
 
def cd(dirname):
127
 
    logfile.write('$ cd %s\n' % dirname)
128
 
    os.chdir(dirname)
129
 
 
130
 
 
131
 
 
132
 
def log_linenumber():
133
 
    """Log the stack frame location two things up."""
134
 
    stack = traceback.extract_stack()[-3]
135
 
    logfile.write('   at %s:%d\n' % stack[:2])
136
 
 
137
 
 
138
 
 
139
 
# prepare an empty scratch directory
140
 
if os.path.exists(TESTDIR):
141
 
    shutil.rmtree(TESTDIR)
142
 
 
143
 
start_dir = os.getcwd()
144
 
 
145
 
 
146
 
logfile = open(LOGFILENAME, 'wt', buffering=1)
147
 
 
148
 
 
149
 
try:
150
 
    from getopt import getopt
151
 
    opts, args = getopt(sys.argv[1:], 'p:')
152
 
 
153
 
    for option, value in opts:
154
 
        if option == '-p':
155
 
            OVERRIDE_PYTHON = value
156
 
            
157
 
    
158
 
    mypath = os.path.abspath(sys.argv[0])
159
 
    print '%-30s %s' % ('running tests from', mypath)
160
 
 
161
 
    global BZRPATH
162
 
 
163
 
    if args:
164
 
        BZRPATH = args[0]
165
 
    else:
166
 
        BZRPATH = os.path.join(os.path.split(mypath)[0], 'bzr')
167
 
 
168
 
    print '%-30s %s' % ('against bzr', BZRPATH)
169
 
    print '%-30s %s' % ('in directory', os.getcwd())
170
 
    print '%-30s %s' % ('with python', (OVERRIDE_PYTHON or '(default)'))
171
 
    print
172
 
    print backtick('bzr version')
173
 
    
174
 
    runcmd(['mkdir', TESTDIR])
175
 
    cd(TESTDIR)
176
 
    test_root = os.getcwd()
177
 
 
178
 
    progress("introductory commands")
179
 
    runcmd("bzr version")
180
 
    runcmd("bzr --version")
181
 
    runcmd("bzr help")
182
 
    runcmd("bzr --help")
183
 
 
184
 
    progress("internal tests")
185
 
    runcmd("bzr selftest")
186
 
 
187
 
    progress("user identity")
188
 
    # this should always identify something, if only "john@localhost"
189
 
    runcmd("bzr whoami")
190
 
    runcmd("bzr whoami --email")
191
 
    assert backtick("bzr whoami --email").count('@') == 1
192
 
 
193
 
    progress("invalid commands")
194
 
    runcmd("bzr pants", retcode=1)
195
 
    runcmd("bzr --pants off", retcode=1)
196
 
    runcmd("bzr diff --message foo", retcode=1)
197
 
 
198
 
    progress("basic branch creation")
199
 
    runcmd(['mkdir', 'branch1'])
200
 
    cd('branch1')
201
 
    runcmd('bzr init')
202
 
 
203
 
    assert backtick('bzr root')[:-1] == os.path.join(test_root, 'branch1')
204
 
 
205
 
    progress("status of new file")
206
 
    
207
 
    f = file('test.txt', 'wt')
208
 
    f.write('hello world!\n')
209
 
    f.close()
210
 
 
211
 
    out = backtick("bzr unknowns")
212
 
    assert out == 'test.txt\n'
213
 
 
214
 
    out = backtick("bzr status")
215
 
    assert out == 'unknown:\n  test.txt\n'
216
 
 
217
 
    out = backtick("bzr status --all")
218
 
    assert out == "unknown:\n  test.txt\n"
219
 
 
220
 
    out = backtick("bzr status test.txt --all")
221
 
    assert out == "unknown:\n  test.txt\n"
222
 
 
223
 
    f = file('test2.txt', 'wt')
224
 
    f.write('goodbye cruel world...\n')
225
 
    f.close()
226
 
 
227
 
    out = backtick("bzr status test.txt")
228
 
    assert out == "unknown:\n  test.txt\n"
229
 
 
230
 
    out = backtick("bzr status")
231
 
    assert out == ("unknown:\n"
232
 
                   "  test.txt\n"
233
 
                   "  test2.txt\n")
234
 
 
235
 
    os.unlink('test2.txt')
236
 
 
237
 
    progress("command aliases")
238
 
    out = backtick("bzr st --all")
239
 
    assert out == ("unknown:\n"
240
 
                   "  test.txt\n")
241
 
    
242
 
    out = backtick("bzr stat")
243
 
    assert out == ("unknown:\n"
244
 
                   "  test.txt\n")
245
 
 
246
 
    progress("command help")
247
 
    runcmd("bzr help st")
248
 
    runcmd("bzr help")
249
 
    runcmd("bzr help commands")
250
 
    runcmd("bzr help slartibartfast", 1)
251
 
 
252
 
    out = backtick("bzr help ci")
253
 
    out.index('aliases: ')
254
 
 
255
 
    progress("can't rename unversioned file")
256
 
    runcmd("bzr rename test.txt new-test.txt", 1)
257
 
 
258
 
    progress("adding a file")
259
 
 
260
 
    runcmd("bzr add test.txt")
261
 
    assert backtick("bzr unknowns") == ''
262
 
    assert backtick("bzr status --all") == ("added:\n"
263
 
                                            "  test.txt\n")
264
 
 
265
 
    progress("rename newly-added file")
266
 
    runcmd("bzr rename test.txt hello.txt")
267
 
    assert os.path.exists("hello.txt")
268
 
    assert not os.path.exists("test.txt")
269
 
 
270
 
    assert backtick("bzr revno") == '0\n'
271
 
 
272
 
    progress("add first revision")
273
 
    runcmd(["bzr", "commit", "-m", 'add first revision'])
274
 
 
275
 
    progress("more complex renames")
276
 
    os.mkdir("sub1")
277
 
    runcmd("bzr rename hello.txt sub1", 1)
278
 
    runcmd("bzr rename hello.txt sub1/hello.txt", 1)
279
 
    runcmd("bzr move hello.txt sub1", 1)
280
 
 
281
 
    runcmd("bzr add sub1")
282
 
    runcmd("bzr rename sub1 sub2")
283
 
    runcmd("bzr move hello.txt sub2")
284
 
    assert backtick("bzr relpath sub2/hello.txt") == os.path.join("sub2", "hello.txt\n")
285
 
 
286
 
    assert exists("sub2")
287
 
    assert exists("sub2/hello.txt")
288
 
    assert not exists("sub1")
289
 
    assert not exists("hello.txt")
290
 
 
291
 
    runcmd(['bzr', 'commit', '-m', 'commit with some things moved to subdirs'])
292
 
 
293
 
    mkdir("sub1")
294
 
    runcmd('bzr add sub1')
295
 
    runcmd('bzr move sub2/hello.txt sub1')
296
 
    assert not exists('sub2/hello.txt')
297
 
    assert exists('sub1/hello.txt')
298
 
    runcmd('bzr move sub2 sub1')
299
 
    assert not exists('sub2')
300
 
    assert exists('sub1/sub2')
301
 
 
302
 
    runcmd(['bzr', 'commit', '-m', 'rename nested subdirectories'])
303
 
 
304
 
    cd('sub1/sub2')
305
 
    assert backtick('bzr root')[:-1] == os.path.join(test_root, 'branch1')
306
 
    runcmd('bzr move ../hello.txt .')
307
 
    assert exists('./hello.txt')
308
 
    assert backtick('bzr relpath hello.txt') == os.path.join('sub1', 'sub2', 'hello.txt\n')
309
 
    assert backtick('bzr relpath ../../sub1/sub2/hello.txt') == os.path.join('sub1', 'sub2', 'hello.txt\n')
310
 
    runcmd(['bzr', 'commit', '-m', 'move to parent directory'])
311
 
    cd('..')
312
 
    assert backtick('bzr relpath sub2/hello.txt') == os.path.join('sub1', 'sub2', 'hello.txt\n')
313
 
 
314
 
    runcmd('bzr move sub2/hello.txt .')
315
 
    assert exists('hello.txt')
316
 
 
317
 
    f = file('hello.txt', 'wt')
318
 
    f.write('some nice new content\n')
319
 
    f.close()
320
 
 
321
 
    f = file('msg.tmp', 'wt')
322
 
    f.write('this is my new commit\n')
323
 
    f.close()
324
 
 
325
 
    runcmd('bzr commit -F msg.tmp')
326
 
 
327
 
    assert backtick('bzr revno') == '5\n'
328
 
    runcmd('bzr export -r 5 export-5.tmp')
329
 
    runcmd('bzr export export.tmp')
330
 
 
331
 
    runcmd('bzr log')
332
 
    runcmd('bzr log -v')
333
 
 
334
 
 
335
 
 
336
 
    progress("file with spaces in name")
337
 
    mkdir('sub directory')
338
 
    file('sub directory/file with spaces ', 'wt').write('see how this works\n')
339
 
    runcmd('bzr add .')
340
 
    runcmd('bzr diff')
341
 
    runcmd('bzr commit -m add-spaces')
342
 
    runcmd('bzr check')
343
 
 
344
 
    runcmd('bzr log')
345
 
    runcmd('bzr log --forward')
346
 
 
347
 
    runcmd('bzr info')
348
 
 
349
 
 
350
 
    
351
 
 
352
 
 
353
 
 
354
 
    cd('..')
355
 
    cd('..')
356
 
    progress('branch')
357
 
    # Can't create a branch if it already exists
358
 
    runcmd('bzr branch branch1', retcode=1)
359
 
    # Can't create a branch if its parent doesn't exist
360
 
    runcmd('bzr branch /unlikely/to/exist', retcode=1)
361
 
    runcmd('bzr branch branch1 branch2')
362
 
 
363
 
    progress("pull")
364
 
    cd('branch1')
365
 
    runcmd('bzr pull', retcode=1)
366
 
    runcmd('bzr pull ../branch2')
367
 
    cd('.bzr')
368
 
    runcmd('bzr pull')
369
 
    runcmd('bzr commit -m empty')
370
 
    runcmd('bzr pull')
371
 
    cd('../../branch2')
372
 
    runcmd('bzr pull')
373
 
    runcmd('bzr commit -m empty')
374
 
    cd('../branch1')
375
 
    runcmd('bzr commit -m empty')
376
 
    runcmd('bzr pull', retcode=1)
377
 
    cd ('..')
378
 
 
379
 
    progress('status after remove')
380
 
    mkdir('status-after-remove')
381
 
    # see mail from William Dodé, 2005-05-25
382
 
    # $ bzr init; touch a; bzr add a; bzr commit -m "add a"
383
 
    #     * looking for changes...
384
 
    #     added a
385
 
    #     * commited r1
386
 
    #     $ bzr remove a
387
 
    #     $ bzr status
388
 
    #     bzr: local variable 'kind' referenced before assignment
389
 
    #     at /vrac/python/bazaar-ng/bzrlib/diff.py:286 in compare_trees()
390
 
    #     see ~/.bzr.log for debug information
391
 
    cd('status-after-remove')
392
 
    runcmd('bzr init')
393
 
    file('a', 'w').write('foo')
394
 
    runcmd('bzr add a')
395
 
    runcmd(['bzr', 'commit', '-m', 'add a'])
396
 
    runcmd('bzr remove a')
397
 
    runcmd('bzr status')
398
 
 
399
 
    cd('..')
400
 
 
401
 
    progress('ignore patterns')
402
 
    mkdir('ignorebranch')
403
 
    cd('ignorebranch')
404
 
    runcmd('bzr init')
405
 
    assert backtick('bzr unknowns') == ''
406
 
 
407
 
    file('foo.tmp', 'wt').write('tmp files are ignored')
408
 
    assert backtick('bzr unknowns') == ''
409
 
 
410
 
    file('foo.c', 'wt').write('int main() {}')
411
 
    assert backtick('bzr unknowns') == 'foo.c\n'
412
 
    runcmd('bzr add foo.c')
413
 
    assert backtick('bzr unknowns') == ''
414
 
 
415
 
    # 'ignore' works when creating the .bzignore file
416
 
    file('foo.blah', 'wt').write('blah')
417
 
    assert backtick('bzr unknowns') == 'foo.blah\n'
418
 
    runcmd('bzr ignore *.blah')
419
 
    assert backtick('bzr unknowns') == ''
420
 
    assert file('.bzrignore', 'rb').read() == '*.blah\n'
421
 
 
422
 
    # 'ignore' works when then .bzrignore file already exists
423
 
    file('garh', 'wt').write('garh')
424
 
    assert backtick('bzr unknowns') == 'garh\n'
425
 
    runcmd('bzr ignore garh')
426
 
    assert backtick('bzr unknowns') == ''
427
 
    assert file('.bzrignore', 'rb').read() == '*.blah\ngarh\n'
428
 
 
429
 
    cd('..')
430
 
 
431
 
 
432
 
 
433
 
 
434
 
    progress("recursive and non-recursive add")
435
 
    mkdir('no-recurse')
436
 
    cd('no-recurse')
437
 
    runcmd('bzr init')
438
 
    mkdir('foo')
439
 
    fp = os.path.join('foo', 'test.txt')
440
 
    f = file(fp, 'w')
441
 
    f.write('hello!\n')
442
 
    f.close()
443
 
    runcmd('bzr add --no-recurse foo')
444
 
    runcmd('bzr file-id foo')
445
 
    runcmd('bzr file-id ' + fp, 1)      # not versioned yet
446
 
    runcmd('bzr commit -m add-dir-only')
447
 
 
448
 
    runcmd('bzr file-id ' + fp, 1)      # still not versioned 
449
 
 
450
 
    runcmd('bzr add foo')
451
 
    runcmd('bzr file-id ' + fp)
452
 
    runcmd('bzr commit -m add-sub-file')
453
 
    
454
 
    cd('..')
455
 
 
456
 
 
457
 
 
458
 
 
459
 
    progress("all tests passed!")
460
 
except Exception, e:
461
 
    sys.stderr.write('*' * 50 + '\n'
462
 
                     + 'testbzr: tests failed\n'
463
 
                     + 'see ' + LOGFILENAME + ' for more information\n'
464
 
                     + '*' * 50 + '\n')
465
 
    logfile.write('tests failed!\n')
466
 
    traceback.print_exc(None, logfile)
467
 
    logfile.close()
468
 
 
469
 
    sys.stdout.writelines(file(os.path.join(start_dir, LOGFILENAME), 'rt').readlines()[-50:])
470
 
    
471
 
    sys.exit(1)
472