From ed280d5ac360e2af796e9bd973d7b4df89f0c449 Mon Sep 17 00:00:00 2001 From: "Jeremy T. Bouse" Date: Fri, 27 Nov 2009 16:20:12 -0500 Subject: Imported Upstream version 1.7.4 --- test.py | 163 +++++++++++++++++++++++++++++++++++++++++----------------------- 1 file changed, 104 insertions(+), 59 deletions(-) mode change 100644 => 100755 test.py (limited to 'test.py') diff --git a/test.py b/test.py old mode 100644 new mode 100755 index 17b6294..b7f6c8a --- a/test.py +++ b/test.py @@ -1,6 +1,6 @@ -#!/usr/bin/python +#!/usr/bin/env python -# Copyright (C) 2003-2005 Robey Pointer +# Copyright (C) 2003-2007 Robey Pointer # # This file is part of paramiko. # @@ -22,82 +22,127 @@ do the unit tests! """ -import sys, os, unittest +import os +import re +import sys +import unittest from optparse import OptionParser import paramiko -sys.path.append('tests/') +sys.path.append('tests') from test_message import MessageTest from test_file import BufferedFileTest +from test_buffered_pipe import BufferedPipeTest from test_util import UtilTest +from test_hostkeys import HostKeysTest from test_pkey import KeyTest from test_kex import KexTest from test_packetizer import PacketizerTest +from test_auth import AuthTest from test_transport import TransportTest from test_sftp import SFTPTest +from test_sftp_big import BigSFTPTest +from test_client import SSHClientTest default_host = 'localhost' default_user = os.environ.get('USER', 'nobody') default_keyfile = os.path.join(os.environ.get('HOME', '/'), '.ssh/id_rsa') default_passwd = None -parser = OptionParser('usage: %prog [options]') -parser.add_option('--verbose', action='store_true', dest='verbose', default=False, - help='verbose display (one line per test)') -parser.add_option('--no-pkey', action='store_false', dest='use_pkey', default=True, - help='skip RSA/DSS private key tests (which can take a while)') -parser.add_option('--no-transport', action='store_false', dest='use_transport', default=True, - help='skip transport tests (which can take a while)') -parser.add_option('--no-sftp', action='store_false', dest='use_sftp', default=True, - help='skip SFTP client/server tests, which can be slow') -parser.add_option('--no-big-file', action='store_false', dest='use_big_file', default=True, - help='skip big file SFTP tests, which are slow as molasses') -parser.add_option('-R', action='store_false', dest='use_loopback_sftp', default=True, - help='perform SFTP tests against a remote server (by default, SFTP tests ' + - 'are done through a loopback socket)') -parser.add_option('-H', '--sftp-host', dest='hostname', type='string', default=default_host, - metavar='', - help='[with -R] host for remote sftp tests (default: %s)' % default_host) -parser.add_option('-U', '--sftp-user', dest='username', type='string', default=default_user, - metavar='', - help='[with -R] username for remote sftp tests (default: %s)' % default_user) -parser.add_option('-K', '--sftp-key', dest='keyfile', type='string', default=default_keyfile, - metavar='', - help='[with -R] location of private key for remote sftp tests (default: %s)' % - default_keyfile) -parser.add_option('-P', '--sftp-passwd', dest='password', type='string', default=default_passwd, - metavar='', - help='[with -R] (optional) password to unlock the private key for remote sftp tests') -options, args = parser.parse_args() -if len(args) > 0: - parser.error('unknown argument(s)') +def iter_suite_tests(suite): + """Return all tests in a suite, recursing through nested suites""" + for item in suite._tests: + if isinstance(item, unittest.TestCase): + yield item + elif isinstance(item, unittest.TestSuite): + for r in iter_suite_tests(item): + yield r + else: + raise Exception('unknown object %r inside test suite %r' + % (item, suite)) -# setup logging -paramiko.util.log_to_file('test.log') -if options.use_sftp: - if options.use_loopback_sftp: - SFTPTest.init_loopback() - else: - SFTPTest.init(options.hostname, options.username, options.keyfile, options.password) - if not options.use_big_file: - SFTPTest.set_big_file_test(False) +def filter_suite_by_re(suite, pattern): + result = unittest.TestSuite() + filter_re = re.compile(pattern) + for test in iter_suite_tests(suite): + if filter_re.search(test.id()): + result.addTest(test) + return result -suite = unittest.TestSuite() -suite.addTest(unittest.makeSuite(MessageTest)) -suite.addTest(unittest.makeSuite(BufferedFileTest)) -suite.addTest(unittest.makeSuite(UtilTest)) -if options.use_pkey: - suite.addTest(unittest.makeSuite(KeyTest)) -suite.addTest(unittest.makeSuite(KexTest)) -suite.addTest(unittest.makeSuite(PacketizerTest)) -if options.use_transport: - suite.addTest(unittest.makeSuite(TransportTest)) -if options.use_sftp: - suite.addTest(unittest.makeSuite(SFTPTest)) -verbosity = 1 -if options.verbose: - verbosity = 2 -unittest.TextTestRunner(verbosity=verbosity).run(suite) + +def main(): + parser = OptionParser('usage: %prog [options]') + parser.add_option('--verbose', action='store_true', dest='verbose', default=False, + help='verbose display (one line per test)') + parser.add_option('--no-pkey', action='store_false', dest='use_pkey', default=True, + help='skip RSA/DSS private key tests (which can take a while)') + parser.add_option('--no-transport', action='store_false', dest='use_transport', default=True, + help='skip transport tests (which can take a while)') + parser.add_option('--no-sftp', action='store_false', dest='use_sftp', default=True, + help='skip SFTP client/server tests, which can be slow') + parser.add_option('--no-big-file', action='store_false', dest='use_big_file', default=True, + help='skip big file SFTP tests, which are slow as molasses') + parser.add_option('-R', action='store_false', dest='use_loopback_sftp', default=True, + help='perform SFTP tests against a remote server (by default, SFTP tests ' + + 'are done through a loopback socket)') + parser.add_option('-H', '--sftp-host', dest='hostname', type='string', default=default_host, + metavar='', + help='[with -R] host for remote sftp tests (default: %s)' % default_host) + parser.add_option('-U', '--sftp-user', dest='username', type='string', default=default_user, + metavar='', + help='[with -R] username for remote sftp tests (default: %s)' % default_user) + parser.add_option('-K', '--sftp-key', dest='keyfile', type='string', default=default_keyfile, + metavar='', + help='[with -R] location of private key for remote sftp tests (default: %s)' % + default_keyfile) + parser.add_option('-P', '--sftp-passwd', dest='password', type='string', default=default_passwd, + metavar='', + help='[with -R] (optional) password to unlock the private key for remote sftp tests') + + options, args = parser.parse_args() + + # setup logging + paramiko.util.log_to_file('test.log') + + if options.use_sftp: + if options.use_loopback_sftp: + SFTPTest.init_loopback() + else: + SFTPTest.init(options.hostname, options.username, options.keyfile, options.password) + if not options.use_big_file: + SFTPTest.set_big_file_test(False) + + suite = unittest.TestSuite() + suite.addTest(unittest.makeSuite(MessageTest)) + suite.addTest(unittest.makeSuite(BufferedFileTest)) + suite.addTest(unittest.makeSuite(BufferedPipeTest)) + suite.addTest(unittest.makeSuite(UtilTest)) + suite.addTest(unittest.makeSuite(HostKeysTest)) + if options.use_pkey: + suite.addTest(unittest.makeSuite(KeyTest)) + suite.addTest(unittest.makeSuite(KexTest)) + suite.addTest(unittest.makeSuite(PacketizerTest)) + if options.use_transport: + suite.addTest(unittest.makeSuite(AuthTest)) + suite.addTest(unittest.makeSuite(TransportTest)) + suite.addTest(unittest.makeSuite(SSHClientTest)) + if options.use_sftp: + suite.addTest(unittest.makeSuite(SFTPTest)) + if options.use_big_file: + suite.addTest(unittest.makeSuite(BigSFTPTest)) + verbosity = 1 + if options.verbose: + verbosity = 2 + + runner = unittest.TextTestRunner(verbosity=verbosity) + if len(args) > 0: + filter = '|'.join(args) + suite = filter_suite_by_re(suite, filter) + runner.run(suite) + + +if __name__ == '__main__': + main() -- cgit v1.2.3