js_parser.rs

//! Integration test: verify JS trace parser matches Rust parser

use dial9_tokio_telemetry::analysis_unstable::TraceReader;
use dial9_tokio_telemetry::telemetry::{RotatingWriter, TracedRuntime};
use std::io::{BufWriter, Write};
use std::process::Command;
use tempfile::TempDir;

#[inline(never)]
fn burn_cpu(iterations: u64) -> u64 {
    let mut result = 0u64;
    for i in 0..iterations {
        result = result.wrapping_add(i.wrapping_mul(i));
    }
    result
}

async fn cpu_task(id: usize) {
    for _ in 0..3 {
        let _ = burn_cpu(1_000_000);
        tokio::task::yield_now().await;
    }
    eprintln!("Task {id} done");
}

#[test]
fn test_js_parser_matches_rust() {
    let temp_dir = TempDir::new().unwrap();
    let trace_path = temp_dir.path().join("test_trace.bin");
    let jsonl_path = temp_dir.path().join("expected.jsonl");

    // Generate a trace — enable CPU profiling on Linux where it's available
    {
        let mut builder = tokio::runtime::Builder::new_multi_thread();
        builder.worker_threads(2).enable_all();

        let writer = RotatingWriter::single_file(&trace_path).unwrap();
        #[allow(unused_mut)]
        let mut tb = TracedRuntime::builder().with_task_tracking(true);
        #[cfg(feature = "cpu-profiling")]
        {
            tb = tb.with_cpu_profiling(
                dial9_tokio_telemetry::telemetry::cpu_profile::CpuProfilingConfig::default(),
            );
        }
        let (runtime, _guard) = tb.build_and_start(builder, writer).unwrap();

        runtime.block_on(async {
            let mut tasks = vec![];
            for i in 0..10 {
                tasks.push(tokio::spawn(cpu_task(i)));
            }
            for task in tasks {
                let _ = task.await;
            }
        });
    }

    let sealed_path = temp_dir.path().join("test_trace.0.bin");
    eprintln!("Generated trace at {}", sealed_path.display());

    // Export to JSONL using Rust parser (in-process to avoid cargo subprocess overhead)
    {
        let reader = TraceReader::new(sealed_path.to_str().unwrap()).unwrap();
        let file = std::fs::File::create(&jsonl_path).unwrap();
        let mut w = BufWriter::new(file);
        for e in &reader.all_events {
            serde_json::to_writer(&mut w, &e).unwrap();
            w.write_all(b"\n").unwrap();
        }
        w.flush().unwrap();
    }

    eprintln!("Exported JSONL to {}", jsonl_path.display());

    // Run JS parser test (use CARGO_MANIFEST_DIR to find ui directory)
    let manifest_dir = std::env::var("CARGO_MANIFEST_DIR").unwrap();
    let test_script = std::path::Path::new(&manifest_dir)
        .parent()
        .unwrap()
        .join("dial9-viewer")
        .join("ui")
        .join("test_parser.js");

    let test_output = Command::new("node")
        .args([
            test_script.to_str().unwrap(),
            sealed_path.to_str().unwrap(),
            jsonl_path.to_str().unwrap(),
        ])
        .output()
        .expect("Failed to run node test_parser.js");

    eprintln!("{}", String::from_utf8_lossy(&test_output.stdout));

    assert!(
        test_output.status.success(),
        "JS parser test failed:\n{}",
        String::from_utf8_lossy(&test_output.stderr)
    );
}

