Imported Upstream version 1.7.4upstream/1.7.4

1 files changed, 104 insertions, 59 deletions

diff --git a/test.py b/test.py
index 17b6294..b7f6c8a 100644..100755
--- a/test.py
+++ b/test.py
@@ -1,6 +1,6 @@
-#!/usr/bin/python
+#!/usr/bin/env python
 
-# Copyright (C) 2003-2005 Robey Pointer <robey@lag.net>
+# Copyright (C) 2003-2007  Robey Pointer <robey@lag.net>
 #
 # 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='<host>',
-                  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='<username>',
-                  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='<keyfile>',
-                  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='<password>',
-                  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='<host>',
+                      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='<username>',
+                      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='<keyfile>',
+                      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='<password>',
+                      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()