~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to testbzr

  • Committer: Martin Pool
  • Date: 2005-05-15 02:36:04 UTC
  • Revision ID: mbp@sourcefrog.net-20050515023603-7328f6cbabd2b09a
- Merge aaron's merge command

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
1
#! /usr/bin/python
2
 
# -*- coding: utf-8 -*-
3
2
 
4
3
# Copyright (C) 2005 Canonical Ltd
5
4
 
17
16
# along with this program; if not, write to the Free Software
18
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19
18
 
20
 
print 'please use "bzr selftest" instead'
21
 
import sys
22
 
sys.exit(1)
23
 
 
 
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 == 'unknown:\n  test.txt\n'
 
183
 
 
184
    out = backtick("bzr status --all")
 
185
    assert out == "unknown:\n  test.txt\n"
 
186
 
 
187
    out = backtick("bzr status test.txt --all")
 
188
    assert out == "unknown:\n  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 == "unknown:\n  test.txt\n"
 
196
 
 
197
    out = backtick("bzr status")
 
198
    assert out == ("unknown:\n"
 
199
                   "  test.txt\n"
 
200
                   "  test2.txt\n")
 
201
 
 
202
    os.unlink('test2.txt')
 
203
 
 
204
    progress("command aliases")
 
205
    out = backtick("bzr st --all")
 
206
    assert out == ("unknown:\n"
 
207
                   "  test.txt\n")
 
208
    
 
209
    out = backtick("bzr stat")
 
210
    assert out == ("unknown:\n"
 
211
                   "  test.txt\n")
 
212
 
 
213
    progress("command help")
 
214
    runcmd("bzr help st")
 
215
    runcmd("bzr help")
 
216
    runcmd("bzr help commands")
 
217
    runcmd("bzr help slartibartfast", 1)
 
218
 
 
219
    out = backtick("bzr help ci")
 
220
    out.index('aliases: ')
 
221
 
 
222
    progress("can't rename unversioned file")
 
223
    runcmd("bzr rename test.txt new-test.txt", 1)
 
224
 
 
225
    progress("adding a file")
 
226
 
 
227
    runcmd("bzr add test.txt")
 
228
    assert backtick("bzr unknowns") == ''
 
229
    assert backtick("bzr status --all") == ("added:\n"
 
230
                                            "  test.txt\n")
 
231
 
 
232
    progress("rename newly-added file")
 
233
    runcmd("bzr rename test.txt hello.txt")
 
234
    assert os.path.exists("hello.txt")
 
235
    assert not os.path.exists("test.txt")
 
236
 
 
237
    assert backtick("bzr revno") == '0\n'
 
238
 
 
239
    progress("add first revision")
 
240
    runcmd(["bzr", "commit", "-m", 'add first revision'])
 
241
 
 
242
    progress("more complex renames")
 
243
    os.mkdir("sub1")
 
244
    runcmd("bzr rename hello.txt sub1", 1)
 
245
    runcmd("bzr rename hello.txt sub1/hello.txt", 1)
 
246
    runcmd("bzr move hello.txt sub1", 1)
 
247
 
 
248
    runcmd("bzr add sub1")
 
249
    runcmd("bzr rename sub1 sub2")
 
250
    runcmd("bzr move hello.txt sub2")
 
251
    assert backtick("bzr relpath sub2/hello.txt") == "sub2/hello.txt\n"
 
252
 
 
253
    assert exists("sub2")
 
254
    assert exists("sub2/hello.txt")
 
255
    assert not exists("sub1")
 
256
    assert not exists("hello.txt")
 
257
 
 
258
    runcmd(['bzr', 'commit', '-m', 'commit with some things moved to subdirs'])
 
259
 
 
260
    mkdir("sub1")
 
261
    runcmd('bzr add sub1')
 
262
    runcmd('bzr move sub2/hello.txt sub1')
 
263
    assert not exists('sub2/hello.txt')
 
264
    assert exists('sub1/hello.txt')
 
265
    runcmd('bzr move sub2 sub1')
 
266
    assert not exists('sub2')
 
267
    assert exists('sub1/sub2')
 
268
 
 
269
    runcmd(['bzr', 'commit', '-m', 'rename nested subdirectories'])
 
270
 
 
271
    cd('sub1/sub2')
 
272
    assert backtick('bzr root')[:-1] == os.path.join(test_root, 'branch1')
 
273
    runcmd('bzr move ../hello.txt .')
 
274
    assert exists('./hello.txt')
 
275
    assert backtick('bzr relpath hello.txt') == 'sub1/sub2/hello.txt\n'
 
276
    assert backtick('bzr relpath ../../sub1/sub2/hello.txt') == 'sub1/sub2/hello.txt\n'
 
277
    runcmd(['bzr', 'commit', '-m', 'move to parent directory'])
 
278
    cd('..')
 
279
    assert backtick('bzr relpath sub2/hello.txt') == 'sub1/sub2/hello.txt\n'
 
280
 
 
281
    runcmd('bzr move sub2/hello.txt .')
 
282
    assert exists('hello.txt')
 
283
 
 
284
    f = file('hello.txt', 'wt')
 
285
    f.write('some nice new content\n')
 
286
    f.close()
 
287
 
 
288
    f = file('msg.tmp', 'wt')
 
289
    f.write('this is my new commit\n')
 
290
    f.close()
 
291
 
 
292
    runcmd('bzr commit -F msg.tmp')
 
293
 
 
294
    assert backtick('bzr revno') == '5\n'
 
295
    runcmd('bzr export -r 5 export-5.tmp')
 
296
    runcmd('bzr export export.tmp')
 
297
 
 
298
    runcmd('bzr log')
 
299
    runcmd('bzr log -v')
 
300
    
 
301
    cd('..')
 
302
    cd('..')
 
303
 
 
304
    progress('ignore patterns')
 
305
    mkdir('ignorebranch')
 
306
    cd('ignorebranch')
 
307
    runcmd('bzr init')
 
308
    assert backtick('bzr unknowns') == ''
 
309
 
 
310
    file('foo.tmp', 'wt').write('tmp files are ignored')
 
311
    assert backtick('bzr unknowns') == ''
 
312
 
 
313
    file('foo.c', 'wt').write('int main() {}')
 
314
    assert backtick('bzr unknowns') == 'foo.c\n'
 
315
    runcmd('bzr add foo.c')
 
316
    assert backtick('bzr unknowns') == ''
 
317
 
 
318
    file('foo.blah', 'wt').write('blah')
 
319
    assert backtick('bzr unknowns') == 'foo.blah\n'
 
320
    runcmd('bzr ignore *.blah')
 
321
    assert backtick('bzr unknowns') == ''
 
322
    assert file('.bzrignore', 'rt').read() == '*.blah\n'
 
323
 
 
324
 
 
325
    progress("all tests passed!")
 
326
except Exception, e:
 
327
    sys.stderr.write('*' * 50 + '\n'
 
328
                     + 'testbzr: tests failed\n'
 
329
                     + 'see ' + LOGFILENAME + ' for more information\n'
 
330
                     + '*' * 50 + '\n')
 
331
    logfile.write('tests failed!\n')
 
332
    traceback.print_exc(None, logfile)
 
333
    sys.exit(1)