beenull/himalaya
1
0
Fork 0
mirror of https://github.com/soywod/himalaya.git synced 2024-07-05 17:15:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

add tests for --header arg of read command (#338)

Browse source
This commit is contained in:
Clément DOUIN 2022-03-12 13:57:27 +01:00
parent 86b41e4914
commit d3968461e2
No known key found for this signature in database
GPG key ID: 353E4A18EE0FAB72
1 changed files with 76 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

79
src/msg/msg_entity.rs
View file

@ -710,7 +710,7 @@ impl Msg {
}
key => {
msg.headers.insert(
key.to_owned(),
key.to_lowercase(),
rfc2047_decoder::decode(val.as_bytes()).unwrap_or(val),
);
}
@ -786,13 +786,17 @@ impl Msg {
},
key => match self.headers.get(key) {
Some(ref val) if !val.is_empty() => {
readable_msg.push_str(&format!("{}: {}\n", key.to_case(Case::Pascal), val));
readable_msg.push_str(&format!("{}: {}\n", key.to_case(Case::Train), val));
}
_ => (),
},
};
}
readable_msg.push_str("\n");
if !readable_msg.is_empty() {
readable_msg.push_str("\n");
}
readable_msg.push_str(&self.fold_text_parts(text_mime));
Ok(readable_msg)
}
@ -829,6 +833,8 @@ impl TryInto<lettre::address::Envelope> for &Msg {
#[cfg(test)]
mod tests {
use std::iter::FromIterator;
use mailparse::SingleInfo;
use crate::msg::Addr;
@ -973,4 +979,71 @@ mod tests {
"\"Sender 1\" <test-sender-1@local>, \"Sender 2\" <test-sender-2@local>"
);
}
#[test]
fn test_to_readable() {
let msg = Msg {
parts: Parts(vec![Part::TextPlain(TextPlainPart {
content: String::from("hello, world!"),
})]),
..Msg::default()
};
// empty msg, empty headers
assert_eq!(
"hello, world!",
msg.to_readable_string("plain", vec![]).unwrap()
);
// empty msg, basic headers
assert_eq!(
"hello, world!",
msg.to_readable_string("plain", vec!["from", "date", "custom-header"])
.unwrap()
);
// empty msg, subject header
assert_eq!(
"Subject: \n\nhello, world!",
msg.to_readable_string("plain", vec!["subject"]).unwrap()
);
let msg = Msg {
headers: HashMap::from_iter([("custom-header".into(), "custom value".into())]),
message_id: Some("<message-id>".into()),
from: Some(
vec![Addr::Single(SingleInfo {
addr: "test@local".into(),
display_name: Some("Test".into()),
})]
.into(),
),
cc: Some(vec![].into()),
parts: Parts(vec![Part::TextPlain(TextPlainPart {
content: String::from("hello, world!"),
})]),
..Msg::default()
};
// header present in msg headers
assert_eq!(
"From: \"Test\" <test@local>\n\nhello, world!",
msg.to_readable_string("plain", vec!["from"]).unwrap()
);
// header present but empty in msg headers
assert_eq!(
"hello, world!",
msg.to_readable_string("plain", vec!["cc"]).unwrap()
);
// custom header present in msg headers
assert_eq!(
"Custom-Header: custom value\n\nhello, world!",
msg.to_readable_string("plain", vec!["custom-header"])
.unwrap()
);
// custom header present in msg headers (case insensitivity)
assert_eq!(
"Custom-Header: custom value\n\nhello, world!",
msg.to_readable_string("plain", vec!["CUSTOM-hEaDer"])
.unwrap()
);
}
}