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);
}