rustcmdpev 0.2.0

Visualize PostgreSQL EXPLAIN JSON output in a terminal-friendly tree
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

use std::io::Write;
use std::process::{Command, Output, Stdio};

fn run_with_args_and_stdin(args: &[&str], stdin_payload: &str, no_color: bool) -> Output {
    let mut cmd = Command::new(env!("CARGO_BIN_EXE_rustcmdpev"));
    cmd.args(args)
        .stdin(Stdio::piped())
        .stdout(Stdio::piped())
        .stderr(Stdio::piped());

    if no_color {
        cmd.env("NO_COLOR", "1");
    }

    let mut child = cmd.spawn().expect("failed to spawn rustcmdpev");

    {
        let stdin = child.stdin.as_mut().expect("failed to open stdin");
        stdin
            .write_all(stdin_payload.as_bytes())
            .expect("failed to write stdin");
    }

    child.wait_with_output().expect("failed to wait on child")
}

#[test]
fn color_always_emits_ansi_sequences() {
    let output = run_with_args_and_stdin(
        &["--color", "always", "--format", "pretty"],
        "[{\"Plan\":{\"Node Type\":\"Seq Scan\"}}]",
        false,
    );
    assert!(output.status.success());
    let stdout = String::from_utf8_lossy(&output.stdout);
    assert!(stdout.contains("\u{1b}["));
}

#[test]
fn color_never_disables_ansi_sequences() {
    let output = run_with_args_and_stdin(
        &["--color", "never", "--format", "pretty"],
        "[{\"Plan\":{\"Node Type\":\"Seq Scan\"}}]",
        false,
    );
    assert!(output.status.success());
    let stdout = String::from_utf8_lossy(&output.stdout);
    assert!(!stdout.contains("\u{1b}["));
}

#[test]
fn color_auto_respects_no_color() {
    let output = run_with_args_and_stdin(
        &["--color", "auto", "--format", "pretty"],
        "[{\"Plan\":{\"Node Type\":\"Seq Scan\"}}]",
        true,
    );
    assert!(output.status.success());
    let stdout = String::from_utf8_lossy(&output.stdout);
    assert!(!stdout.contains("\u{1b}["));
}