Function install_tests

crates/ruff_python_parser/tests/generate_inline_tests.rs:101–141

(tests: &HashMap<String, Test>, target_dir: &str)

}

fn install_tests(tests: &HashMap<String, Test>, target_dir: &str) -> Result<TestFiles> {
    let root_dir = project_root();
    let tests_dir = root_dir.join(target_dir);
    if !tests_dir.is_dir() {
        fs::create_dir_all(&tests_dir)?;
    }

    // Test kind is irrelevant for existing test cases.
    let existing = existing_tests(&tests_dir)?;

    let mut updated_files = vec![];
    // Sort so `TestFiles` reports generated files that need updating deterministically.
    let mut sorted_tests = tests.iter().collect::<Vec<_>>();
    sorted_tests.sort_unstable_by_key(|(name, _)| *name);

    for (name, test) in sorted_tests {
        let path = match existing.get(name) {
            Some(path) => path.clone(),
            None => tests_dir.join(name).with_extension("py"),
        };
        match fs::read_to_string(&path) {
            Ok(old_contents) if old_contents == test.contents => continue,
            _ => {}
        }
        fs::write(&path, &test.contents)
            .with_context(|| format!("Failed to write to {:?}", path.display()))?;
        updated_files.push(path);
    }

    let mut unreferenced = existing
        .into_iter()
        .filter(|(name, _)| !tests.contains_key(name))
        .map(|(_, path)| path)
        .collect::<Vec<_>>();
    unreferenced.sort_unstable();

    Ok(TestFiles {
        unreferenced,
        updated: updated_files,
    })
}

#[derive(Default, Debug)]
struct TestCollection {