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
|
//
// VMime library (http://vmime.sourceforge.net)
// Copyright (C) 2002-2005 Vincent Richard <[email protected]>
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License as
// published by the Free Software Foundation; either version 2 of
// the License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
//
#include "../lib/unit++/unit++.h"
#include <iostream>
#include <ostream>
#include "vmime/vmime.hpp"
#include "vmime/platforms/posix/posixHandler.hpp"
using namespace unitpp;
namespace
{
class bodyPartTest : public suite
{
static const vmime::string extractComponentString
(const vmime::string& buffer, const vmime::component& c)
{
return vmime::string(buffer.begin() + c.getParsedOffset(),
buffer.begin() + c.getParsedOffset() + c.getParsedLength());
}
void testParse()
{
vmime::string str1 = "HEADER\r\n\r\nBODY";
vmime::bodyPart p1;
p1.parse(str1);
assert_eq("1", "HEADER\r\n\r\n", extractComponentString(str1, *p1.getHeader()));
assert_eq("2", "BODY", extractComponentString(str1, *p1.getBody()));
vmime::string str2 = "HEADER\n\nBODY";
vmime::bodyPart p2;
p2.parse(str2);
assert_eq("3", "HEADER\n\n", extractComponentString(str2, *p2.getHeader()));
assert_eq("4", "BODY", extractComponentString(str2, *p2.getBody()));
vmime::string str3 = "HEADER\r\n\nBODY";
vmime::bodyPart p3;
p3.parse(str3);
assert_eq("5", "HEADER\r\n\n", extractComponentString(str3, *p3.getHeader()));
assert_eq("6", "BODY", extractComponentString(str3, *p3.getBody()));
}
void testGenerate()
{
vmime::bodyPart p1;
p1.getHeader()->getField("Foo")->setValue(vmime::string("bar"));
p1.getBody()->setContents(vmime::stringContentHandler("Baz"));
assert_eq("1", "Foo: bar\r\n\r\nBaz", p1.generate());
}
public:
bodyPartTest() : suite("vmime::bodyPart")
{
// VMime initialization
vmime::platformDependant::setHandler<vmime::platforms::posix::posixHandler>();
add("Parse", testcase(this, "Parse", &bodyPartTest::testParse));
add("Generate", testcase(this, "Generate", &bodyPartTest::testGenerate));
suite::main().add("vmime::bodyPart", this);
}
};
bodyPartTest* theTest = new bodyPartTest();
}
|
