102 lines
2.8 KiB
C++
102 lines
2.8 KiB
C++
|
#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();
|
||
|
|
}
|