From d8b3d2b6414f307bb166fe4f5d4aa16003007ce6 Mon Sep 17 00:00:00 2001 From: Vincent Richard Date: Sun, 17 Feb 2013 22:18:32 +0100 Subject: Added helper function to replace header field. --- tests/parser/headerTest.cpp | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'tests/parser/headerTest.cpp') diff --git a/tests/parser/headerTest.cpp b/tests/parser/headerTest.cpp index d096b28c..47bd0ebe 100644 --- a/tests/parser/headerTest.cpp +++ b/tests/parser/headerTest.cpp @@ -43,6 +43,8 @@ VMIME_TEST_SUITE_BEGIN VMIME_TEST(testInsertFieldAfter1) VMIME_TEST(testInsertFieldAfter2) + VMIME_TEST(testReplaceField) + VMIME_TEST(testRemoveField1) VMIME_TEST(testRemoveField2) @@ -192,6 +194,23 @@ VMIME_TEST_SUITE_BEGIN VASSERT_EQ("Third value", "C: c", headerTest::getFieldValue(*res[2])); } + // replaceField + void testReplaceField() + { + vmime::header hdr; + hdr.parse("A: a\r\nB: b\r\nC: c\r\n"); + + vmime::ref hf = vmime::headerFieldFactory::getInstance()->create("Z", "z"); + hdr.replaceField(hdr.getField("B"), hf); + + std::vector > res = hdr.getFieldList(); + + VASSERT_EQ("Count", static_cast (3), res.size()); + VASSERT_EQ("First value", "A: a", headerTest::getFieldValue(*res[0])); + VASSERT_EQ("Second value", "Z: z", headerTest::getFieldValue(*res[1])); + VASSERT_EQ("Third value", "C: c", headerTest::getFieldValue(*res[2])); + } + // removeField void testRemoveField1() { -- cgit v1.2.3