     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

    use std::{io, io::prelude::Write};

    use crate::{
        console::ConsoleTestState,
        test_result::TestResult,
        time,
        types::{TestDesc, TestName},
    };

    mod json;
    mod pretty;
    mod terse;

    pub(crate) use self::json::JsonFormatter;
    pub(crate) use self::pretty::PrettyFormatter;
    pub(crate) use self::terse::TerseFormatter;

    pub(crate) trait OutputFormatter {
        fn write_run_start(&mut self, test_count: usize) -> io::Result<()>;
        fn write_test_start(&mut self, desc: &TestDesc) -> io::Result<()>;
        fn write_timeout(&mut self, desc: &TestDesc) -> io::Result<()>;
        fn write_result(
            &mut self,
            desc: &TestDesc,
            result: &TestResult,
            exec_time: Option<&time::TestExecTime>,
            stdout: &[u8],
            state: &ConsoleTestState,
        ) -> io::Result<()>;
        fn write_run_finish(&mut self, state: &ConsoleTestState) -> io::Result<bool>;
    }

    pub(crate) fn write_stderr_delimiter(test_output: &mut Vec<u8>, test_name: &TestName) {
        match test_output.last() {
            Some(b'\n') => (),
            Some(_) => test_output.push(b'\n'),
            None => (),
        }
        writeln!(test_output, "---- {} stderr ----", test_name).unwrap();
    }