Add test for recent changes in SessionSerializer

1 files changed, 6 insertions, 1 deletions

diff --git a/test/SessionSerializerTest.cc b/test/SessionSerializerTest.cc
index b2dc5bd5..9ebc6acd 100644
--- a/test/SessionSerializerTest.cc
+++ b/test/SessionSerializerTest.cc
@@ -59,6 +59,11 @@ void SessionSerializerTest::testSave()
     createDownloadResult(error_code::FINISHED, "http://finished"),
     createDownloadResult(error_code::FINISHED, "http://force-save")
   };
+  // This URI will be discarded because same URI exists in remaining
+  // URIs.
+  drs[1]->fileEntries[0]->getSpentUris().push_back("http://error");
+  drs[1]->fileEntries[0]->getSpentUris().push_back("http://error2");
+
   drs[3]->option->put(PREF_FORCE_SAVE, A2_V_TRUE);
   for(size_t i = 0; i < sizeof(drs)/sizeof(drs[0]); ++i) {
     rgman->addDownloadResult(drs[i]);
@@ -74,7 +79,7 @@ void SessionSerializerTest::testSave()
   std::ifstream ss(filename.c_str(), std::ios::binary);
   std::string line;
   std::getline(ss, line);
-  CPPUNIT_ASSERT_EQUAL(std::string("http://error\t"), line);
+  CPPUNIT_ASSERT_EQUAL(std::string("http://error2\thttp://error\t"), line);
   std::getline(ss, line);
   CPPUNIT_ASSERT_EQUAL(fmt(" gid=%s", drs[1]->gid->toHex().c_str()), line);
   std::getline(ss, line);