Example #1
0
        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)