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)
config.user_count = 2 config.bucket_count = 10 config.objects_count = 2 config.objects_size_range = {"min": 10, "max": 50} else: with open(yaml_file, "r") as f: doc = yaml.load(f) config.user_count = doc["config"]["user_count"] config.cluster_name = doc["config"]["cluster_name"] config.bucket_count = doc["config"]["bucket_count"] config.objects_count = doc["config"]["objects_count"] config.objects_size_range = { "min": doc["config"]["objects_size_range"]["min"], "max": doc["config"]["objects_size_range"]["max"], } config.test_ops = doc["config"]["test_ops"] log.info( "user_count:%s\n" "bucket_count: %s\n" "objects_count: %s\n" "objects_size_range: %s\n" % ( config.user_count, config.bucket_count, config.objects_count, config.objects_size_range, ) ) log.info("test_ops: %s" % config.test_ops) test_exec(config)