def main(): global HOST, CREDS, LP parser = optparse.OptionParser("rodc.py [options] <host>") sambaopts = options.SambaOptions(parser) versionopts = options.VersionOptions(parser) credopts = options.CredentialsOptions(parser) subunitopts = SubunitOptions(parser) parser.add_option_group(sambaopts) parser.add_option_group(versionopts) parser.add_option_group(credopts) parser.add_option_group(subunitopts) opts, args = parser.parse_args() LP = sambaopts.get_loadparm() CREDS = credopts.get_credentials(LP) try: HOST = args[0] except IndexError: parser.print_usage() sys.exit(1) if "://" not in HOST: if os.path.isfile(HOST): HOST = "tdb://%s" % HOST else: HOST = "ldap://%s" % HOST TestProgram(module=__name__, opts=subunitopts)
def main(): global RODC, RWDC, CREDS, LP parser = optparse.OptionParser( "rodc_rwdc.py [options] <rodc host> <rwdc host>") sambaopts = options.SambaOptions(parser) versionopts = options.VersionOptions(parser) credopts = options.CredentialsOptions(parser) subunitopts = SubunitOptions(parser) parser.add_option_group(sambaopts) parser.add_option_group(versionopts) parser.add_option_group(credopts) parser.add_option_group(subunitopts) opts, args = parser.parse_args() LP = sambaopts.get_loadparm() CREDS = credopts.get_credentials(LP) CREDS.set_gensec_features(CREDS.get_gensec_features() | gensec.FEATURE_SEAL) try: RODC, RWDC = args except ValueError: parser.print_usage() sys.exit(1) set_auto_replication(RWDC, True) try: TestProgram(module=__name__, opts=subunitopts) finally: set_auto_replication(RWDC, True)
# performance tests that want to apply to old versions. try: from samba.tests.subunitrun import SubunitOptions, TestProgram ANCIENT_SAMBA = False except ImportError: ANCIENT_SAMBA = True samba.ensure_external_module("testtools", "testtools") samba.ensure_external_module("subunit", "subunit/python") from subunit.run import SubunitTestRunner import unittest parser = optparse.OptionParser( "ad_dc_provision_performance.py [options] <host>") sambaopts = options.SambaOptions(parser) parser.add_option_group(sambaopts) parser.add_option_group(options.VersionOptions(parser)) if not ANCIENT_SAMBA: subunitopts = SubunitOptions(parser) parser.add_option_group(subunitopts) # use command line creds if available credopts = options.CredentialsOptions(parser) parser.add_option_group(credopts) opts, args = parser.parse_args() if len(args) < 1: parser.print_usage() sys.exit(1) host = args[0]