sys.exit(1) if __name__ == '__main__': project_dir = os.path.abspath(os.path.join(__file__, "../../..")) test_data_dir = 'test_data' TEST_DATA_PATH = (os.path.join(project_dir, test_data_dir)) log.info('TEST_DATA_PATH: %s' % TEST_DATA_PATH) if not os.path.exists(TEST_DATA_PATH): log.info('test data dir not exists, creating.. ') os.makedirs(TEST_DATA_PATH) parser = argparse.ArgumentParser(description='RGW S3 Automation') parser.add_argument('-c', dest="config", help='RGW Test yaml configuration') args = parser.parse_args() yaml_file = args.config config = Config() if yaml_file is None: config.bucket_count = 10 else: with open(yaml_file, 'r') as f: doc = yaml.load(f) config.bucket_count = doc['config']['bucket_count'] config.cluster_name = doc['config']['cluster_name'] config.test_ops = doc['config']['test_ops'] log.info('bucket_count: %s\n' 'cluster_name: %s' % (config.bucket_count, config.cluster_name)) log.info('test_ops: %s' % config.test_ops) test_exec(config)
if __name__ == "__main__": project_dir = os.path.abspath(os.path.join(__file__, "../../..")) test_data_dir = "test_data" TEST_DATA_PATH = os.path.join(project_dir, test_data_dir) log.info("TEST_DATA_PATH: %s" % TEST_DATA_PATH) if not os.path.exists(TEST_DATA_PATH): log.info("test data dir not exists, creating.. ") os.makedirs(TEST_DATA_PATH) parser = argparse.ArgumentParser(description="RGW S3 Automation") parser.add_argument("-c", dest="config", help="RGW Test yaml configuration") args = parser.parse_args() yaml_file = args.config config = Config() with open(yaml_file, "r") as f: doc = yaml.load(f) config.container_count = doc["config"]["container_count"] config.objects_count = doc["config"]["objects_count"] config.cluster_name = doc["config"]["cluster_name"] config.objects_size_range = { "min": doc["config"]["objects_size_range"]["min"], "max": doc["config"]["objects_size_range"]["max"], } log.info( "bucket_count: %s\n" "objects_count: %s\n" "objects_size_range: %s\n" % (config.container_count, config.objects_count, config.objects_size_range) ) test_exec(config)