Function test_data_dir

datafusion/common/src/test_util.rs:727–766

()

    #[test]
    fn test_data_dir() {
        let udf_env = "get_data_dir";
        let cwd = env::current_dir().unwrap();

        let existing_pb = cwd.join("..");
        let existing = existing_pb.display().to_string();
        let existing_str = existing.as_str();

        let non_existing = cwd.join("non-existing-dir").display().to_string();
        let non_existing_str = non_existing.as_str();

        unsafe {
            env::set_var(udf_env, non_existing_str);
            let res = get_data_dir(udf_env, existing_str);
            assert!(res.is_err());

            env::set_var(udf_env, "");
            let res = get_data_dir(udf_env, existing_str);
            assert!(res.is_ok());
            assert_eq!(res.unwrap(), existing_pb);

            env::set_var(udf_env, " ");
            let res = get_data_dir(udf_env, existing_str);
            assert!(res.is_ok());
            assert_eq!(res.unwrap(), existing_pb);

            env::set_var(udf_env, existing_str);
            let res = get_data_dir(udf_env, existing_str);
            assert!(res.is_ok());
            assert_eq!(res.unwrap(), existing_pb);

            env::remove_var(udf_env);
            let res = get_data_dir(udf_env, non_existing_str);
            assert!(res.is_err());

            let res = get_data_dir(udf_env, existing_str);
            assert!(res.is_ok());
            assert_eq!(res.unwrap(), existing_pb);
        }
    }

    #[test]
    #[cfg(feature = "parquet")]