/// Verify that SymbolTableEntry frames at the end of a trace are still resolved
/// even when the event cap is reached (i.e., the parser doesn't break out of
/// the frame loop early and skip trailing metadata).
#[cfg(feature = "cpu-profiling")]
#[test]
fn test_js_parser_resolves_symbols_past_event_cap() {
    use dial9_perf_self_profile::offline_symbolize::SymbolTableEntry;
    use dial9_tokio_telemetry::telemetry::{PollEndEvent, WorkerId};
    use dial9_trace_format::encoder::Encoder;

    let temp_dir = TempDir::new().unwrap();
    let trace_path = temp_dir.path().join("capped_trace.bin");

    {
        let mut enc = Encoder::new();
        for i in 0..10u64 {
            enc.write(&PollEndEvent {
                timestamp_ns: i * 1_000_000,
                worker_id: WorkerId::from(0usize),
            })
            .unwrap();
        }
        let sym_name = enc.intern_string("my_function").unwrap();
        let empty_file = enc.intern_string("").unwrap();
        enc.write(&SymbolTableEntry {
            timestamp_ns: 0,
            addr: 0x1234,
            size: 256,
            symbol_name: sym_name,
            inline_depth: 0,
            source_file: empty_file,
            source_line: 0,
        })
        .unwrap();
        std::fs::write(&trace_path, enc.finish()).unwrap();
    }

    // Node script: parse with maxEvents=5, verify the symbol is still resolved.
    let manifest_dir = std::env::var("CARGO_MANIFEST_DIR").unwrap();
    let script = format!(
        r#"
const {{ parseTrace }} = require("{viewer}/trace_parser.js");
const fs = require("fs");
async function main() {{
    const result = await parseTrace(fs.readFileSync("{trace}"), {{ maxEvents: 5 }});
    if (result.events.length > 5) {{
        console.error("expected at most 5 events, got " + result.events.length);
        process.exit(1);
    }}
    if (!result.truncated) {{
        console.error("expected truncated=true");
        process.exit(1);
    }}
    const sym = result.callframeSymbols.get("0x1234");
    if (!sym || sym.symbol !== "my_function") {{
        console.error("symbol not resolved: " + JSON.stringify(sym));
        process.exit(1);
    }}
    console.log("OK: " + result.events.length + " events, symbol resolved");
}}
main().catch((e) => {{ console.error(e); process.exit(1); }});
"#,
        viewer = std::path::Path::new(&manifest_dir)
            .parent()
            .unwrap()
            .join("dial9-viewer")
            .join("ui")
            .display(),
        trace = trace_path.display(),
    );

    let output = Command::new("node")
        .args(["-e", &script])
        .output()
        .expect("Failed to run node");

    eprintln!("{}", String::from_utf8_lossy(&output.stdout));
    assert!(
        output.status.success(),
        "JS parser symbol resolution test failed:\n{}",
        String::from_utf8_lossy(&output.stderr)
    );
}

#[test]
fn test_js_parser_alloc_free_events() {
    use dial9_tokio_telemetry::telemetry::{AllocEvent, FreeEvent};
    use dial9_trace_format::encoder::Encoder;

    let temp_dir = TempDir::new().unwrap();
    let trace_path = temp_dir.path().join("alloc_trace.bin");

    {
        let mut enc = Encoder::new();
        let stack = enc.intern_stack_frames(&[0xAAAA, 0xBBBB, 0xCCCC]).unwrap();
        enc.write(&AllocEvent {
            timestamp_ns: 5_000_000,
            tid: 42,
            size: 4096,
            addr: 0xDEAD_BEEF,
            callchain: stack,
        })
        .unwrap();
        enc.write(&FreeEvent {
            timestamp_ns: 10_000_000,
            tid: 7,
            addr: 0xDEAD_BEEF,
            size: 4096,
            alloc_timestamp_ns: 5_000_000,
        })
        .unwrap();
        std::fs::write(&trace_path, enc.finish()).unwrap();
    }

    let manifest_dir = std::env::var("CARGO_MANIFEST_DIR").unwrap();
    let script = format!(
        r#"
const {{ parseTrace }} = require("{viewer}/trace_parser.js");
const fs = require("fs");
async function main() {{
    const result = await parseTrace(fs.readFileSync("{trace}"));
    if (result.allocEvents.length !== 1) {{
        console.error("expected 1 allocEvent, got " + result.allocEvents.length);
        process.exit(1);
    }}
    const a = result.allocEvents[0];
    if (a.tid !== 42) {{ console.error("bad tid: " + a.tid); process.exit(1); }}
    if (a.size !== 4096) {{ console.error("bad size: " + a.size); process.exit(1); }}
    if (a.callchain.length !== 3) {{ console.error("bad callchain len: " + a.callchain.length); process.exit(1); }}

    if (result.freeEvents.length !== 1) {{
        console.error("expected 1 freeEvent, got " + result.freeEvents.length);
        process.exit(1);
    }}
    const f = result.freeEvents[0];
    if (f.tid !== 7) {{ console.error("bad free tid: " + f.tid); process.exit(1); }}
    if (f.addr !== "3735928559") {{ console.error("bad free addr: " + f.addr); process.exit(1); }}
    if (f.size !== 4096) {{ console.error("bad free size: " + f.size); process.exit(1); }}
    if (f.allocTimestampNs !== 5000000) {{ console.error("bad allocTimestampNs: " + f.allocTimestampNs); process.exit(1); }}

    console.log("OK: allocEvents=" + result.allocEvents.length + " freeEvents=" + result.freeEvents.length);
}}
main().catch((e) => {{ console.error(e); process.exit(1); }});
"#,
        viewer = std::path::Path::new(&manifest_dir)
            .parent()
            .unwrap()
            .join("dial9-viewer")
            .join("ui")
            .display(),
        trace = trace_path.display(),
    );

    let output = Command::new("node")
        .args(["-e", &script])
        .output()
        .expect("Failed to run node");

    eprintln!("{}", String::from_utf8_lossy(&output.stdout));
    assert!(
        output.status.success(),
        "JS parser alloc/free test failed:\nstdout: {}\nstderr: {}",
        String::from_utf8_lossy(&output.stdout),
        String::from_utf8_lossy(&output.stderr)
    );
}