mirror of https://github.com/aria2/aria2
71 lines
1.9 KiB
C++
71 lines
1.9 KiB
C++
#include "ChunkedEncoding.h"
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
namespace aria2 {
|
|
|
|
class ChunkedEncodingTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(ChunkedEncodingTest);
|
|
CPPUNIT_TEST(testInflate1);
|
|
CPPUNIT_TEST(testInflateLargeChunk);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
ChunkedEncoding* enc;
|
|
public:
|
|
void setUp() {
|
|
enc = new ChunkedEncoding();
|
|
enc->init();
|
|
}
|
|
|
|
void testInflate1();
|
|
void testInflateLargeChunk();
|
|
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION( ChunkedEncodingTest );
|
|
|
|
void ChunkedEncodingTest::testInflate1() {
|
|
std::string msg = "a\r\n1234567890\r\n";
|
|
unsigned char buf[100];
|
|
size_t len = sizeof(buf);
|
|
enc->inflate(buf, len,
|
|
reinterpret_cast<const unsigned char*>(msg.c_str()), msg.size());
|
|
CPPUNIT_ASSERT_EQUAL(std::string("1234567890"), std::string(&buf[0], &buf[len]));
|
|
// second pass
|
|
len = sizeof(buf);
|
|
msg = "3;extensionIgnored\r\n123\r\n0\r\n";
|
|
enc->inflate(buf, len,
|
|
reinterpret_cast<const unsigned char*>(msg.c_str()), msg.size());
|
|
CPPUNIT_ASSERT_EQUAL(std::string("123"), std::string(&buf[0], &buf[len]));
|
|
// input is over
|
|
CPPUNIT_ASSERT(enc->finished());
|
|
}
|
|
|
|
void ChunkedEncodingTest::testInflateLargeChunk() {
|
|
// give over 4096 character chunk
|
|
std::fstream is("4096chunk.txt", std::ios::in);
|
|
if(is.fail()) {
|
|
CPPUNIT_FAIL("cannot open file 4096chunk.txt");
|
|
}
|
|
std::string body;
|
|
is >> body;
|
|
unsigned char buf[4097];
|
|
size_t len = sizeof(buf);
|
|
for(int i = 0; i < 2; i++) {
|
|
std::string msg = "1000\r\n"+body+"\r\n";
|
|
len = sizeof(buf);
|
|
enc->inflate(buf, len,
|
|
reinterpret_cast<const unsigned char*>(msg.c_str()), msg.size());
|
|
CPPUNIT_ASSERT_EQUAL(body, std::string(&buf[0], &buf[len]));
|
|
}
|
|
enc->inflate(buf, len, reinterpret_cast<const unsigned char*>("0\r\n"), 3);
|
|
CPPUNIT_ASSERT_EQUAL((size_t)0, len);
|
|
CPPUNIT_ASSERT(enc->finished());
|
|
}
|
|
|
|
} // namespace aria2
|