tva 0.3.3

Tab-separated Values Assistant
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

use assert_cmd::assert::Assert;
use assert_cmd::cargo::cargo_bin_cmd;
use assert_cmd::Command;

pub struct TvaCmd {
    cmd: Command,
    stdin: Option<String>,
}

impl TvaCmd {
    pub fn new() -> Self {
        let mut cmd = cargo_bin_cmd!("tva");
        cmd.env("RUST_BACKTRACE", "1");
        Self { cmd, stdin: None }
    }

    #[allow(dead_code)]
    pub fn stdin<S: Into<String>>(mut self, input: S) -> Self {
        self.stdin = Some(input.into());
        self
    }

    #[allow(dead_code)]
    pub fn args(mut self, args: &[&str]) -> Self {
        self.cmd.args(args);
        self
    }

    #[allow(dead_code)]
    pub fn current_dir<P: AsRef<std::path::Path>>(mut self, dir: P) -> Self {
        self.cmd.current_dir(dir);
        self
    }

    pub fn run(mut self) -> (String, String) {
        if let Some(input) = self.stdin {
            self.cmd.write_stdin(input);
        }

        let output = self.cmd.output().expect("Failed to execute command");

        let stdout = String::from_utf8(output.stdout).expect("Invalid UTF-8 in stdout");
        let stderr = String::from_utf8(output.stderr).expect("Invalid UTF-8 in stderr");

        (stdout, stderr)
    }

    #[allow(dead_code)]
    pub fn assert(mut self) -> Assert {
        if let Some(input) = self.stdin {
            self.cmd.write_stdin(input);
        }
        self.cmd.assert()
    }

    #[allow(dead_code)]
    pub fn run_fail(self) -> (String, String) {
        let assert = self.assert().failure();
        let output = assert.get_output();
        let stdout =
            String::from_utf8(output.stdout.clone()).expect("Invalid UTF-8 in stdout");
        let stderr =
            String::from_utf8(output.stderr.clone()).expect("Invalid UTF-8 in stderr");
        (stdout, stderr)
    }
}

#[allow(dead_code)]
pub fn assert_close(actual: f64, expected: f64, epsilon: f64) {
    if (actual - expected).abs() > epsilon {
        panic!(
            "Assertion failed: {} is not close to {} (epsilon: {}, diff: {})",
            actual,
            expected,
            epsilon,
            (actual - expected).abs()
        );
    }
}