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
|
#include "UriListParser.h"
#include <sstream>
#include <algorithm>
#include <iostream>
#include <fstream>
#include <iterator>
#include <cppunit/extensions/HelperMacros.h>
#include "Exception.h"
#include "util.h"
#include "prefs.h"
#include "OptionHandler.h"
namespace aria2 {
class UriListParserTest : public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(UriListParserTest);
CPPUNIT_TEST(testHasNext);
CPPUNIT_TEST_SUITE_END();
private:
std::string list2String(const std::vector<std::string>& src);
public:
void setUp() {
}
void testHasNext();
};
CPPUNIT_TEST_SUITE_REGISTRATION( UriListParserTest );
std::string UriListParserTest::list2String(const std::vector<std::string>& src)
{
std::ostringstream strm;
std::copy(src.begin(), src.end(), std::ostream_iterator<std::string>(strm, " "));
return util::trim(strm.str());
}
void UriListParserTest::testHasNext()
{
std::ifstream in("filelist1.txt", std::ios::binary);
UriListParser flp(in);
std::vector<std::string> uris;
Option reqOp;
CPPUNIT_ASSERT(flp.hasNext());
flp.parseNext(uris, reqOp);
CPPUNIT_ASSERT_EQUAL
(std::string("http://localhost/index.html http://localhost2/index.html"),
list2String(uris));
uris.clear();
reqOp.clear();
CPPUNIT_ASSERT(flp.hasNext());
flp.parseNext(uris, reqOp);
CPPUNIT_ASSERT_EQUAL(std::string("ftp://localhost/aria2.tar.bz2"),
list2String(uris));
CPPUNIT_ASSERT_EQUAL(std::string("/tmp"), reqOp.get(PREF_DIR));
CPPUNIT_ASSERT_EQUAL(std::string("chunky_chocolate"), reqOp.get(PREF_OUT));
uris.clear();
reqOp.clear();
CPPUNIT_ASSERT(!flp.hasNext());
flp.parseNext(uris, reqOp);
CPPUNIT_ASSERT_EQUAL(std::string(""), list2String(uris));
CPPUNIT_ASSERT(!flp.hasNext());
}
} // namespace aria2
|