build.rs

// Copyright 2022 RisingLight Project Authors. Licensed under Apache-2.0.

use std::env;
use std::fmt::Write as _;
use std::io::Write;
use std::path::PathBuf;

fn main() {
    // Scan test scripts and generate test cases.
    println!("cargo:rerun-if-changed=tests/sql");
    const PATTERN: &str = "tests/sql/**/[!_]*.slt"; // ignore files start with '_'
    const MEM_BLOCKLIST: &[&str] = &["statistics.slt"];
    const DISK_BLOCKLIST: &[&str] = &["blob.slt"];

    let path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("testcase.rs");
    let mut fout = std::fs::File::create(path).expect("failed to create file");

    let paths = glob::glob(PATTERN).expect("failed to find test files");
    let mut mem_attrs = String::new();
    let mut disk_attrs = String::new();
    for entry in paths {
        let path = entry.expect("failed to read glob entry");
        let subpath = path.strip_prefix("tests/sql").unwrap().to_str().unwrap();
        if !MEM_BLOCKLIST.iter().any(|p| subpath.contains(p)) {
            writeln!(mem_attrs, "#[test_case::test_case({:?})]", subpath).unwrap();
        }
        if !DISK_BLOCKLIST.iter().any(|p| subpath.contains(p)) {
            writeln!(disk_attrs, "#[test_case::test_case({:?})]", subpath).unwrap();
        }
    }
    writeln!(
        fout,
        "{mem_attrs}
        fn mem(name: &str) {{ test_mem(name); }}
        {disk_attrs}
        fn disk(name: &str) {{ test_disk(name); }}"
    )
    .expect("failed to write file");
}