mirrors/uv
1
0
Fork 0
mirror of https://github.com/astral-sh/uv.git synced 2025-07-08 05:45:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Collapse duplicate toolchain search in test setup (#4442)

Browse source
This commit is contained in:
Zanie Blue 2024-06-22 10:43:03 -04:00 • committed by GitHub
parent 8f2d8b3f4f
commit c7149e96ca
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

54
crates/uv/tests/common/mod.rs
View file

@ -709,47 +709,21 @@ pub fn python_toolchains_for_versions(
let cache = Cache::from_path(temp_dir.child("cache").to_path_buf()).init()?;
let selected_pythons = python_versions
.iter()
.flat_map(|python_version| {
let inner = InstalledToolchains::from_settings()
.map(|toolchains| {
toolchains
.find_version(
&PythonVersion::from_str(python_version)
.expect("Tests should use a valid Python version"),
)
.expect("Tests are run on a supported platform")
.map(|toolchain| {
toolchain
.executable()
.parent()
.expect("Executables must exist in a directory")
.to_path_buf()
})
.collect::<Vec<_>>()
})
.unwrap_or_default();
if inner.is_empty() {
// TODO(zanieb): Collapse these two cases now that we support `ToolchainPreference`
// Fallback to a system lookup if we failed to find one in the toolchain directory
if let Ok(toolchain) = Toolchain::find(
&ToolchainRequest::parse(python_version),
// Without required, we could pick the current venv here and the test fails
// because the venv subcommand requires a system interpreter.
EnvironmentPreference::OnlySystem,
ToolchainPreference::PreferInstalledManaged,
&cache,
) {
vec![toolchain
.into_interpreter()
.sys_executable()
.parent()
.expect("Python executable should always be in a directory")
.to_path_buf()]
} else {
panic!("Could not find Python {python_version} for test");
}
.map(|python_version| {
if let Ok(toolchain) = Toolchain::find(
&ToolchainRequest::parse(python_version),
EnvironmentPreference::OnlySystem,
ToolchainPreference::PreferInstalledManaged,
&cache,
) {
toolchain
.into_interpreter()
.sys_executable()
.parent()
.expect("Python executable should always be in a directory")
.to_path_buf()
} else {
inner
panic!("Could not find Python {python_version} for test");
}
})
.collect::<Vec<_>>();