mirror of https://github.com/aria2/aria2
54 lines
1.3 KiB
C++
54 lines
1.3 KiB
C++
#include "StreamUriListParser.h"
|
|
#include "Exception.h"
|
|
#include "Util.h"
|
|
#include <sstream>
|
|
#include <algorithm>
|
|
#include <iostream>
|
|
#include <iterator>
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
using namespace std;
|
|
|
|
class StreamUriListParserTest : public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(StreamUriListParserTest);
|
|
CPPUNIT_TEST(testHasNext);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
string list2String(const Strings& src);
|
|
public:
|
|
void setUp() {
|
|
}
|
|
|
|
void testHasNext();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION( StreamUriListParserTest );
|
|
|
|
string StreamUriListParserTest::list2String(const Strings& src)
|
|
{
|
|
ostringstream strm;
|
|
copy(src.begin(), src.end(), ostream_iterator<string>(strm, " "));
|
|
return Util::trim(strm.str());
|
|
}
|
|
|
|
void StreamUriListParserTest::testHasNext()
|
|
{
|
|
stringstream s;
|
|
s << "http://localhost/index.html http://localhost2/index.html\n"
|
|
<< "ftp://localhost/aria2.tar.bz2";
|
|
|
|
StreamUriListParser flp(s);
|
|
|
|
CPPUNIT_ASSERT(flp.hasNext());
|
|
CPPUNIT_ASSERT_EQUAL(string("http://localhost/index.html http://localhost2/index.html"), list2String(flp.next()));
|
|
CPPUNIT_ASSERT(flp.hasNext());
|
|
CPPUNIT_ASSERT_EQUAL(string("ftp://localhost/aria2.tar.bz2"),
|
|
list2String(flp.next()));
|
|
CPPUNIT_ASSERT(flp.hasNext());
|
|
CPPUNIT_ASSERT_EQUAL(string(""),
|
|
list2String(flp.next()));
|
|
CPPUNIT_ASSERT(!flp.hasNext());
|
|
}
|