testcontainers-rs 0.0.1

rust testcontainers
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

use futures::{
    stream::{BoxStream, Stream},
    StreamExt,
};
use std::{fmt, io};

/// WaitError describes errors when waiting for a log line
#[derive(thiserror::Error, Debug)]
pub enum WaitError {
    /// End of stream without matching log line.
    /// Contains all received lines of the log.
    #[error("EOF without matching log line")]
    EOF(Vec<String>),

    /// IO error while reading from the stream
    #[error("IO error: {0}")]
    Io(
        #[from]
        #[source]
        io::Error,
    ),
}

pub struct LogStream<'s> {
    inner: BoxStream<'s, Result<String, io::Error>>,
}

impl<'s> fmt::Debug for LogStream<'s> {
    #[inline]
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("LogStream").finish()
    }
}

impl<'s> LogStream<'s> {
    #[inline]
    // pub fn new(stream: BoxStream<'d, Result<String, io::Error>>) -> Self {
    pub fn new(stream: impl Stream<Item = Result<String, io::Error>> + 's + Send) -> Self {
        Self {
            inner: stream.boxed(),
        }
    }

    #[inline]
    pub async fn wait_for_message(mut self, message: &str) -> Result<(), WaitError> {
        let mut lines = vec![];

        while let Some(line) = self.inner.next().await.transpose()? {
            // TODO: use regex here
            if line.contains(message) {
                return Ok(());
            }
            lines.push(line);
        }

        Err(WaitError::EOF(lines))
    }
}