testdata-macros 0.1.0

File-based testing helper -- subpackage for proc-macro
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

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

pub(crate) fn rustfmt(s: &str) -> io::Result<String> {
    let mut rustfmt = Command::new("rustfmt")
        .args(&["--edition", "2021"])
        .args(&["--color", "never"])
        .stdin(Stdio::piped())
        .stdout(Stdio::piped())
        .spawn()?;

    let rustfmt_stdin = rustfmt.stdin.take().unwrap();
    let mut rustfmt_stdin = BufWriter::new(rustfmt_stdin);
    writeln!(rustfmt_stdin, "{}\n", s)?;
    rustfmt_stdin.flush()?;
    drop(rustfmt_stdin);

    let rustfmt_output = rustfmt.wait_with_output()?;
    if !rustfmt_output.status.success() {
        return Err(io::Error::new(
            io::ErrorKind::Other,
            format!("rustfmt failed: {:?}", rustfmt_output),
        ));
    }
    let src = String::from_utf8_lossy(&rustfmt_output.stdout).into_owned();
    Ok(src)
}

#[macro_export]
macro_rules! assert_ts_eq {
    ($lhs:expr, $rhs:expr) => {
        match ($lhs, $rhs) {
            (lhs, rhs) => {
                let lhs = lhs.to_string();
                let rhs = rhs.to_string();
                if lhs != rhs {
                    if let (Ok(lhs), Ok(rhs)) = (
                        $crate::testing::rustfmt(&lhs),
                        $crate::testing::rustfmt(&rhs),
                    ) {
                        pretty_assertions::assert_eq!(lhs, rhs);
                    }
                    pretty_assertions::assert_eq!(lhs, rhs);
                    unreachable!();
                }
            }
        }
    };
}