hotpath 0.30.3

#[path = "common/support.rs"]
mod common;

#[cfg(all(test, feature = "hotpath"))]
pub mod tests {
    use {
        hotpath::json::{JsonThreadEntry, JsonThreadsList},
        std::time::Duration,
    };

    use crate::common::assert_contains_all;
    use crate::common::example::{poll_endpoint, Example};

    fn poll_threads(port: u16) -> JsonThreadsList {
        let json_text = poll_endpoint(
            &format!("http://localhost:{port}/threads"),
            30,
            Duration::from_millis(1000),
            |body| body.contains("thread_count") && !body.contains("\"thread_count\":0"),
        );
        serde_json::from_str(&json_text).expect("Failed to parse threads JSON")
    }

    fn internal_threads(threads: &JsonThreadsList) -> Vec<&JsonThreadEntry> {
        threads
            .data
            .iter()
            .filter(|t| t.name.starts_with("hp-"))
            .collect()
    }

    // HOTPATH_METRICS_PORT=6775 TEST_SLEEP_SECONDS=10 cargo run -p test-tokio-async --example basic --features hotpath-alloc
    #[test]
    fn test_threads_endpoint() {
        let _running = Example::new("test-tokio-async", "basic")
            .features("hotpath-alloc")
            .env("HOTPATH_METRICS_PORT", "6775")
            .env("TEST_SLEEP_SECONDS", "10")
            .spawn();

        // The first poll only waits for the server; the second runs after
        // tiny_http's workers have parsed requests and allocated.
        poll_threads(6775);
        let threads_response = poll_threads(6775);

        assert!(
            threads_response.thread_count > 0,
            "Expected at least 1 thread, got {}",
            threads_response.thread_count
        );

        assert_eq!(
            threads_response.thread_count,
            threads_response.data.len(),
            "thread_count should match data.len()"
        );

        let hp_threads = internal_threads(&threads_response);
        assert!(
            hp_threads.is_empty(),
            "Internal threads should be hidden by default, found: {:?}",
            hp_threads.iter().map(|t| &t.name).collect::<Vec<_>>()
        );

        for thread in &threads_response.data {
            assert!(thread.os_tid > 0, "Thread should have valid os_tid");
        }
    }

    // HOTPATH_SHOW_INTERNAL_THREADS=1 HOTPATH_METRICS_PORT=6788 TEST_SLEEP_SECONDS=10 cargo run -p test-tokio-async --example basic --features hotpath-alloc
    #[test]
    fn test_threads_endpoint_shows_internal_threads() {
        let _running = Example::new("test-tokio-async", "basic")
            .features("hotpath-alloc")
            .env("HOTPATH_SHOW_INTERNAL_THREADS", "1")
            .env("HOTPATH_METRICS_PORT", "6788")
            .env("TEST_SLEEP_SECONDS", "10")
            .spawn();

        poll_threads(6788);
        let threads_response = poll_threads(6788);

        let hp_threads = internal_threads(&threads_response);
        assert!(
            !hp_threads.is_empty(),
            "Expected at least one hp- thread, found none. Threads: {:?}",
            threads_response
                .data
                .iter()
                .map(|t| &t.name)
                .collect::<Vec<_>>()
        );

        // On Linux tiny_http's unnamed workers inherit `hp-server`; shown,
        // their request-parsing allocations are counted again.
        #[cfg(target_os = "linux")]
        assert!(
            hp_threads.iter().any(|t| {
                t.name == "hp-server"
                    && t.alloc_bytes
                        .as_deref()
                        .and_then(hotpath::parse_bytes)
                        .is_some_and(|b| b > 0)
            }),
            "Expected an allocating tiny_http worker named hp-server: {hp_threads:?}"
        );
    }

    // HOTPATH_OUTPUT_FORMAT=json cargo run -p test-tokio-async --example guard_timeout_threads --features hotpath-alloc
    #[test]
    fn test_report_hides_internal_threads() {
        let report = Example::new("test-tokio-async", "guard_timeout_threads")
            .features("hotpath-alloc")
            .env("HOTPATH_THREADS_LIMIT", "0")
            .json()
            .report();
        let threads = report.threads.expect("threads section");

        assert!(!threads.data.is_empty(), "Expected the main thread row");
        let hp_threads = internal_threads(&threads);
        assert!(
            hp_threads.is_empty(),
            "Internal threads should be hidden from the report, found: {:?}",
            hp_threads.iter().map(|t| &t.name).collect::<Vec<_>>()
        );
    }

    // cargo run -p test-tokio-async --example guard_timeout_threads --features hotpath
    #[test]
    fn test_guard_timeout_output() {
        let stdout = Example::new("test-tokio-async", "guard_timeout_threads").stdout();

        assert_contains_all(
            &stdout,
            &[
                "[hotpath]",
                "| threads",
                "Thread CPU and memory statistics.",
                "Max RSS:",
            ],
        );
    }

    // HOTPATH_OUTPUT_FORMAT=json cargo run -p test-tokio-async --example rss_exec --features hotpath
    #[cfg(target_os = "linux")]
    #[test]
    fn test_max_rss_excludes_pre_exec_peak() {
        let report = Example::new("test-tokio-async", "rss_exec").json().report();
        let threads = report.threads.expect("threads section");
        let max_rss = threads.rss_bytes_max.as_deref().expect("rss_bytes_max");
        let max_rss = hotpath::parse_bytes(max_rss).expect("rss_bytes_max parses");

        // The example touches 200 MB before exec'ing into the profiled run.
        assert!(
            max_rss < 100 * 1024 * 1024,
            "Max RSS {max_rss} B includes the pre-exec ballast"
        );
    }
}