1 files changed, 42 insertions, 0 deletions
diff --git a/test/BtPostDownloadHandlerTest.cc b/test/BtPostDownloadHandlerTest.cc
new file mode 100644
index 00000000..b4228ee2
--- /dev/null
+++ b/test/BtPostDownloadHandlerTest.cc
@@ -0,0 +1,42 @@
+#include "BtPostDownloadHandler.h"
+#include "BtContext.h"
+#include "RequestGroup.h"
+#include "Option.h"
+#include <cppunit/extensions/HelperMacros.h>
+
+class BtPostDownloadHandlerTest:public CppUnit::TestFixture {
+
+  CPPUNIT_TEST_SUITE(BtPostDownloadHandlerTest);
+  CPPUNIT_TEST(testCanHandle);
+  CPPUNIT_TEST(testGetNextRequestGroups);
+  CPPUNIT_TEST_SUITE_END();
+private:
+
+public:
+  void setUp() {}
+
+  void testCanHandle();
+  void testGetNextRequestGroups();
+};
+
+
+CPPUNIT_TEST_SUITE_REGISTRATION( BtPostDownloadHandlerTest );
+
+void BtPostDownloadHandlerTest::testCanHandle()
+{
+  Option op;
+  BtPostDownloadHandler handler(&op);
+  CPPUNIT_ASSERT(!handler.canHandle(".torrent!!"));
+  CPPUNIT_ASSERT(handler.canHandle(".torrent"));
+}
+
+void BtPostDownloadHandlerTest::testGetNextRequestGroups()
+{
+  Option op;
+  BtPostDownloadHandler handler(&op);
+  RequestGroups groups = handler.getNextRequestGroups("test.torrent");
+  CPPUNIT_ASSERT_EQUAL((size_t)1, groups.size());
+  BtContextHandle btctx = groups.front()->getDownloadContext();
+  CPPUNIT_ASSERT(!btctx.isNull());
+  CPPUNIT_ASSERT_EQUAL(string("aria2-test"), btctx->getName());
+}