65b44c3b3a
* src/engine/rfc822/rfc822-part.vala (Part): Add a blacklist for text types that shouldn't have CRLF's stripped. Add unit tests.
102 lines
3.6 KiB
Vala
102 lines
3.6 KiB
Vala
/*
|
|
* Copyright 2018 Michael Gratton <mike@vee.net>
|
|
*
|
|
* This software is licensed under the GNU Lesser General Public License
|
|
* (version 2.1 or later). See the COPYING file in this distribution.
|
|
*/
|
|
|
|
class Geary.RFC822.PartTest : TestCase {
|
|
|
|
private const string CR_BODY = "This is an attachment.\n";
|
|
private const string CRLF_BODY = "This is an attachment.\r\n";
|
|
private const string ICAL_BODY = "BEGIN:VCALENDAR\r\nEND:VCALENDAR\r\n";
|
|
|
|
|
|
public PartTest() {
|
|
base("Geary.RFC822.PartTest");
|
|
add_test("new_from_empty_mime_part", new_from_empty_mime_part);
|
|
add_test("new_from_complete_mime_part", new_from_complete_mime_part);
|
|
add_test("write_to_buffer_plain", write_to_buffer_plain);
|
|
add_test("write_to_buffer_plain_crlf", write_to_buffer_plain_crlf);
|
|
add_test("write_to_buffer_plain_ical", write_to_buffer_plain_ical);
|
|
}
|
|
|
|
public void new_from_empty_mime_part() throws Error {
|
|
GMime.Part part = new_part(null, CR_BODY.data);
|
|
part.set_header("Content-Type", "");
|
|
|
|
Part test = new Part(part);
|
|
|
|
assert_null(test.content_type, "content_type");
|
|
assert_null_string(test.content_id, "content_id");
|
|
assert_null_string(test.content_description, "content_description");
|
|
assert_null(test.content_disposition, "content_disposition");
|
|
}
|
|
|
|
public void new_from_complete_mime_part() throws Error {
|
|
const string TYPE = "text/plain";
|
|
const string ID = "test-id";
|
|
const string DESC = "test description";
|
|
|
|
GMime.Part part = new_part(TYPE, CR_BODY.data);
|
|
part.set_content_id(ID);
|
|
part.set_content_description(DESC);
|
|
part.set_content_disposition(
|
|
new GMime.ContentDisposition.from_string("inline")
|
|
);
|
|
|
|
Part test = new Part(part);
|
|
|
|
assert_string(TYPE, test.content_type.to_string());
|
|
assert_string(ID, test.content_id);
|
|
assert_string(DESC, test.content_description);
|
|
assert_non_null(test.content_disposition, "content_disposition");
|
|
assert_int(
|
|
Geary.Mime.DispositionType.INLINE,
|
|
test.content_disposition.disposition_type
|
|
);
|
|
}
|
|
|
|
public void write_to_buffer_plain() throws Error {
|
|
Part test = new Part(new_part("text/plain", CR_BODY.data));
|
|
|
|
Memory.Buffer buf = test.write_to_buffer();
|
|
|
|
assert_string(CR_BODY, buf.to_string());
|
|
}
|
|
|
|
public void write_to_buffer_plain_crlf() throws Error {
|
|
Part test = new Part(new_part("text/plain", CRLF_BODY.data));
|
|
|
|
Memory.Buffer buf = test.write_to_buffer();
|
|
|
|
// CRLF should be stripped
|
|
assert_string(CR_BODY, buf.to_string());
|
|
}
|
|
|
|
public void write_to_buffer_plain_ical() throws Error {
|
|
Part test = new Part(new_part("text/calendar", ICAL_BODY.data));
|
|
|
|
Memory.Buffer buf = test.write_to_buffer();
|
|
|
|
// CRLF should not be stripped
|
|
assert_string(ICAL_BODY, buf.to_string());
|
|
}
|
|
|
|
private GMime.Part new_part(string? mime_type,
|
|
uint8[] body,
|
|
GMime.ContentEncoding encoding = GMime.ContentEncoding.DEFAULT) {
|
|
GMime.Part part = new GMime.Part();
|
|
if (mime_type != null) {
|
|
part.set_content_type(new GMime.ContentType.from_string(mime_type));
|
|
}
|
|
GMime.DataWrapper body_wrapper = new GMime.DataWrapper.with_stream(
|
|
new GMime.StreamMem.with_buffer(body),
|
|
encoding
|
|
);
|
|
part.set_content_object(body_wrapper);
|
|
part.encode(GMime.EncodingConstraint.7BIT);
|
|
return part;
|
|
}
|
|
|
|
}
|