Function track_command

crates/memtrack/tests/shared.rs:131–163

Track a spawned command, collecting all memory events. When `discover_system_allocators` is true, the tracker will scan for all allocators on the system (slower). When false, only `extra_allocators` are used.

(
    mut command: Command,
    extra_allocators: &[AllocatorLib],
    discover_system_allocators: bool,
)

/// When `discover_system_allocators` is true, the tracker will scan for all
/// allocators on the system (slower). When false, only `extra_allocators` are used.
pub fn track_command(
    mut command: Command,
    extra_allocators: &[AllocatorLib],
    discover_system_allocators: bool,
) -> TrackResult {
    // IMPORTANT: Always initialize the tracker BEFORE spawning the binary, as it can take some time to
    // attach to all the allocator libraries (especially when using NixOS).
    let mut tracker = if discover_system_allocators {
        memtrack::Tracker::new()?
    } else {
        memtrack::Tracker::new_without_allocators()?
    };
    tracker.attach_allocators(extra_allocators)?;

    let child = command.spawn().context("Failed to spawn command")?;
    let root_pid = child.id() as i32;

    tracker.enable()?;
    let rx = tracker.track(root_pid)?;

    let mut events = Vec::new();
    while let Ok(event) = rx.recv_timeout(Duration::from_secs(10)) {
        events.push(event);
    }

    // Drop the tracker in a new thread to not block the test
    let thread_handle = std::thread::spawn(move || core::mem::drop(tracker));

    info!("Tracked {} events", events.len());
    trace!("Events: {events:#?}");

    Ok((events, thread_handle))
}

pub fn track_binary_with_opts(binary: &Path, extra_allocators: &[AllocatorLib]) -> TrackResult {
    track_command(Command::new(binary), extra_allocators, true)