mod common;
use milter::*;
#[test]
fn add_header() {
let test_name = common::test_name(file!());
let miltertest = common::spawn_miltertest_runner(file!());
Milter::new("inet:3333@localhost")
.name(test_name.to_str().unwrap())
.on_eom(eom_callback)
.actions(Actions::ADD_HEADER)
.run()
.expect("milter execution failed");
let exit_code = miltertest.join().expect("panic in miltertest runner");
assert!(exit_code.success(), "miltertest returned error exit code");
}
#[on_eom(eom_callback)]
fn handle_eom(ctx: Context<()>) -> milter::Result<Status> {
ctx.api.add_header("Test-Name", "Test-Value")?;
Ok(Status::Continue)
}