#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());
}