capture-stdio 0.1.1

Capture stdin/out/err of the current process
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

//! Intercept [std::io::Stdout]

use super::PipedFd;
use crate::Capture;
use os_pipe::PipeReader;
use std::io::Error;

/// Intercept stdout
pub struct PipedStdout {
    internal: PipedFd,
}

impl Capture for PipedStdout {
    fn capture() -> Result<Self, Error> {
        let internal = PipedFd::capture(1, false)?;
        Ok(Self { internal })
    }

    fn restore(&mut self) {
        self.internal.restore();
    }
}

impl PipedStdout {
    /// Get reader of pipe
    pub fn get_reader(&mut self) -> &mut PipeReader {
        &mut self.internal.reader
    }
}

#[cfg(test)]
mod tests {
    use crate::pipe::stdout::PipedStdout;
    use crate::Capture;
    use std::io::{set_output_capture, BufRead, BufReader};

    #[test]
    fn test_stdout() {
        // stdout is captured by testing
        let original = set_output_capture(None);

        let mut piped_stdout = PipedStdout::capture().unwrap();
        let string = "Write something to stdout\n";
        print!("{}", string);

        set_output_capture(original);

        let mut output = String::new();
        let mut buf_reader = BufReader::new(piped_stdout.get_reader());
        buf_reader.read_line(&mut output).unwrap();

        assert_eq!(output, string);
        piped_stdout.restore();
    }
}