milter 0.2.4

Bindings to the sendmail milter library
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
61
62
63
64
65
66
67
68
69
70
71
72
73

//! Common integration test utilities.
//!
//! `miltertest` is spawned in a separate thread (not main), rather than have
//! the actual `Milter` be in a spawned thread and `miltertest` in main: just as
//! it is outside of tests.

use std::{
    ffi::OsString,
    path::PathBuf,
    process::{Command, ExitStatus},
    thread::{self, JoinHandle},
    time::Duration,
};

// Integration tests require the `miltertest` program.
const MILTERTEST: &str = "miltertest";

pub fn test_name(test_file_name: &str) -> OsString {
    PathBuf::from(test_file_name)
        .as_path()
        .file_stem()
        .expect("no file stem")
        .to_os_string()
}

pub fn spawn_miltertest_runner(test_file_name: &str) -> JoinHandle<ExitStatus> {
    let _timeout_thread = thread::spawn(|| {
        thread::sleep(Duration::from_secs(15));

        eprintln!("miltertest runner timed out");

        milter::shutdown();
    });

    let file_name = to_miltertest_file_name(test_file_name);

    thread::spawn(move || {
        thread::sleep(Duration::from_millis(100));

        let output = Command::new(MILTERTEST)
            .arg("-s")
            .arg(&file_name)
            .output()
            .expect("miltertest execution failed");

        print_output_stream("STDOUT", output.stdout);
        print_output_stream("STDERR", output.stderr);

        milter::shutdown();

        output.status
    })
}

fn to_miltertest_file_name(file_name: &str) -> OsString {
    let mut path = PathBuf::from(file_name);
    path.set_extension("lua");
    path.into_os_string()
}

fn print_output_stream(name: &str, output: Vec<u8>) {
    if !output.is_empty() {
        let output = String::from_utf8(output).unwrap();

        eprintln!("{}:", name);

        if output.ends_with('\n') {
            eprint!("{}", &output)
        } else {
            eprintln!("{}", &output)
        }
    }
}