ferencd@0: // ferencd@0: // VMime library (http://www.vmime.org) ferencd@0: // Copyright (C) 2002-2013 Vincent Richard ferencd@0: // ferencd@0: // This program is free software; you can redistribute it and/or ferencd@0: // modify it under the terms of the GNU General Public License as ferencd@0: // published by the Free Software Foundation; either version 3 of ferencd@0: // the License, or (at your option) any later version. ferencd@0: // ferencd@0: // This program is distributed in the hope that it will be useful, ferencd@0: // but WITHOUT ANY WARRANTY; without even the implied warranty of ferencd@0: // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU ferencd@0: // General Public License for more details. ferencd@0: // ferencd@0: // You should have received a copy of the GNU General Public License along ferencd@0: // with this program; if not, write to the Free Software Foundation, Inc., ferencd@0: // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. ferencd@0: // ferencd@0: // Linking this library statically or dynamically with other modules is making ferencd@0: // a combined work based on this library. Thus, the terms and conditions of ferencd@0: // the GNU General Public License cover the whole combination. ferencd@0: // ferencd@0: ferencd@0: #include "tests/testUtils.hpp" ferencd@0: ferencd@0: ferencd@0: VMIME_TEST_SUITE_BEGIN(headerTest) ferencd@0: ferencd@0: VMIME_TEST_LIST_BEGIN ferencd@0: VMIME_TEST(testHas1) ferencd@0: VMIME_TEST(testHas2) ferencd@0: ferencd@0: VMIME_TEST(testAppend1) ferencd@0: VMIME_TEST(testAppend2) ferencd@0: ferencd@0: VMIME_TEST(testInsertFieldBefore1) ferencd@0: VMIME_TEST(testInsertFieldBefore2) ferencd@0: ferencd@0: VMIME_TEST(testInsertFieldAfter1) ferencd@0: VMIME_TEST(testInsertFieldAfter2) ferencd@0: ferencd@0: VMIME_TEST(testReplaceField) ferencd@0: ferencd@0: VMIME_TEST(testRemoveField1) ferencd@0: VMIME_TEST(testRemoveField2) ferencd@0: ferencd@0: VMIME_TEST(testRemoveAllFields) ferencd@0: ferencd@0: VMIME_TEST(testgetFieldCount) ferencd@0: ferencd@0: VMIME_TEST(testIsEmpty1) ferencd@0: VMIME_TEST(testIsEmpty2) ferencd@0: ferencd@0: VMIME_TEST(testGetFieldAt) ferencd@0: ferencd@0: VMIME_TEST(testGetFieldList1) ferencd@0: VMIME_TEST(testGetFieldList2) ferencd@0: ferencd@0: VMIME_TEST(testFind1) ferencd@0: ferencd@0: VMIME_TEST(testFindAllFields1) ferencd@0: VMIME_TEST(testFindAllFields2) ferencd@0: VMIME_TEST(testFindAllFields3) ferencd@0: VMIME_TEST_LIST_END ferencd@0: ferencd@0: ferencd@0: static const std::string getFieldValue(const vmime::headerField& field) ferencd@0: { ferencd@0: std::ostringstream oss; ferencd@0: vmime::utility::outputStreamAdapter voss(oss); ferencd@0: field.generate(voss); ferencd@0: ferencd@0: return (oss.str()); ferencd@0: } ferencd@0: ferencd@0: // has function tests ferencd@0: void testHas1() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("From: x\r\nTo: y\r\nTo: z\r\n"); ferencd@0: ferencd@0: bool res = hdr.hasField("Z"); ferencd@0: ferencd@0: VASSERT_EQ("Value", false, res); ferencd@0: } ferencd@0: ferencd@0: void testHas2() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("X: x\r\nTo: y\r\nTo: z\r\n"); ferencd@0: ferencd@0: bool res = hdr.hasField("To"); ferencd@0: ferencd@0: VASSERT_EQ("Value", true, res); ferencd@0: } ferencd@0: ferencd@0: // appendField function tests ferencd@0: void testAppend1() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse(""); ferencd@0: ferencd@0: vmime::shared_ptr hf = vmime::headerFieldFactory::getInstance()->create("A", "a"); ferencd@0: hdr.appendField(hf); ferencd@0: ferencd@0: std::vector > res = hdr.getFieldList(); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (1), res.size()); ferencd@0: VASSERT_EQ("First value", "A: a", headerTest::getFieldValue(*res[0])); ferencd@0: } ferencd@0: ferencd@0: void testAppend2() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("A: a\r\n"); ferencd@0: ferencd@0: vmime::shared_ptr hf = vmime::headerFieldFactory::getInstance()->create("B", "b"); ferencd@0: hdr.appendField(hf); ferencd@0: ferencd@0: std::vector > res = hdr.getFieldList(); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (2), res.size()); ferencd@0: VASSERT_EQ("First value", "A: a", headerTest::getFieldValue(*res[0])); ferencd@0: VASSERT_EQ("Second value", "B: b", headerTest::getFieldValue(*res[1])); ferencd@0: } ferencd@0: ferencd@0: // insertFieldBefore ferencd@0: void testInsertFieldBefore1() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("A: a\r\nC: c\r\n"); ferencd@0: ferencd@0: vmime::shared_ptr hf = vmime::headerFieldFactory::getInstance()->create("B", "b"); ferencd@0: hdr.insertFieldBefore(hdr.getField("C"), hf); ferencd@0: ferencd@0: std::vector > res = hdr.getFieldList(); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (3), res.size()); ferencd@0: VASSERT_EQ("First value", "A: a", headerTest::getFieldValue(*res[0])); ferencd@0: VASSERT_EQ("Second value", "B: b", headerTest::getFieldValue(*res[1])); ferencd@0: VASSERT_EQ("Third value", "C: c", headerTest::getFieldValue(*res[2])); ferencd@0: } ferencd@0: ferencd@0: void testInsertFieldBefore2() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("A: a\r\nC: c\r\n"); ferencd@0: ferencd@0: vmime::shared_ptr hf = vmime::headerFieldFactory::getInstance()->create("B", "b"); ferencd@0: hdr.insertFieldBefore(1, hf); ferencd@0: ferencd@0: std::vector > res = hdr.getFieldList(); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (3), res.size()); ferencd@0: VASSERT_EQ("First value", "A: a", headerTest::getFieldValue(*res[0])); ferencd@0: VASSERT_EQ("Second value", "B: b", headerTest::getFieldValue(*res[1])); ferencd@0: VASSERT_EQ("Third value", "C: c", headerTest::getFieldValue(*res[2])); ferencd@0: } ferencd@0: ferencd@0: // insertFieldAfter ferencd@0: void testInsertFieldAfter1() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("A: a\r\nC: c\r\n"); ferencd@0: ferencd@0: vmime::shared_ptr hf = vmime::headerFieldFactory::getInstance()->create("B", "b"); ferencd@0: hdr.insertFieldAfter(hdr.getField("A"), hf); ferencd@0: ferencd@0: std::vector > res = hdr.getFieldList(); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (3), res.size()); ferencd@0: VASSERT_EQ("First value", "A: a", headerTest::getFieldValue(*res[0])); ferencd@0: VASSERT_EQ("Second value", "B: b", headerTest::getFieldValue(*res[1])); ferencd@0: VASSERT_EQ("Third value", "C: c", headerTest::getFieldValue(*res[2])); ferencd@0: } ferencd@0: ferencd@0: void testInsertFieldAfter2() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("A: a\r\nC: c\r\n"); ferencd@0: ferencd@0: vmime::shared_ptr hf = vmime::headerFieldFactory::getInstance()->create("B", "b"); ferencd@0: hdr.insertFieldAfter(0, hf); ferencd@0: ferencd@0: std::vector > res = hdr.getFieldList(); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (3), res.size()); ferencd@0: VASSERT_EQ("First value", "A: a", headerTest::getFieldValue(*res[0])); ferencd@0: VASSERT_EQ("Second value", "B: b", headerTest::getFieldValue(*res[1])); ferencd@0: VASSERT_EQ("Third value", "C: c", headerTest::getFieldValue(*res[2])); ferencd@0: } ferencd@0: ferencd@0: // replaceField ferencd@0: void testReplaceField() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("A: a\r\nB: b\r\nC: c\r\n"); ferencd@0: ferencd@0: vmime::shared_ptr hf = vmime::headerFieldFactory::getInstance()->create("Z", "z"); ferencd@0: hdr.replaceField(hdr.getField("B"), hf); ferencd@0: ferencd@0: std::vector > res = hdr.getFieldList(); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (3), res.size()); ferencd@0: VASSERT_EQ("First value", "A: a", headerTest::getFieldValue(*res[0])); ferencd@0: VASSERT_EQ("Second value", "Z: z", headerTest::getFieldValue(*res[1])); ferencd@0: VASSERT_EQ("Third value", "C: c", headerTest::getFieldValue(*res[2])); ferencd@0: } ferencd@0: ferencd@0: // removeField ferencd@0: void testRemoveField1() ferencd@0: { ferencd@0: vmime::header hdr1, hdr2; ferencd@0: hdr1.parse("A: a\r\nB: b\r\nC: c\r\n"); ferencd@0: hdr2.parse("A: a\r\nB: b\r\nC: c\r\n"); ferencd@0: ferencd@0: hdr1.removeField(hdr1.getField("B")); ferencd@0: hdr2.removeField(1); ferencd@0: ferencd@0: std::vector > res1 = hdr1.getFieldList(); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (2), res1.size()); ferencd@0: VASSERT_EQ("First value", "A: a", headerTest::getFieldValue(*res1[0])); ferencd@0: VASSERT_EQ("Second value", "C: c", headerTest::getFieldValue(*res1[1])); ferencd@0: ferencd@0: std::vector > res2 = hdr2.getFieldList(); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (2), res2.size()); ferencd@0: VASSERT_EQ("First value", "A: a", headerTest::getFieldValue(*res2[0])); ferencd@0: VASSERT_EQ("Second value", "C: c", headerTest::getFieldValue(*res2[1])); ferencd@0: } ferencd@0: ferencd@0: void testRemoveField2() ferencd@0: { ferencd@0: vmime::header hdr1, hdr2; ferencd@0: hdr1.parse("A: a\r\n"); ferencd@0: hdr2.parse("A: a\r\n"); ferencd@0: ferencd@0: hdr1.removeField(hdr1.getField("A")); ferencd@0: hdr2.removeField(0); ferencd@0: ferencd@0: std::vector > res1 = hdr1.getFieldList(); ferencd@0: VASSERT_EQ("Count", static_cast (0), res1.size()); ferencd@0: ferencd@0: std::vector > res2 = hdr2.getFieldList(); ferencd@0: VASSERT_EQ("Count", static_cast (0), res2.size()); ferencd@0: } ferencd@0: ferencd@0: // removeAllFields ferencd@0: void testRemoveAllFields() ferencd@0: { ferencd@0: vmime::header hdr1, hdr2; ferencd@0: hdr1.parse("A: a\r\n"); ferencd@0: hdr2.parse("A: a\r\nB: b\r\n"); ferencd@0: ferencd@0: hdr1.removeAllFields(); ferencd@0: hdr2.removeAllFields(); ferencd@0: ferencd@0: std::vector > res1 = hdr1.getFieldList(); ferencd@0: VASSERT_EQ("Count", static_cast (0), res1.size()); ferencd@0: ferencd@0: std::vector > res2 = hdr2.getFieldList(); ferencd@0: VASSERT_EQ("Count", static_cast (0), res2.size()); ferencd@0: } ferencd@0: ferencd@0: // getFieldCount ferencd@0: void testgetFieldCount() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("A: a\r\nB: b\r\nC: c\r\nD: d\r\n"); ferencd@0: ferencd@0: VASSERT_EQ("Value", 4, hdr.getFieldCount()); ferencd@0: } ferencd@0: ferencd@0: // isEmpty ferencd@0: void testIsEmpty1() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("A: a\r\nB: b\r\nC: c\r\n"); ferencd@0: ferencd@0: VASSERT_EQ("Value", false, hdr.isEmpty()); ferencd@0: } ferencd@0: ferencd@0: void testIsEmpty2() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("\r\n"); ferencd@0: ferencd@0: VASSERT_EQ("Value", true, hdr.isEmpty()); ferencd@0: } ferencd@0: ferencd@0: // getFieldAt ferencd@0: void testGetFieldAt() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("B: b\r\nA: a\r\nC: c\r\n"); ferencd@0: ferencd@0: vmime::shared_ptr res = hdr.getFieldAt(2); ferencd@0: ferencd@0: VASSERT_EQ("Value", "C: c", getFieldValue(*res)); ferencd@0: } ferencd@0: ferencd@0: // getFieldList ferencd@0: void testGetFieldList1() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("A: a\r\nB: b1\r\nC: c\r\nB: b2\r\n"); ferencd@0: ferencd@0: std::vector > res = hdr.getFieldList(); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (4), res.size()); ferencd@0: VASSERT_EQ("First value", "A: a", headerTest::getFieldValue(*res[0])); ferencd@0: VASSERT_EQ("Second value", "B: b1", headerTest::getFieldValue(*res[1])); ferencd@0: VASSERT_EQ("Third value", "C: c", headerTest::getFieldValue(*res[2])); ferencd@0: VASSERT_EQ("Thourth value", "B: b2", headerTest::getFieldValue(*res[3])); ferencd@0: } ferencd@0: ferencd@0: void testGetFieldList2() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("\r\n"); ferencd@0: ferencd@0: std::vector > res = hdr.getFieldList(); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (0), res.size()); ferencd@0: } ferencd@0: ferencd@0: // find function tests ferencd@0: void testFind1() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("A: a\r\nB: b\r\nC: c\r\nB: d\r\n"); ferencd@0: ferencd@0: vmime::shared_ptr res = hdr.findField("B"); ferencd@0: ferencd@0: VASSERT_EQ("Value", "B: b", getFieldValue(*res)); ferencd@0: } ferencd@0: ferencd@0: // getAllByName function tests ferencd@0: void testFindAllFields1() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("A: a1\nC: c1\n"); ferencd@0: ferencd@0: std::vector > res = hdr.findAllFields("B"); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (0), res.size()); ferencd@0: } ferencd@0: ferencd@0: void testFindAllFields2() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("A: a1\nB: b1\nB: b2\nC: c1\n"); ferencd@0: ferencd@0: std::vector > res = hdr.findAllFields("B"); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (2), res.size()); ferencd@0: VASSERT_EQ("First value", "B: b1", headerTest::getFieldValue(*res[0])); ferencd@0: VASSERT_EQ("Second value", "B: b2", headerTest::getFieldValue(*res[1])); ferencd@0: } ferencd@0: ferencd@0: void testFindAllFields3() ferencd@0: { ferencd@0: vmime::header hdr; ferencd@0: hdr.parse("A: a1\nB: b1\nB: b2\nC: c1\nC: c3\nC: c2\n"); ferencd@0: ferencd@0: std::vector > res = hdr.findAllFields("C"); ferencd@0: ferencd@0: VASSERT_EQ("Count", static_cast (3), res.size()); ferencd@0: VASSERT_EQ("First value", "C: c1", headerTest::getFieldValue(*res[0])); ferencd@0: VASSERT_EQ("Second value", "C: c3", headerTest::getFieldValue(*res[1])); ferencd@0: VASSERT_EQ("Second value", "C: c2", headerTest::getFieldValue(*res[2])); ferencd@0: } ferencd@0: ferencd@0: VMIME_TEST_SUITE_END ferencd@0: