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(yaml_file)
config.shards = None
config.max_objects = None
if yaml_file is None:
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(
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='NFS-Ganesha-RGW Automation')
parser.add_argument('-r', dest="rgw_user_info", help='RGW user info')
parser.add_argument('-c', dest="test_config", help='Test Configuration')
args = parser.parse_args()
rgw_user_info_yaml = args.rgw_user_info
test_config_yaml = args.test_config
test_config = Config() # test config
with open(test_config_yaml, 'r') as f:
doc = yaml.load(f)
test_config.bucket_count = doc['config']['bucket_count']
test_config.objects_count = doc['config']['objects_count']
test_config.objects_size_range = {
'min': doc['config']['objects_size_range']['min'],
'max': doc['config']['objects_size_range']['max']
}
test_config.io_op_config = doc['io_op_config']
test_exec(rgw_user_info_yaml, test_config)
except Exception as e:
log.info(e)
log.info(traceback.format_exc())
test_info.failed_status('test failed')
sys.exit(1)
except TestExecError as e:
log.info(e)
log.info(traceback.format_exc())
test_info.failed_status('test failed')
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',
default=None)
args = parser.parse_args()
config = Config()
config.user_count = 2
config.bucket_count = 2
test_exec(config)
