mirror of
https://github.com/aria2/aria2.git
synced 2025-04-05 13:37:40 +03:00
2010-09-06 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>
Data from remote server in HTTP/FTP download are now written to the disk(or memory) through StreamFilter. Decoding chunked and gziped streams are done cascading StreamFilter. Removed inefficient 1byte read code. * src/ChunkedDecodingStreamFilter.cc * src/ChunkedDecodingStreamFilter.h * src/DownloadCommand.cc * src/DownloadCommand.h * src/GZipDecodingStreamFilter.cc * src/GZipDecodingStreamFilter.h * src/HttpConnection.cc * src/HttpDownloadCommand.cc * src/HttpResponse.cc * src/HttpResponse.h * src/HttpResponseCommand.cc * src/HttpResponseCommand.h * src/HttpSkipResponseCommand.cc * src/HttpSkipResponseCommand.h * src/Makefile.am * src/NullSinkStreamFilter.cc * src/NullSinkStreamFilter.h * src/RequestGroup.cc * src/SinkStreamFilter.cc * src/SinkStreamFilter.h * src/StreamFilter.cc * src/StreamFilter.h * test/ChunkedDecodingStreamFilterTest.cc * test/GZipDecodingStreamFilterTest.cc * test/HttpResponseTest.cc * test/Makefile.am * test/MockSegment.h
This commit is contained in:
parent
552c753dbb
commit
efbfe4c006
30 changed files with 1767 additions and 231 deletions
81
test/GZipDecodingStreamFilterTest.cc
Normal file
81
test/GZipDecodingStreamFilterTest.cc
Normal file
|
@ -0,0 +1,81 @@
|
|||
#include "GZipDecodingStreamFilter.h"
|
||||
|
||||
#include <cassert>
|
||||
#include <iostream>
|
||||
#include <fstream>
|
||||
|
||||
#include <cppunit/extensions/HelperMacros.h>
|
||||
|
||||
#include "Exception.h"
|
||||
#include "util.h"
|
||||
#include "Segment.h"
|
||||
#include "ByteArrayDiskWriter.h"
|
||||
#include "SinkStreamFilter.h"
|
||||
#include "MockSegment.h"
|
||||
#ifdef ENABLE_MESSAGE_DIGEST
|
||||
# include "MessageDigestHelper.h"
|
||||
#endif // ENABLE_MESSAGE_DIGEST
|
||||
|
||||
namespace aria2 {
|
||||
|
||||
class GZipDecodingStreamFilterTest:public CppUnit::TestFixture {
|
||||
|
||||
CPPUNIT_TEST_SUITE(GZipDecodingStreamFilterTest);
|
||||
CPPUNIT_TEST(testTransform);
|
||||
CPPUNIT_TEST_SUITE_END();
|
||||
|
||||
class MockSegment2:public MockSegment {
|
||||
private:
|
||||
off_t positionToWrite_;
|
||||
public:
|
||||
MockSegment2():positionToWrite_(0) {}
|
||||
|
||||
virtual void updateWrittenLength(size_t bytes)
|
||||
{
|
||||
positionToWrite_ += bytes;
|
||||
}
|
||||
|
||||
virtual off_t getPositionToWrite() const
|
||||
{
|
||||
return positionToWrite_;
|
||||
}
|
||||
};
|
||||
|
||||
SharedHandle<GZipDecodingStreamFilter> filter_;
|
||||
SharedHandle<SinkStreamFilter> sinkFilter_;
|
||||
SharedHandle<ByteArrayDiskWriter> writer_;
|
||||
SharedHandle<MockSegment2> segment_;
|
||||
public:
|
||||
void setUp()
|
||||
{
|
||||
writer_.reset(new ByteArrayDiskWriter());
|
||||
sinkFilter_.reset(new SinkStreamFilter());
|
||||
filter_.reset(new GZipDecodingStreamFilter(sinkFilter_));
|
||||
sinkFilter_->init();
|
||||
filter_->init();
|
||||
segment_.reset(new MockSegment2());
|
||||
}
|
||||
|
||||
void testTransform();
|
||||
};
|
||||
|
||||
|
||||
CPPUNIT_TEST_SUITE_REGISTRATION(GZipDecodingStreamFilterTest);
|
||||
|
||||
void GZipDecodingStreamFilterTest::testTransform()
|
||||
{
|
||||
unsigned char buf[4096];
|
||||
std::ifstream in("gzip_decode_test.gz", std::ios::binary);
|
||||
while(in) {
|
||||
in.read(reinterpret_cast<char*>(buf), sizeof(buf));
|
||||
filter_->transform(writer_, segment_, buf, in.gcount());
|
||||
}
|
||||
CPPUNIT_ASSERT(filter_->finished());
|
||||
#ifdef ENABLE_MESSAGE_DIGEST
|
||||
CPPUNIT_ASSERT_EQUAL(std::string("8b577b33c0411b2be9d4fa74c7402d54a8d21f96"),
|
||||
MessageDigestHelper::digestString
|
||||
(MessageDigestContext::SHA1, writer_->getString()));
|
||||
#endif // ENABLE_MESSAGE_DIGEST
|
||||
}
|
||||
|
||||
} // namespace aria2
|
Loading…
Add table
Add a link
Reference in a new issue