mirror of
https://github.com/astral-sh/uv.git
synced 2025-07-15 01:05:01 +00:00
e7742070c1
## Summary Add netrc support to the uv-client. closes #1405 ## Test Plan I've added a corresponding test case to validate the correct header. Furthermore a tested it against a real world private repository.
68 lines
2 KiB
Rust
68 lines
2 KiB
Rust
use std::env;
|
|
use std::io::Write;
|
|
|
|
use anyhow::Result;
|
|
use futures::future;
|
|
use hyper::header::AUTHORIZATION;
|
|
use hyper::server::conn::Http;
|
|
use hyper::service::service_fn;
|
|
use hyper::{Body, Request, Response};
|
|
use tempfile::NamedTempFile;
|
|
use tokio::net::TcpListener;
|
|
|
|
use uv_cache::Cache;
|
|
use uv_client::RegistryClientBuilder;
|
|
|
|
#[tokio::test]
|
|
async fn test_client_with_netrc_credentials() -> Result<()> {
|
|
// Set up the TCP listener on a random available port
|
|
let listener = TcpListener::bind("127.0.0.1:0").await?;
|
|
let addr = listener.local_addr()?;
|
|
|
|
// Spawn the server loop in a background task
|
|
tokio::spawn(async move {
|
|
let svc = service_fn(move |req: Request<Body>| {
|
|
// Get User Agent Header and send it back in the response
|
|
let auth = req
|
|
.headers()
|
|
.get(AUTHORIZATION)
|
|
.and_then(|v| v.to_str().ok())
|
|
.map(|s| s.to_string())
|
|
.unwrap_or_default(); // Empty Default
|
|
future::ok::<_, hyper::Error>(Response::new(Body::from(auth)))
|
|
});
|
|
// Start Hyper Server
|
|
let (socket, _) = listener.accept().await.unwrap();
|
|
Http::new()
|
|
.http1_keep_alive(false)
|
|
.serve_connection(socket, svc)
|
|
.with_upgrades()
|
|
.await
|
|
.expect("Server Started");
|
|
});
|
|
|
|
// Create a netrc file
|
|
let mut netrc_file = NamedTempFile::new()?;
|
|
env::set_var("NETRC", netrc_file.path());
|
|
writeln!(netrc_file, "machine 127.0.0.1 login user password 1234")?;
|
|
|
|
// Initialize uv-client
|
|
let cache = Cache::temp()?;
|
|
let client = RegistryClientBuilder::new(cache).build();
|
|
|
|
// Send request to our dummy server
|
|
let res = client
|
|
.cached_client()
|
|
.uncached()
|
|
.get(format!("http://{addr}"))
|
|
.send()
|
|
.await?;
|
|
|
|
// Check the HTTP status
|
|
assert!(res.status().is_success());
|
|
|
|
// Verify auth header
|
|
assert_eq!(res.text().await?, "Basic dXNlcjoxMjM0");
|
|
|
|
Ok(())
|
|
}
|