1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
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();
}
|
