Function test_substitute_tilde

datafusion-cli/src/catalog.rs:341–379

()

    #[cfg(not(target_os = "windows"))]
    #[test]
    fn test_substitute_tilde() {
        use std::{env, path::PathBuf};
        let original_home = home_dir();
        let test_home_path = if cfg!(windows) {
            "C:\\Users\\user"
        } else {
            "/home/user"
        };
        unsafe {
            env::set_var(
                if cfg!(windows) { "USERPROFILE" } else { "HOME" },
                test_home_path,
            );
        }
        let input = "~/Code/datafusion/benchmarks/data/tpch_sf1/part/part-0.parquet";
        let expected = PathBuf::from(test_home_path)
            .join("Code")
            .join("datafusion")
            .join("benchmarks")
            .join("data")
            .join("tpch_sf1")
            .join("part")
            .join("part-0.parquet")
            .to_string_lossy()
            .to_string();
        let actual = substitute_tilde(input.to_string());
        assert_eq!(actual, expected);
        unsafe {
            match original_home {
                Some(home_path) => env::set_var(
                    if cfg!(windows) { "USERPROFILE" } else { "HOME" },
                    home_path.to_str().unwrap(),
                ),
                None => {
                    env::remove_var(if cfg!(windows) { "USERPROFILE" } else { "HOME" })
                }
            }
        }
    }
}