2009-03-13 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>

	Renamed ServerStatURISelector as FeedbackURISelector.
	* src/FeedbackURISelector.cc
	* src/FeedbackURISelector.h
	* src/Makefile.am
	* src/RequestGroupMan.cc
	* src/ServerStatURISelector.cc
	* src/ServerStatURISelector.h
	* test/FeedbackURISelectorTest.cc
	* test/Makefile.am
	* test/ServerStatURISelectorTest.cc

1 files changed, 100 insertions, 0 deletions

diff --git a/test/FeedbackURISelectorTest.cc b/test/FeedbackURISelectorTest.cc
new file mode 100644
index 00000000..5dcf2c76
--- /dev/null
+++ b/test/FeedbackURISelectorTest.cc
@@ -0,0 +1,100 @@
+#include "FeedbackURISelector.h"
+
+#include <iostream>
+
+#include <cppunit/extensions/HelperMacros.h>
+
+#include "Exception.h"
+#include "Util.h"
+#include "array_fun.h"
+#include "ServerStatMan.h"
+#include "ServerStat.h"
+
+namespace aria2 {
+
+class FeedbackURISelectorTest:public CppUnit::TestFixture {
+
+  CPPUNIT_TEST_SUITE(FeedbackURISelectorTest);
+  CPPUNIT_TEST(testSelect_withoutServerStat);
+  CPPUNIT_TEST(testSelect);
+  CPPUNIT_TEST(testSelect_skipErrorHost);
+  CPPUNIT_TEST_SUITE_END();
+private:
+
+  std::deque<std::string> uris;
+
+  SharedHandle<ServerStatMan> ssm;
+
+  SharedHandle<FeedbackURISelector> sel;
+  
+public:
+  void setUp()
+  {
+    static const char* urisSrc[] = {
+      "http://alpha/file",
+      "ftp://alpha/file",
+      "http://bravo/file"
+    };
+    uris.assign(&urisSrc[0], &urisSrc[arrayLength(urisSrc)]);
+    
+    ssm.reset(new ServerStatMan());
+    sel.reset(new FeedbackURISelector(ssm));
+  }
+
+  void tearDown() {}
+
+  void testSelect_withoutServerStat();
+  
+  void testSelect();
+
+  void testSelect_skipErrorHost();
+};
+
+
+CPPUNIT_TEST_SUITE_REGISTRATION(FeedbackURISelectorTest);
+
+void FeedbackURISelectorTest::testSelect_withoutServerStat()
+{
+  // Without ServerStat, selector returns first URI
+  std::string uri = sel->select(uris);
+  CPPUNIT_ASSERT_EQUAL(std::string("http://alpha/file"), uri);
+  CPPUNIT_ASSERT_EQUAL((size_t)2, uris.size());
+}
+
+void FeedbackURISelectorTest::testSelect()
+{
+  SharedHandle<ServerStat> bravo(new ServerStat("bravo", "http"));
+  bravo->updateDownloadSpeed(100000);
+  SharedHandle<ServerStat> alphaFTP(new ServerStat("alpha", "ftp"));
+  alphaFTP->updateDownloadSpeed(80000);
+  SharedHandle<ServerStat> alphaHTTP(new ServerStat("alpha", "http"));
+  alphaHTTP->updateDownloadSpeed(180000);
+  alphaHTTP->setError();
+
+  ssm->add(bravo);
+  ssm->add(alphaFTP);
+  ssm->add(alphaHTTP);
+
+  CPPUNIT_ASSERT_EQUAL(std::string("http://bravo/file"), sel->select(uris));
+  CPPUNIT_ASSERT_EQUAL((size_t)2, uris.size());
+  
+  CPPUNIT_ASSERT_EQUAL(std::string("ftp://alpha/file"), sel->select(uris));
+  CPPUNIT_ASSERT_EQUAL((size_t)1, uris.size());
+}
+
+void FeedbackURISelectorTest::testSelect_skipErrorHost()
+{
+  SharedHandle<ServerStat> alphaHTTP(new ServerStat("alpha", "http"));
+  alphaHTTP->setError();
+  SharedHandle<ServerStat> alphaFTP(new ServerStat("alpha", "ftp"));
+  alphaFTP->setError();
+
+  ssm->add(alphaHTTP);
+  ssm->add(alphaFTP);
+
+  // See error URIs are removed from URI List.
+  CPPUNIT_ASSERT_EQUAL(std::string("http://bravo/file"), sel->select(uris));
+  CPPUNIT_ASSERT_EQUAL((size_t)0, uris.size());
+}
+
+} // namespace aria2