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)
