~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to testbzr

  • Committer: John Arbash Meinel
  • Date: 2006-07-21 19:49:48 UTC
  • mto: This revision was merged to the branch mainline in revision 1882.
  • Revision ID: john@arbash-meinel.com-20060721194948-5378854eeb7d2cad
some test cleanups

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
#! /usr/bin/python
2
 
 
3
 
# Copyright (C) 2005 Canonical Ltd
4
 
 
5
 
# This program is free software; you can redistribute it and/or modify
6
 
# it under the terms of the GNU General Public License as published by
7
 
# the Free Software Foundation; either version 2 of the License, or
8
 
# (at your option) any later version.
9
 
 
10
 
# This program is distributed in the hope that it will be useful,
11
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
 
# GNU General Public License for more details.
14
 
 
15
 
# You should have received a copy of the GNU General Public License
16
 
# along with this program; if not, write to the Free Software
17
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
 
 
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
 
 
29
 
try:
30
 
    import shutil
31
 
    from subprocess import call, Popen, PIPE
32
 
except ImportError, e:
33
 
    sys.stderr.write("testbzr: sorry, this test suite requires modules from python2.4\n"
34
 
                     + '    ' + str(e))
35
 
    sys.exit(1)
36
 
 
37
 
 
38
 
class CommandFailed(Exception):
39
 
    pass
40
 
 
41
 
 
42
 
def formcmd(cmd):
43
 
    if isinstance(cmd, basestring):
44
 
        logfile.write('$ %s\n' % cmd)
45
 
        cmd = cmd.split()
46
 
    else:
47
 
        logfile.write('$ %r\n' % cmd)
48
 
 
49
 
    return cmd
50
 
 
51
 
 
52
 
def runcmd(cmd, retcode=0):
53
 
    """Run one command and check the return code.
54
 
 
55
 
    Returns a tuple of (stdout,stderr) strings.
56
 
 
57
 
    If a single string is based, it is split into words.
58
 
    For commands that are not simple space-separated words, please
59
 
    pass a list instead."""
60
 
    cmd = formcmd(cmd)
61
 
    log_linenumber()
62
 
    
63
 
    actual_retcode = call(cmd, stdout=logfile, stderr=logfile)
64
 
    
65
 
    if retcode != actual_retcode:
66
 
        raise CommandFailed("test failed: %r returned %d, expected %d"
67
 
                            % (cmd, actual_retcode, retcode))
68
 
 
69
 
 
70
 
 
71
 
def backtick(cmd, retcode=0):
72
 
    cmd = formcmd(cmd)
73
 
    log_linenumber()
74
 
    child = Popen(cmd, stdout=PIPE, stderr=logfile)
75
 
    outd, errd = child.communicate()
76
 
    logfile.write(outd)
77
 
    actual_retcode = child.wait()
78
 
 
79
 
    outd = outd.replace('\r', '')
80
 
    
81
 
    if retcode != actual_retcode:
82
 
        raise CommandFailed("test failed: %r returned %d, expected %d"
83
 
                            % (cmd, actual_retcode, retcode))
84
 
 
85
 
    return outd
86
 
 
87
 
 
88
 
 
89
 
def progress(msg):
90
 
    print '* ' + msg
91
 
    logfile.write('* '+ msg + '\n')
92
 
    log_linenumber()
93
 
 
94
 
 
95
 
def cd(dirname):
96
 
    logfile.write('$ cd %s\n' % dirname)
97
 
    os.chdir(dirname)
98
 
 
99
 
 
100
 
 
101
 
def log_linenumber():
102
 
    """Log the stack frame location two things up."""
103
 
    stack = traceback.extract_stack()[-3]
104
 
    logfile.write('   at %s:%d\n' % stack[:2])
105
 
 
106
 
 
107
 
TESTDIR = "testbzr.tmp"
108
 
 
109
 
# prepare an empty scratch directory
110
 
if os.path.exists(TESTDIR):
111
 
    shutil.rmtree(TESTDIR)
112
 
 
113
 
 
114
 
logfile = open('testbzr.log', 'wt', buffering=1)
115
 
 
116
 
 
117
 
try:
118
 
    runcmd(['mkdir', TESTDIR])
119
 
    cd(TESTDIR)
120
 
 
121
 
    progress("introductory commands")
122
 
    runcmd("bzr version")
123
 
    runcmd("bzr help")
124
 
    runcmd("bzr --help")
125
 
 
126
 
    progress("user identity")
127
 
    # this should always identify something, if only "john@localhost"
128
 
    runcmd("bzr whoami")
129
 
    runcmd("bzr whoami --email")
130
 
    assert backtick("bzr whoami --email").count('@') == 1
131
 
 
132
 
    progress("invalid commands")
133
 
    runcmd("bzr pants", retcode=1)
134
 
    runcmd("bzr --pants off", retcode=1)
135
 
 
136
 
    progress("basic branch creation")
137
 
    runcmd(['mkdir', 'branch1'])
138
 
    cd('branch1')
139
 
    runcmd('bzr init')
140
 
 
141
 
    progress("status of new file")
142
 
    
143
 
    f = file('test.txt', 'wt')
144
 
    f.write('hello world!\n')
145
 
    f.close()
146
 
 
147
 
    out = backtick("bzr unknowns")
148
 
    assert out == 'test.txt\n'
149
 
 
150
 
    out = backtick("bzr status")
151
 
    assert out == '''?       test.txt\n'''
152
 
 
153
 
    out = backtick("bzr status --all")
154
 
    assert out == "?       test.txt\n"
155
 
 
156
 
    progress("can't rename unversioned file")
157
 
    runcmd("bzr rename test.txt new-test.txt", 1)
158
 
 
159
 
    progress("adding a file")
160
 
 
161
 
    runcmd("bzr add test.txt")
162
 
    assert backtick("bzr unknowns") == ''
163
 
    assert backtick("bzr status --all") == "A       test.txt\n"
164
 
 
165
 
    progress("rename newly-added file")
166
 
    runcmd("bzr rename test.txt hello.txt")
167
 
    assert os.path.exists("hello.txt")
168
 
    assert not os.path.exists("test.txt")
169
 
 
170
 
    assert backtick("bzr revno") == '0\n'
171
 
 
172
 
    cd('..')
173
 
 
174
 
    progress("all tests passed!")
175
 
except Exception, e:
176
 
    sys.stderr.write('*' * 50 + '\n'
177
 
                     + 'testbzr: tests failed\n'
178
 
                     + 'see testbzr.log for more information\n'
179
 
                     + '*' * 50 + '\n')
180
 
    logfile.write('tests failed!\n')
181
 
    traceback.print_exc(None, logfile)
182
 
    sys.exit(1)