50 lines
1.6 KiB
Rust
50 lines
1.6 KiB
Rust
use melib;
|
|
use ui::mime_apps;
|
|
|
|
use melib::email::Draft;
|
|
use mime_apps::query_mime_info;
|
|
|
|
#[test]
|
|
fn build_draft() {
|
|
let mut new_draft = Draft::default();
|
|
let mut attachment = melib::email::attachment_from_file(&"./tests/test_image.gif")
|
|
.expect("Could not open test_image.gif.");
|
|
if let Ok(mime_type) = query_mime_info("./tests/test_image.gif") {
|
|
match attachment.content_type {
|
|
melib::email::ContentType::Other { ref mut tag, .. } => {
|
|
*tag = mime_type;
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
if new_draft.headers().contains_key("User-Agent") {
|
|
new_draft.headers_mut().remove("User-Agent");
|
|
let pos = new_draft
|
|
.header_order
|
|
.iter()
|
|
.position(|k| k == "User-Agent")
|
|
.unwrap();
|
|
new_draft.header_order.remove(pos);
|
|
}
|
|
{
|
|
new_draft.headers_mut().remove("Date");
|
|
let pos = new_draft
|
|
.header_order
|
|
.iter()
|
|
.position(|k| k == "Date")
|
|
.unwrap();
|
|
new_draft.header_order.remove(pos);
|
|
}
|
|
|
|
new_draft.attachments_mut().push(attachment);
|
|
new_draft.set_body("hello world.".to_string());
|
|
let raw = new_draft.finalise().expect("could not finalise draft");
|
|
let boundary_def = raw.find("bzz_bzz__bzz__").unwrap();
|
|
let boundary_end = boundary_def + raw[boundary_def..].find("\"").unwrap();
|
|
let boundary = raw[boundary_def..boundary_end].to_string();
|
|
let boundary_str = &boundary["bzz_bzz__bzz__".len()..];
|
|
|
|
let raw = raw.replace(boundary_str, "");
|
|
assert_eq!(include_str!("generated.mail"), &raw);
|
|
}
|