parser.add_option('-v', '--verbose', action="store_const",
dest="verbose", const=2,
help="verbose output")
parser.add_option('-q', '--quiet', action="store_const",
dest="verbose", const=0,
help="minimal output")
parser.add_option('--protocol', dest="proto", type="string",
help="protocol to use, one of: accel, accelc, binary, compact, json, multi, multia, multiac, multic, multij")
parser.add_option('--transport', dest="trans", type="string",
help="transport to use, one of: buffered, framed, http")
parser.set_defaults(framed=False, http_path=None, verbose=1, host='localhost', port=9090, proto='binary')
options, args = parser.parse_args()
if options.genpydir:
sys.path.insert(0, os.path.join(SCRIPT_DIR, options.genpydir))
sys.path.insert(0, local_libpath())
if options.http_path:
options.trans = 'http'
from ThriftTest import SecondService
from ThriftTest import ThriftTest
from ThriftTest.ttypes import Xtruct, Xtruct2, Numberz, Xception, Xception2
from thrift.Thrift import TException
from thrift.transport import TTransport
from thrift.transport import TSocket
from thrift.transport import THttpClient
from thrift.transport import TZlibTransport
from thrift.protocol import TBinaryProtocol
from thrift.protocol import TCompactProtocol
from thrift.protocol import TJSONProtocol
def main():
parser = OptionParser()
parser.add_option('--all', action="store_true", dest='all')
parser.add_option(
'--genpydirs',
type='string',
dest='genpydirs',
default='default,slots,oldstyle,no_utf8strings,dynamic,dynamicslots',
help=
'directory extensions for generated code, used as suffixes for \"gen-py-*\" added sys.path for individual tests'
)
parser.add_option("--port",
type="int",
dest="port",
default=9090,
help="port number for server to listen on")
parser.add_option('-v',
'--verbose',
action="store_const",
dest="verbose",
const=2,
help="verbose output")
parser.add_option('-q',
'--quiet',
action="store_const",
dest="verbose",
const=0,
help="minimal output")
parser.add_option(
'-L',
'--libdir',
dest="libdir",
default=local_libpath(),
help="directory path that contains Thrift Python library")
parser.add_option(
'--gen-base',
dest="gen_base",
default=SCRIPT_DIR,
help="directory path that contains Thrift Python library")
parser.set_defaults(verbose=1)
options, args = parser.parse_args()
generated_dirs = []
for gp_dir in options.genpydirs.split(','):
generated_dirs.append('gen-py-%s' % (gp_dir))
# commandline permits a single class name to be specified to override SERVERS=[...]
servers = SERVERS
if len(args) == 1:
if args[0] in SERVERS:
servers = args
else:
print('Unavailable server type "%s", please choose one of: %s' %
(args[0], servers))
sys.exit(0)
tests = TestCases(options.gen_base, options.libdir, options.port,
generated_dirs, servers, options.verbose)
# run tests without a client/server first
print('----------------')
print(
' Executing individual test scripts with various generated code directories'
)
print(' Directories to be tested: ' + ', '.join(generated_dirs))
print(' Scripts to be tested: ' + ', '.join(SCRIPTS))
print('----------------')
for genpydir in generated_dirs:
for script in SCRIPTS:
runScriptTest(options.libdir, options.gen_base, genpydir, script)
print('----------------')
print(
' Executing Client/Server tests with various generated code directories'
)
print(' Servers to be tested: ' + ', '.join(servers))
print(' Directories to be tested: ' + ', '.join(generated_dirs))
print(' Protocols to be tested: ' + ', '.join(PROTOS))
print(' Options to be tested: ZLIB(yes/no), SSL(yes/no)')
print('----------------')
if options.all:
tests.run_all_tests()
else:
tests.test_feature('gendir', generated_dirs)
tests.test_feature('server', servers)
tests.test_feature('proto', PROTOS)
tests.test_feature('zlib', [False, True])
tests.test_feature('ssl', [False, True])
def main():
parser = OptionParser()
parser.add_option('--all', action="store_true", dest='all')
parser.add_option('--genpydirs', type='string', dest='genpydirs',
default='default,slots,oldstyle,no_utf8strings,dynamic,dynamicslots',
help='directory extensions for generated code, used as suffixes for \"gen-py-*\" added sys.path for individual tests')
parser.add_option("--port", type="int", dest="port", default=9090,
help="port number for server to listen on")
parser.add_option('-v', '--verbose', action="store_const",
dest="verbose", const=2,
help="verbose output")
parser.add_option('-q', '--quiet', action="store_const",
dest="verbose", const=0,
help="minimal output")
parser.add_option('-L', '--libdir', dest="libdir", default=local_libpath(),
help="directory path that contains Thrift Python library")
parser.add_option('--gen-base', dest="gen_base", default=SCRIPT_DIR,
help="directory path that contains Thrift Python library")
parser.set_defaults(verbose=1)
options, args = parser.parse_args()
generated_dirs = []
for gp_dir in options.genpydirs.split(','):
generated_dirs.append('gen-py-%s' % (gp_dir))
# commandline permits a single class name to be specified to override SERVERS=[...]
servers = default_servers()
if len(args) == 1:
if args[0] in servers:
servers = args
else:
print('Unavailable server type "%s", please choose one of: %s' % (args[0], servers))
sys.exit(0)
tests = TestCases(options.gen_base, options.libdir, options.port, generated_dirs, servers, options.verbose)
# run tests without a client/server first
print('----------------')
print(' Executing individual test scripts with various generated code directories')
print(' Directories to be tested: ' + ', '.join(generated_dirs))
print(' Scripts to be tested: ' + ', '.join(SCRIPTS))
print('----------------')
for genpydir in generated_dirs:
for script in SCRIPTS:
runScriptTest(options.libdir, options.gen_base, genpydir, script)
print('----------------')
print(' Executing Client/Server tests with various generated code directories')
print(' Servers to be tested: ' + ', '.join(servers))
print(' Directories to be tested: ' + ', '.join(generated_dirs))
print(' Protocols to be tested: ' + ', '.join(PROTOS))
print(' Options to be tested: ZLIB(yes/no), SSL(yes/no)')
print('----------------')
if options.all:
tests.run_all_tests()
else:
tests.test_feature('gendir', generated_dirs)
tests.test_feature('server', servers)
tests.test_feature('proto', PROTOS)
tests.test_feature('zlib', [False, True])
tests.test_feature('ssl', [False, True])
