dovecot: Added message-parser unit test.

[dovecot at dovecot.org]

details:   http://hg.dovecot.org/dovecot/rev/c49084d2148e
changeset: 6727:c49084d2148e
user:      Timo Sirainen <tss at iki.fi>
date:      Thu Nov 08 03:08:03 2007 +0200
description:
Added message-parser unit test.

diffstat:

1 file changed, 118 insertions(+)
src/tests/test-mail.c |  118 +++++++++++++++++++++++++++++++++++++++++++++++++

diffs (143 lines):

diff -r 1a551d95ac9c -r c49084d2148e src/tests/test-mail.c
--- a/src/tests/test-mail.c	Thu Nov 08 03:07:30 2007 +0200
+++ b/src/tests/test-mail.c	Thu Nov 08 03:08:03 2007 +0200
@@ -2,9 +2,44 @@
 
 #include "lib.h"
 #include "str.h"
+#include "istream-internal.h"
 #include "message-address.h"
 #include "message-date.h"
+#include "message-parser.h"
 #include "test-common.h"
+
+static const char test_msg[] =
+"Return-Path: <test at example.org>\n"
+"Subject: Hello world\n"
+"From: Test User <test at example.org>\n"
+"To: Another User <test2 at example.org>\n"
+"Message-Id: <1.2.3.4 at example>\n"
+"Mime-Version: 1.0\n"
+"Date: Sun, 23 May 2007 04:58:08 +0300\n"
+"Content-Type: multipart/signed; micalg=pgp-sha1;\n"
+"	protocol=\"application/pgp-signature\";\n"
+"	boundary=\"=-GNQXLhuj24Pl1aCkk4/d\"\n"
+"\n"
+"--=-GNQXLhuj24Pl1aCkk4/d\n"
+"Content-Type: text/plain\n"
+"Content-Transfer-Encoding: quoted-printable\n"
+"\n"
+"There was a day=20\n"
+"a happy=20day\n"
+"\n"
+"--=-GNQXLhuj24Pl1aCkk4/d\n"
+"Content-Type: application/pgp-signature; name=signature.asc\n"
+"\n"
+"-----BEGIN PGP SIGNATURE-----\n"
+"Version: GnuPG v1.2.4 (GNU/Linux)\n"
+"\n"
+"invalid\n"
+"-----END PGP SIGNATURE-----\n"
+"\n"
+"--=-GNQXLhuj24Pl1aCkk4/d--\n"
+"\n"
+"\n";
+#define TEST_MSG_LEN (sizeof(test_msg)-1)
 
 static bool cmp_addr(const struct message_address *a1,
 		     const struct message_address *a2)
@@ -137,11 +172,94 @@ static void test_message_date_parse(void
 	}
 }
 
+static bool msg_parts_cmp(struct message_part *p1, struct message_part *p2)
+{
+	while (p1 != NULL || p2 != NULL) {
+		if ((p1 != NULL) != (p2 != NULL))
+			return FALSE;
+		if ((p1->children != NULL) != (p2->children != NULL))
+			return FALSE;
+
+		if (p1->children) {
+			if (!msg_parts_cmp(p1->children, p2->children))
+				return FALSE;
+		}
+
+		if (p1->physical_pos != p2->physical_pos ||
+		    p1->header_size.physical_size != p2->header_size.physical_size ||
+		    p1->header_size.virtual_size != p2->header_size.virtual_size ||
+		    p1->header_size.lines != p2->header_size.lines ||
+		    p1->body_size.physical_size != p2->body_size.physical_size ||
+		    p1->body_size.virtual_size != p2->body_size.virtual_size ||
+		    p1->body_size.lines != p2->body_size.lines ||
+		    p1->flags != p2->flags)
+			return FALSE;
+
+		p1 = p1->next;
+		p2 = p2->next;
+	}
+	return TRUE;
+}
+
+static ssize_t test_read(struct istream_private *stream)
+{
+	if (stream->pos < TEST_MSG_LEN)
+		return 0;
+
+	stream->istream.eof = TRUE;
+	return -1;
+}
+
+static void test_message_parser(void)
+{
+	struct message_parser_ctx *parser;
+	struct istream *input;
+	struct message_part *parts, *parts2;
+	struct message_block block;
+	unsigned int i;
+	bool success = TRUE;
+	pool_t pool;
+	int ret;
+
+	pool = pool_alloconly_create("message parser", 10240);
+	input = i_stream_create_from_data(test_msg, TEST_MSG_LEN);
+
+	parser = message_parser_init(pool, input, 0, 0);
+	while ((ret = message_parser_parse_next_block(parser, &block)) > 0) ;
+	i_assert(ret < 0);
+	parts = message_parser_deinit(&parser);
+	i_stream_unref(&input);
+
+	input = i_stream_create_from_data(test_msg, TEST_MSG_LEN);
+	input->blocking = FALSE;
+	input->real_stream->read = test_read;
+
+	parser = message_parser_init(pool, input, 0, 0);
+	for (i = 1; i <= TEST_MSG_LEN; i++) {
+		input->real_stream->pos = i;
+		while ((ret = message_parser_parse_next_block(parser,
+							      &block)) > 0) ;
+		if (ret < 0 && i < TEST_MSG_LEN) {
+			success = FALSE;
+			break;
+		}
+	}
+	parts2 = message_parser_deinit(&parser);
+	i_stream_unref(&input);
+
+	if (!msg_parts_cmp(parts, parts2))
+		success = FALSE;
+
+	pool_unref(&pool);
+	test_out("message_parser()", success);
+}
+
 int main(void)
 {
 	test_init();
 
 	test_message_address();
 	test_message_date_parse();
+	test_message_parser();
 	return test_deinit();
 }