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
|
#include "WrDiskCacheEntry.h"
#include <cstring>
#include <cppunit/extensions/HelperMacros.h>
#include "TestUtil.h"
#include "DirectDiskAdaptor.h"
#include "ByteArrayDiskWriter.h"
namespace aria2 {
class WrDiskCacheEntryTest:public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(WrDiskCacheEntryTest);
CPPUNIT_TEST(testWriteToDisk);
CPPUNIT_TEST(testAppend);
CPPUNIT_TEST(testClear);
CPPUNIT_TEST_SUITE_END();
SharedHandle<DirectDiskAdaptor> adaptor_;
SharedHandle<ByteArrayDiskWriter> writer_;
public:
void setUp()
{
adaptor_.reset(new DirectDiskAdaptor());
writer_.reset(new ByteArrayDiskWriter());
adaptor_->setDiskWriter(writer_);
}
void testWriteToDisk();
void testAppend();
void testClear();
};
CPPUNIT_TEST_SUITE_REGISTRATION( WrDiskCacheEntryTest );
void WrDiskCacheEntryTest::testWriteToDisk()
{
WrDiskCacheEntry e(adaptor_);
e.cacheData(createDataCell(0, "??01234567", 2));
e.cacheData(createDataCell(8, "890"));
e.writeToDisk();
CPPUNIT_ASSERT_EQUAL((size_t)0, e.getSize());
CPPUNIT_ASSERT_EQUAL(std::string("01234567890"), writer_->getString());
}
void WrDiskCacheEntryTest::testAppend()
{
WrDiskCacheEntry e(adaptor_);
WrDiskCacheEntry::DataCell* cell = new WrDiskCacheEntry::DataCell();
cell->goff = 0;
size_t capacity = 6;
size_t offset = 2;
cell->data = new unsigned char[offset+capacity];
memcpy(cell->data, "??foo", 3);
cell->offset = offset;
cell->len = 3;
cell->capacity = capacity;
e.cacheData(cell);
CPPUNIT_ASSERT_EQUAL((size_t)3,
e.append(3, (const unsigned char*)"barbaz", 6));
CPPUNIT_ASSERT_EQUAL((size_t)6, cell->len);
CPPUNIT_ASSERT_EQUAL((size_t)6, e.getSize());
CPPUNIT_ASSERT_EQUAL((size_t)0,
e.append(7, (const unsigned char*)"FOO", 3));
}
void WrDiskCacheEntryTest::testClear()
{
WrDiskCacheEntry e(adaptor_);
e.cacheData(createDataCell(0, "foo"));
e.clear();
CPPUNIT_ASSERT_EQUAL((size_t)0, e.getSize());
CPPUNIT_ASSERT_EQUAL(std::string(), writer_->getString());
}
} // namespace aria2
|
