hline 0.2.1

a grep-like tool that highlights lines in files
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

#![cfg(test)]
use crate::print;
use crate::print::Printer;
use std::cell::RefCell;
use std::fmt;
use termion::color;

#[derive(Default)]
pub(crate) struct MockPrinter {
    pub(crate) uncolored_messages: RefCell<Vec<String>>,
    pub(crate) colored_messages: RefCell<Vec<String>>,
    next_error: RefCell<Option<print::Error>>,
}

impl MockPrinter {
    pub(crate) fn fail_next(&mut self, error: print::Error) {
        self.next_error.replace(Some(error));
    }
}

impl Printer for &MockPrinter {
    fn print<S: fmt::Display>(&self, msg: S) -> print::Result {
        self.uncolored_messages.borrow_mut().push(msg.to_string());

        if self.next_error.borrow().is_some() {
            Err(self.next_error.replace(None).unwrap())
        } else {
            Ok(())
        }
    }

    fn colored_print<S: fmt::Display, C: color::Color>(
        &self,
        _color: color::Fg<C>,
        msg: S,
    ) -> print::Result {
        // Unfortunately, termion colors don't implement PartialEq, so checking for the exact color is not
        // feasible unless we wanted to write a wrapper, which I don't care enough to just for unit testing
        self.colored_messages.borrow_mut().push(msg.to_string());

        if self.next_error.borrow().is_some() {
            Err(self.next_error.replace(None).unwrap())
        } else {
            Ok(())
        }
    }
}

/// Similar to [`MockPrinter`], except that it only implements required methods
#[derive(Default)]
pub(crate) struct BarebonesMockPrinter {
    pub(crate) messages: RefCell<Vec<String>>,
}

impl Printer for BarebonesMockPrinter {
    fn print<S: fmt::Display>(&self, msg: S) -> print::Result {
        self.messages.borrow_mut().push(msg.to_string());
        Ok(())
    }
}