1 files changed, 101 insertions, 0 deletions
diff --git a/tests/parser/headerTest.cpp b/tests/parser/headerTest.cpp
new file mode 100644
index 00000000..b8eab349
--- /dev/null
+++ b/tests/parser/headerTest.cpp
@@ -0,0 +1,101 @@
+#include "../lib/unit++/unit++.h"
+
+#include <iostream>
+#include <ostream>
+
+#include "../../src/vmime"
+#include "../../examples/common.inc"
+
+using namespace unitpp;
+
+namespace
+{
+	class headerTest : public suite
+	{
+		static const std::string getFieldValue(const vmime::headerField& field)
+		{
+			std::ostringstream oss;
+			vmime::utility::outputStreamAdapter voss(oss);
+			field.generate(voss);
+
+			return (oss.str());
+		}
+
+		// getAllByType function tests
+		void testGetAllByType1()
+		{
+			vmime::header hdr;
+			hdr.parse("To: a\r\nFrom: b\r\n");
+
+			std::vector <vmime::headerField*> res = hdr.fields.findAllByType(vmime::headerField::Subject);
+
+			assert_eq("Count", (unsigned int) 0, res.size());
+		}
+
+		void testGetAllByType2()
+		{
+			vmime::header hdr;
+			hdr.parse("To: b\r\nTo : a\r\nFrom: c\r\n");
+
+			std::vector <vmime::headerField*> res = hdr.fields.findAllByType(vmime::headerField::To);
+
+			assert_eq("Count", (unsigned int) 2, res.size());
+			assert_eq("First value", "To: b", headerTest::getFieldValue(*res[0]));
+			assert_eq("First value", "To: a", headerTest::getFieldValue(*res[1]));
+		}
+
+		// getAllByName function tests
+		void testGetAllByName1()
+		{
+			vmime::header hdr;
+			hdr.parse("A: a1\nC: c1\n");
+
+			std::vector <vmime::headerField*> res = hdr.fields.findAllByName("B");
+
+			assert_eq("Count", (unsigned int) 0, res.size());
+		}
+
+		void testGetAllByName2()
+		{
+			vmime::header hdr;
+			hdr.parse("A: a1\nB: b1\nB: b2\nC: c1\n");
+
+			std::vector <vmime::headerField*> res = hdr.fields.findAllByName("B");
+
+			assert_eq("Count", (unsigned int) 2, res.size());
+			assert_eq("First value", "B: b1", headerTest::getFieldValue(*res[0]));
+			assert_eq("Second value", "B: b2", headerTest::getFieldValue(*res[1]));
+		}
+
+		void testGetAllByName3()
+		{
+			vmime::header hdr;
+			hdr.parse("A: a1\nB: b1\nB: b2\nC: c1\nC: c3\nC: c2\n");
+
+			std::vector <vmime::headerField*> res = hdr.fields.findAllByName("C");
+
+			assert_eq("Count", (unsigned int) 3, res.size());
+			assert_eq("First value", "C: c1", headerTest::getFieldValue(*res[0]));
+			assert_eq("Second value", "C: c3", headerTest::getFieldValue(*res[1]));
+			assert_eq("Second value", "C: c2", headerTest::getFieldValue(*res[2]));
+		}
+
+	public:
+
+		headerTest() : suite("vmime::header")
+		{
+			// VMime initialization
+			vmime::platformDependant::setHandler<my_handler>();
+
+			add("GetAllByType", testcase(this, "GetAllByType1", &headerTest::testGetAllByType1));
+			add("GetAllByType", testcase(this, "GetAllByType2", &headerTest::testGetAllByType2));
+			add("GetAllByName", testcase(this, "GetAllByName1", &headerTest::testGetAllByName1));
+			add("GetAllByName", testcase(this, "GetAllByName2", &headerTest::testGetAllByName2));
+			add("GetAllByName", testcase(this, "GetAllByName3", &headerTest::testGetAllByName3));
+			suite::main().add("vmime::header", this);
+		}
+
+	};
+
+	headerTest* theTest = new headerTest();
+}