Don't import test_sftp or test_sftp_big unless we are going to do the tests

This commit is contained in:
Scott Maxwell 2013-11-01 01:06:17 -07:00
parent 201a61d66d
commit 06b866cf40
1 changed files with 4 additions and 3 deletions

View File

@ -42,8 +42,6 @@ from tests.test_kex import KexTest
from tests.test_packetizer import PacketizerTest from tests.test_packetizer import PacketizerTest
from tests.test_auth import AuthTest from tests.test_auth import AuthTest
from tests.test_transport import TransportTest from tests.test_transport import TransportTest
from tests.test_sftp import SFTPTest
from tests.test_sftp_big import BigSFTPTest
from tests.test_client import SSHClientTest from tests.test_client import SSHClientTest
default_host = 'localhost' default_host = 'localhost'
@ -109,12 +107,15 @@ def main():
paramiko.util.log_to_file('test.log') paramiko.util.log_to_file('test.log')
if options.use_sftp: if options.use_sftp:
from tests.test_sftp import SFTPTest
if options.use_loopback_sftp: if options.use_loopback_sftp:
SFTPTest.init_loopback() SFTPTest.init_loopback()
else: else:
SFTPTest.init(options.hostname, options.username, options.keyfile, options.password) SFTPTest.init(options.hostname, options.username, options.keyfile, options.password)
if not options.use_big_file: if not options.use_big_file:
SFTPTest.set_big_file_test(False) SFTPTest.set_big_file_test(False)
if options.use_big_file:
from tests.test_sftp_big import BigSFTPTest
suite = unittest.TestSuite() suite = unittest.TestSuite()
suite.addTest(unittest.makeSuite(MessageTest)) suite.addTest(unittest.makeSuite(MessageTest))