uv-test 0.0.52

This is an internal component crate of uv
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

use std::path::Path;
use std::thread;
use std::time::Duration;

use wiremock::matchers::any;
use wiremock::{Mock, MockServer, Request, ResponseTemplate};

/// Background wiremock server shared by the local test indexes in this crate.
pub(crate) struct HttpServer {
    url: String,
    shutdown: Option<tokio::sync::oneshot::Sender<()>>,
    thread: Option<thread::JoinHandle<()>>,
}

impl HttpServer {
    pub(crate) fn start(
        handler: impl Fn(&Request, &str) -> ResponseTemplate + Send + Sync + 'static,
    ) -> Self {
        let (url_tx, url_rx) = std::sync::mpsc::channel::<String>();
        let (shutdown_tx, shutdown_rx) = tokio::sync::oneshot::channel::<()>();

        let thread = thread::spawn(move || {
            let runtime = tokio::runtime::Builder::new_current_thread()
                .enable_all()
                .build()
                .expect("failed to create tokio runtime for local HTTP test server");

            runtime.block_on(async move {
                let server = MockServer::start().await;
                let server_uri = server.uri();

                Mock::given(any())
                    .respond_with(move |request: &Request| handler(request, &server_uri))
                    .mount(&server)
                    .await;

                url_tx.send(server.uri()).ok();
                let _ = shutdown_rx.await;
            });
        });

        let url = url_rx
            .recv_timeout(Duration::from_secs(30))
            .expect("timed out waiting for local HTTP test server to start");

        Self {
            url,
            shutdown: Some(shutdown_tx),
            thread: Some(thread),
        }
    }

    pub(crate) fn url(&self) -> &str {
        &self.url
    }
}

pub(crate) fn content_type_for_filename(filename: &str) -> &'static str {
    if Path::new(filename)
        .extension()
        .is_some_and(|extension| extension.eq_ignore_ascii_case("whl"))
    {
        "application/zip"
    } else {
        "application/gzip"
    }
}

impl Drop for HttpServer {
    fn drop(&mut self) {
        drop(self.shutdown.take());
        if let Some(thread) = self.thread.take() {
            thread.join().ok();
        }
    }
}