testcontainers 0.27.3

A library for integration-testing against docker containers from within Rust.
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

// Socat can be useful if we would want to allow exposing not exposed ports of a container.
// But for now it's unused, so dead code is allowed.
#![allow(dead_code)]

use std::{borrow::Cow, collections::HashMap};

use itertools::Itertools;

use crate::{core::WaitFor, Image};

/// A socat container is used as a TCP proxy, enabling any port of another container to be exposed
/// publicly, even if that container does not make the port public itself.
pub(crate) struct Socat {
    targets: HashMap<u16, String>,
}

impl Socat {
    pub(crate) fn new() -> Self {
        Self {
            targets: HashMap::new(),
        }
    }

    pub(crate) fn add_target(&mut self, port: u16, value: String) {
        self.targets.insert(port, value);
    }
}

impl Image for Socat {
    fn name(&self) -> &str {
        "alpine/socat"
    }

    fn tag(&self) -> &str {
        "1.8.0.1"
    }

    fn ready_conditions(&self) -> Vec<WaitFor> {
        vec![WaitFor::healthcheck()]
    }

    fn entrypoint(&self) -> Option<&str> {
        Some("/bin/sh")
    }

    fn cmd(&self) -> impl IntoIterator<Item = impl Into<Cow<'_, str>>> {
        [
            "-c".to_string(),
            self.targets
                .iter()
                .map(|(port, value)| format!("socat TCP-LISTEN:{port},fork,reuseaddr TCP:{value}"))
                .join(" & "),
        ]
    }
}