diff options
| author | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2013-06-28 00:25:06 +0900 |
|---|---|---|
| committer | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2013-06-29 23:09:09 +0900 |
| commit | 5a223115e09cd568169fe2efd71d09689a980335 (patch) | |
| tree | 61250fc5f89229e4ecc63d70ae2069207388fea0 /test/CookieStorageTest.cc | |
| parent | 0b1e05b13bfd7cfbf75b299c1f32cdd6e6aed953 (diff) | |
Rewrite CookieStorage
Diffstat (limited to 'test/CookieStorageTest.cc')
| -rw-r--r-- | test/CookieStorageTest.cc | 481 |
1 files changed, 274 insertions, 207 deletions
diff --git a/test/CookieStorageTest.cc b/test/CookieStorageTest.cc index 4740f969..e88332b8 100644 --- a/test/CookieStorageTest.cc +++ b/test/CookieStorageTest.cc @@ -12,6 +12,7 @@ #include "RecoverableException.h" #include "File.h" #include "TestUtil.h" +#include "TimerA2.h" namespace aria2 { @@ -29,14 +30,13 @@ class CookieStorageTest:public CppUnit::TestFixture { CPPUNIT_TEST(testSaveNsFormat_fail); CPPUNIT_TEST(testCookieIsFull); CPPUNIT_TEST(testDomainIsFull); + CPPUNIT_TEST(testEviction); CPPUNIT_TEST_SUITE_END(); public: void setUp() {} void tearDown() {} - void dumpCookie(std::vector<Cookie>& cookies, const CookieStorage& cs); - void testStore(); void testParseAndStore(); void testCriteriaFind(); @@ -48,56 +48,67 @@ public: void testSaveNsFormat_fail(); void testCookieIsFull(); void testDomainIsFull(); + void testEviction(); }; CPPUNIT_TEST_SUITE_REGISTRATION(CookieStorageTest); -void CookieStorageTest::dumpCookie -(std::vector<Cookie>& cookies, const CookieStorage& st) +namespace { +std::vector<const Cookie*> dumpCookie(const CookieStorage& st) { - st.dumpCookie(std::back_inserter(cookies)); - std::sort(cookies.begin(), cookies.end(), CookieSorter()); + auto res = std::vector<const Cookie*>{}; + st.dumpCookie(std::back_inserter(res)); + std::sort(res.begin(), res.end(), CookieSorter()); + return res; } +} // namespace void CookieStorageTest::testStore() { time_t now = 1000; - CookieStorage st; - Cookie goodCookie(createCookie("k", "v", "localhost", true, "/", false)); - CPPUNIT_ASSERT(st.store(goodCookie, now)); + auto st = CookieStorage{}; + auto goodCookie = []() { + return createCookie("k", "v", "localhost", true, "/", false); + }; + CPPUNIT_ASSERT(st.store(goodCookie(), now)); CPPUNIT_ASSERT_EQUAL((size_t)1, st.size()); - CPPUNIT_ASSERT(st.contains(goodCookie)); + CPPUNIT_ASSERT(st.contains(*goodCookie())); - Cookie anotherCookie(createCookie("k", "v", "mirror", true, "/", true)); - CPPUNIT_ASSERT(st.store(anotherCookie, now)); + auto anotherCookie = []() { + return createCookie("k", "v", "mirror", true, "/", true); + }; + CPPUNIT_ASSERT(st.store(anotherCookie(), now)); CPPUNIT_ASSERT_EQUAL((size_t)2, st.size()); - CPPUNIT_ASSERT(st.contains(anotherCookie)); - CPPUNIT_ASSERT(st.contains(goodCookie)); + CPPUNIT_ASSERT(st.contains(*anotherCookie())); + CPPUNIT_ASSERT(st.contains(*goodCookie())); - Cookie updateGoodCookie(createCookie("k", "v2", "localhost", true, - "/", false)); - CPPUNIT_ASSERT(st.store(updateGoodCookie, now)); + auto updateGoodCookie = []() { + return createCookie("k", "v2", "localhost", true, "/", false); + }; + CPPUNIT_ASSERT(st.store(updateGoodCookie(), now)); CPPUNIT_ASSERT_EQUAL((size_t)2, st.size()); - CPPUNIT_ASSERT(st.contains(updateGoodCookie)); - CPPUNIT_ASSERT(st.contains(anotherCookie)); + CPPUNIT_ASSERT(st.contains(*updateGoodCookie())); + CPPUNIT_ASSERT(st.contains(*anotherCookie())); - Cookie expireGoodCookie(createCookie("k", "v3", 0, "localhost", true, - "/", false)); - CPPUNIT_ASSERT(!st.store(expireGoodCookie, now)); + auto expireGoodCookie = []() { + return createCookie("k", "v3", 0, "localhost", true, "/", false); + }; + CPPUNIT_ASSERT(!st.store(expireGoodCookie(), now)); CPPUNIT_ASSERT_EQUAL((size_t)1, st.size()); - CPPUNIT_ASSERT(st.contains(anotherCookie)); + CPPUNIT_ASSERT(st.contains(*anotherCookie())); - Cookie fromNumericHost(createCookie("k", "v", "192.168.1.1", true, - "/", false)); - CPPUNIT_ASSERT(st.store(fromNumericHost, now)); + auto fromNumericHost = []() { + return createCookie("k", "v", "192.168.1.1", true, "/", false); + }; + CPPUNIT_ASSERT(st.store(fromNumericHost(), now)); CPPUNIT_ASSERT_EQUAL((size_t)2, st.size()); - CPPUNIT_ASSERT(st.contains(fromNumericHost)); + CPPUNIT_ASSERT(st.contains(*fromNumericHost())); } void CookieStorageTest::testParseAndStore() { - CookieStorage st; + auto st = CookieStorage{}; time_t now = 1000; std::string localhostCookieStr = "k=v;" " expires=Fri, 01 Jan 2038 00:00:00 GMT; path=/; domain=localhost;"; @@ -122,221 +133,233 @@ void CookieStorageTest::testParseAndStore() void CookieStorageTest::testCriteriaFind() { - CookieStorage st; + auto st = CookieStorage{}; time_t now = 1000; - Cookie alpha(createCookie("alpha", "ALPHA", "aria2.org", false, "/", false)); - Cookie bravo(createCookie("bravo", "BRAVO", 1060, "aria2.org", false, - "/foo", false)); - Cookie charlie(createCookie("charlie", "CHARLIE", "aria2.org", false, - "/", true)); - Cookie delta(createCookie("delta", "DELTA", "aria2.org", false, - "/foo/bar", false)); - Cookie echo(createCookie("echo", "ECHO", "www.dl.aria2.org", false, - "/", false)); - Cookie foxtrot(createCookie("foxtrot", "FOXTROT", "sf.net", false, - "/", false)); - Cookie golf(createCookie("golf", "GOLF", "192.168.1.1", true, - "/", false)); - Cookie hotel1(createCookie("hotel", "HOTEL1", "samename.x", false, - "/", false)); - Cookie hotel2(createCookie("hotel", "HOTEL2", "samename.x", false, - "/hotel", false)); - Cookie hotel3(createCookie("hotel", "HOTEL3", "samename.x", false, - "/bar/wine", false)); - Cookie hotel4(createCookie("hotel", "HOTEL4", "samename.x", false, - "/bar/", false)); - Cookie india1(createCookie("india", "INDIA1", "default.domain", true, - "/foo", false)); - Cookie india2(createCookie("india", "INDIA2", "default.domain", false, - "/", false)); - Cookie juliet1(createCookie("juliet", "JULIET1", "localhost", true, - "/foo", false)); - - CPPUNIT_ASSERT(st.store(alpha, now)); - CPPUNIT_ASSERT(st.store(bravo, now)); - CPPUNIT_ASSERT(st.store(charlie, now)); - CPPUNIT_ASSERT(st.store(delta, now)); - CPPUNIT_ASSERT(st.store(echo, now)); - CPPUNIT_ASSERT(st.store(foxtrot, now)); - CPPUNIT_ASSERT(st.store(golf, now)); - CPPUNIT_ASSERT(st.store(hotel1, now)); - CPPUNIT_ASSERT(st.store(hotel2, now)); - CPPUNIT_ASSERT(st.store(hotel3, now)); - CPPUNIT_ASSERT(st.store(hotel4, now)); - CPPUNIT_ASSERT(st.store(india1, now)); - CPPUNIT_ASSERT(st.store(india2, now)); - CPPUNIT_ASSERT(st.store(juliet1, now)); - - std::vector<Cookie> aria2Slash = st.criteriaFind("www.dl.aria2.org", "/", - 0, false); + auto alpha = []() { + return createCookie("alpha", "ALPHA", "aria2.org", false, "/", false); + }; + auto bravo = []() { + return createCookie("bravo", "BRAVO", 1060, "aria2.org", false, + "/foo", false); + }; + auto charlie = []() { + return createCookie("charlie", "CHARLIE", "aria2.org", false, + "/", true); + }; + auto delta = []() { + return createCookie("delta", "DELTA", "aria2.org", false, + "/foo/bar", false); + }; + auto echo = []() { + return createCookie("echo", "ECHO", "www.dl.aria2.org", false, + "/", false); + }; + auto foxtrot = []() { + return createCookie("foxtrot", "FOXTROT", "sf.net", false, + "/", false); + }; + auto golf = []() { + return createCookie("golf", "GOLF", "192.168.1.1", true, + "/", false); + }; + auto hotel1 = []() { + return createCookie("hotel", "HOTEL1", "samename.x", false, + "/", false); + }; + auto hotel2 = []() { + return createCookie("hotel", "HOTEL2", "samename.x", false, + "/hotel", false); + }; + auto hotel3 = []() { + return createCookie("hotel", "HOTEL3", "samename.x", false, + "/bar/wine", false); + }; + auto hotel4 = []() { + return createCookie("hotel", "HOTEL4", "samename.x", false, + "/bar/", false); + }; + auto india1 = []() { + return createCookie("india", "INDIA1", "default.domain", true, + "/foo", false); + }; + auto india2 = []() { + return createCookie("india", "INDIA2", "default.domain", false, + "/", false); + }; + auto juliet1 = []() { + return createCookie("juliet", "JULIET1", "localhost", true, + "/foo", false); + }; + + CPPUNIT_ASSERT(st.store(alpha(), now)); + CPPUNIT_ASSERT(st.store(bravo(), now)); + CPPUNIT_ASSERT(st.store(charlie(), now)); + CPPUNIT_ASSERT(st.store(delta(), now)); + CPPUNIT_ASSERT(st.store(echo(), now)); + CPPUNIT_ASSERT(st.store(foxtrot(), now)); + CPPUNIT_ASSERT(st.store(golf(), now)); + CPPUNIT_ASSERT(st.store(hotel1(), now)); + CPPUNIT_ASSERT(st.store(hotel2(), now)); + CPPUNIT_ASSERT(st.store(hotel3(), now)); + CPPUNIT_ASSERT(st.store(hotel4(), now)); + CPPUNIT_ASSERT(st.store(india1(), now)); + CPPUNIT_ASSERT(st.store(india2(), now)); + CPPUNIT_ASSERT(st.store(juliet1(), now)); + + auto aria2Slash = st.criteriaFind("www.dl.aria2.org", "/", 0, false); CPPUNIT_ASSERT_EQUAL((size_t)2, aria2Slash.size()); - CPPUNIT_ASSERT(std::find(aria2Slash.begin(), aria2Slash.end(), alpha) - != aria2Slash.end()); - CPPUNIT_ASSERT(std::find(aria2Slash.begin(), aria2Slash.end(), echo) - != aria2Slash.end()); + CPPUNIT_ASSERT(derefFind(aria2Slash, alpha())); + CPPUNIT_ASSERT(derefFind(aria2Slash, echo())); - std::vector<Cookie> aria2SlashFoo = st.criteriaFind("www.dl.aria2.org","/foo", - 0, false); + auto aria2SlashFoo = st.criteriaFind("www.dl.aria2.org","/foo", 0, false); CPPUNIT_ASSERT_EQUAL((size_t)3, aria2SlashFoo.size()); - CPPUNIT_ASSERT_EQUAL(std::string("bravo"), aria2SlashFoo[0].getName()); - CPPUNIT_ASSERT(std::find(aria2SlashFoo.begin(), aria2SlashFoo.end(), alpha) - != aria2SlashFoo.end()); - CPPUNIT_ASSERT(std::find(aria2SlashFoo.begin(), aria2SlashFoo.end(), echo) - != aria2SlashFoo.end()); - - std::vector<Cookie> aria2Expires = st.criteriaFind("www.dl.aria2.org", "/foo", - 1120, - false); + CPPUNIT_ASSERT_EQUAL(std::string("bravo"), aria2SlashFoo[0]->getName()); + CPPUNIT_ASSERT(derefFind(aria2SlashFoo, alpha())); + CPPUNIT_ASSERT(derefFind(aria2SlashFoo, echo())); + + auto aria2Expires = st.criteriaFind("www.dl.aria2.org", "/foo", 1120, false); CPPUNIT_ASSERT_EQUAL((size_t)2, aria2Expires.size()); - CPPUNIT_ASSERT(std::find(aria2Expires.begin(), aria2Expires.end(), alpha) - != aria2Expires.end()); - CPPUNIT_ASSERT(std::find(aria2Expires.begin(), aria2Expires.end(), echo) - != aria2Expires.end()); + CPPUNIT_ASSERT(derefFind(aria2Expires, alpha())); + CPPUNIT_ASSERT(derefFind(aria2Expires, echo())); - std::vector<Cookie> dlAria2 = st.criteriaFind("dl.aria2.org", "/", 0, false); + auto dlAria2 = st.criteriaFind("dl.aria2.org", "/", 0, false); CPPUNIT_ASSERT_EQUAL((size_t)1, dlAria2.size()); - CPPUNIT_ASSERT_EQUAL(std::string("alpha"), dlAria2[0].getName()); + CPPUNIT_ASSERT_EQUAL(std::string("alpha"), dlAria2[0]->getName()); - std::vector<Cookie> tailmatchAria2 = st.criteriaFind("myaria2.org", "/", 0, - false); + auto tailmatchAria2 = st.criteriaFind("myaria2.org", "/", 0, false); CPPUNIT_ASSERT(tailmatchAria2.empty()); - std::vector<Cookie> numericHostCookies = st.criteriaFind("192.168.1.1", "/",0, - false); + auto numericHostCookies = st.criteriaFind("192.168.1.1", "/",0, false); CPPUNIT_ASSERT_EQUAL((size_t)1, numericHostCookies.size()); - CPPUNIT_ASSERT_EQUAL(std::string("golf"), numericHostCookies[0].getName()); + CPPUNIT_ASSERT_EQUAL(std::string("golf"), numericHostCookies[0]->getName()); - std::vector<Cookie> sameNameCookies = - st.criteriaFind("samename.x", "/bar/wine", 0, false); + auto sameNameCookies = st.criteriaFind("samename.x", "/bar/wine", 0, false); CPPUNIT_ASSERT_EQUAL((size_t)3, sameNameCookies.size()); - CPPUNIT_ASSERT_EQUAL(std::string("HOTEL3"), sameNameCookies[0].getValue()); - CPPUNIT_ASSERT_EQUAL(std::string("HOTEL4"), sameNameCookies[1].getValue()); - CPPUNIT_ASSERT_EQUAL(std::string("HOTEL1"), sameNameCookies[2].getValue()); + CPPUNIT_ASSERT_EQUAL(std::string("HOTEL3"), sameNameCookies[0]->getValue()); + CPPUNIT_ASSERT_EQUAL(std::string("HOTEL4"), sameNameCookies[1]->getValue()); + CPPUNIT_ASSERT_EQUAL(std::string("HOTEL1"), sameNameCookies[2]->getValue()); - std::vector<Cookie> defaultDomainCookies = - st.criteriaFind("default.domain", "/foo", 0, false); + auto defaultDomainCookies = st.criteriaFind("default.domain", "/foo", 0, + false); CPPUNIT_ASSERT_EQUAL((size_t)2, defaultDomainCookies.size()); CPPUNIT_ASSERT_EQUAL(std::string("INDIA1"), - defaultDomainCookies[0].getValue()); + defaultDomainCookies[0]->getValue()); CPPUNIT_ASSERT_EQUAL(std::string("INDIA2"), - defaultDomainCookies[1].getValue()); + defaultDomainCookies[1]->getValue()); defaultDomainCookies = st.criteriaFind("sub.default.domain", "/foo", 0, false); CPPUNIT_ASSERT_EQUAL((size_t)1, defaultDomainCookies.size()); CPPUNIT_ASSERT_EQUAL(std::string("INDIA2"), - defaultDomainCookies[0].getValue()); + defaultDomainCookies[0]->getValue()); // localhost.local case - std::vector<Cookie> localDomainCookies = - st.criteriaFind("localhost", "/foo", 0, false); + auto localDomainCookies = st.criteriaFind("localhost", "/foo", 0, false); CPPUNIT_ASSERT_EQUAL((size_t)1, localDomainCookies.size()); CPPUNIT_ASSERT_EQUAL(std::string("JULIET1"), - localDomainCookies[0].getValue()); + localDomainCookies[0]->getValue()); } void CookieStorageTest::testCriteriaFind_cookieOrder() { - CookieStorage st; - Cookie a(createCookie("a", "0", "host", true, "/", false)); - a.setCreationTime(1000); - Cookie b(createCookie("b", "0", "host", true, "/foo", false)); - b.setCreationTime(5000); - Cookie c(createCookie("c", "0", "host", true, "/foo", false)); - c.setCreationTime(4000); - Cookie d(createCookie("d", "0", "host", true, "/foo/bar", false)); - d.setCreationTime(6000); - - st.store(a, 0); - st.store(b, 0); - st.store(c, 0); - st.store(d, 0); - - std::vector<Cookie> cookies = st.criteriaFind("host", "/foo/bar", 0, false); + auto st = CookieStorage{}; + auto a = createCookie("a", "0", "host", true, "/", false); + a->setCreationTime(1000); + auto b = createCookie("b", "0", "host", true, "/foo", false); + b->setCreationTime(5000); + auto c = createCookie("c", "0", "host", true, "/foo", false); + c->setCreationTime(4000); + auto d = createCookie("d", "0", "host", true, "/foo/bar", false); + d->setCreationTime(6000); + + st.store(std::move(a), 0); + st.store(std::move(b), 0); + st.store(std::move(c), 0); + st.store(std::move(d), 0); + + auto cookies = st.criteriaFind("host", "/foo/bar", 0, false); CPPUNIT_ASSERT_EQUAL((size_t)4, cookies.size()); - CPPUNIT_ASSERT_EQUAL(std::string("d"), cookies[0].getName()); - CPPUNIT_ASSERT_EQUAL(std::string("c"), cookies[1].getName()); - CPPUNIT_ASSERT_EQUAL(std::string("b"), cookies[2].getName()); - CPPUNIT_ASSERT_EQUAL(std::string("a"), cookies[3].getName()); + CPPUNIT_ASSERT_EQUAL(std::string("d"), cookies[0]->getName()); + CPPUNIT_ASSERT_EQUAL(std::string("c"), cookies[1]->getName()); + CPPUNIT_ASSERT_EQUAL(std::string("b"), cookies[2]->getName()); + CPPUNIT_ASSERT_EQUAL(std::string("a"), cookies[3]->getName()); } void CookieStorageTest::testLoad() { - CookieStorage st; + auto st = CookieStorage{}; st.load(A2_TEST_DIR"/nscookietest.txt", 1001); CPPUNIT_ASSERT_EQUAL((size_t)4, st.size()); - std::vector<Cookie> cookies; - dumpCookie(cookies, st); + auto cookies = dumpCookie(st); - Cookie c = cookies[0]; - CPPUNIT_ASSERT_EQUAL(std::string("passwd"), c.getName()); - CPPUNIT_ASSERT_EQUAL(std::string("secret"), c.getValue()); - CPPUNIT_ASSERT_EQUAL(std::numeric_limits<time_t>::max(), c.getExpiryTime()); - CPPUNIT_ASSERT(!c.getPersistent()); - CPPUNIT_ASSERT_EQUAL(std::string("/cgi-bin"), c.getPath()); - CPPUNIT_ASSERT_EQUAL(std::string("192.168.0.1"), c.getDomain()); - CPPUNIT_ASSERT(c.getHostOnly()); - CPPUNIT_ASSERT(!c.getSecure()); + auto c = cookies[0]; + CPPUNIT_ASSERT_EQUAL(std::string("passwd"), c->getName()); + CPPUNIT_ASSERT_EQUAL(std::string("secret"), c->getValue()); + CPPUNIT_ASSERT_EQUAL(std::numeric_limits<time_t>::max(), c->getExpiryTime()); + CPPUNIT_ASSERT(!c->getPersistent()); + CPPUNIT_ASSERT_EQUAL(std::string("/cgi-bin"), c->getPath()); + CPPUNIT_ASSERT_EQUAL(std::string("192.168.0.1"), c->getDomain()); + CPPUNIT_ASSERT(c->getHostOnly()); + CPPUNIT_ASSERT(!c->getSecure()); c = cookies[1]; - CPPUNIT_ASSERT_EQUAL(std::string("novalue"), c.getName()); - CPPUNIT_ASSERT_EQUAL(std::string(""), c.getValue()); - CPPUNIT_ASSERT_EQUAL((time_t)2147483647, c.getExpiryTime()); - CPPUNIT_ASSERT(c.getPersistent()); - CPPUNIT_ASSERT_EQUAL(std::string("/"), c.getPath()); - CPPUNIT_ASSERT_EQUAL(std::string("example.org"), c.getDomain()); - CPPUNIT_ASSERT(!c.getHostOnly()); - CPPUNIT_ASSERT(!c.getSecure()); + CPPUNIT_ASSERT_EQUAL(std::string("novalue"), c->getName()); + CPPUNIT_ASSERT_EQUAL(std::string(""), c->getValue()); + CPPUNIT_ASSERT_EQUAL((time_t)2147483647, c->getExpiryTime()); + CPPUNIT_ASSERT(c->getPersistent()); + CPPUNIT_ASSERT_EQUAL(std::string("/"), c->getPath()); + CPPUNIT_ASSERT_EQUAL(std::string("example.org"), c->getDomain()); + CPPUNIT_ASSERT(!c->getHostOnly()); + CPPUNIT_ASSERT(!c->getSecure()); c = cookies[2]; - CPPUNIT_ASSERT_EQUAL(std::string("JSESSIONID"), c.getName()); - CPPUNIT_ASSERT_EQUAL(std::string("123456789"), c.getValue()); - CPPUNIT_ASSERT_EQUAL((time_t)2147483647, c.getExpiryTime()); - CPPUNIT_ASSERT_EQUAL(std::string("/"), c.getPath()); - CPPUNIT_ASSERT_EQUAL(std::string("localhost"), c.getDomain()); - CPPUNIT_ASSERT(c.getHostOnly()); - CPPUNIT_ASSERT(c.getSecure()); + CPPUNIT_ASSERT_EQUAL(std::string("JSESSIONID"), c->getName()); + CPPUNIT_ASSERT_EQUAL(std::string("123456789"), c->getValue()); + CPPUNIT_ASSERT_EQUAL((time_t)2147483647, c->getExpiryTime()); + CPPUNIT_ASSERT_EQUAL(std::string("/"), c->getPath()); + CPPUNIT_ASSERT_EQUAL(std::string("localhost"), c->getDomain()); + CPPUNIT_ASSERT(c->getHostOnly()); + CPPUNIT_ASSERT(c->getSecure()); c = cookies[3]; - CPPUNIT_ASSERT_EQUAL(std::string("TAX"), c.getName()); - CPPUNIT_ASSERT_EQUAL(std::string("1000"), c.getValue()); - CPPUNIT_ASSERT((time_t)INT32_MAX <= c.getExpiryTime()); - CPPUNIT_ASSERT(c.getPersistent()); - CPPUNIT_ASSERT_EQUAL(std::string("/"), c.getPath()); - CPPUNIT_ASSERT_EQUAL(std::string("overflow"), c.getDomain()); - CPPUNIT_ASSERT(!c.getSecure()); + CPPUNIT_ASSERT_EQUAL(std::string("TAX"), c->getName()); + CPPUNIT_ASSERT_EQUAL(std::string("1000"), c->getValue()); + CPPUNIT_ASSERT((time_t)INT32_MAX <= c->getExpiryTime()); + CPPUNIT_ASSERT(c->getPersistent()); + CPPUNIT_ASSERT_EQUAL(std::string("/"), c->getPath()); + CPPUNIT_ASSERT_EQUAL(std::string("overflow"), c->getDomain()); + CPPUNIT_ASSERT(!c->getSecure()); } void CookieStorageTest::testLoad_sqlite3() { - CookieStorage st; + auto st = CookieStorage{}; #ifdef HAVE_SQLITE3 st.load(A2_TEST_DIR"/cookies.sqlite", 1000); CPPUNIT_ASSERT_EQUAL((size_t)2, st.size()); - std::vector<Cookie> cookies; - dumpCookie(cookies, st); - Cookie c = cookies[0]; - CPPUNIT_ASSERT_EQUAL(std::string("JSESSIONID"), c.getName()); - CPPUNIT_ASSERT_EQUAL(std::string("123456789"), c.getValue()); - CPPUNIT_ASSERT_EQUAL((time_t)INT32_MAX, c.getExpiryTime()); - CPPUNIT_ASSERT(c.getPersistent()); - CPPUNIT_ASSERT_EQUAL(std::string("localhost"), c.getDomain()); - CPPUNIT_ASSERT(c.getHostOnly()); - CPPUNIT_ASSERT_EQUAL(std::string("/"), c.getPath()); - CPPUNIT_ASSERT(c.getSecure()); + auto cookies = dumpCookie(st); + auto c = cookies[0]; + CPPUNIT_ASSERT_EQUAL(std::string("JSESSIONID"), c->getName()); + CPPUNIT_ASSERT_EQUAL(std::string("123456789"), c->getValue()); + CPPUNIT_ASSERT_EQUAL((time_t)INT32_MAX, c->getExpiryTime()); + CPPUNIT_ASSERT(c->getPersistent()); + CPPUNIT_ASSERT_EQUAL(std::string("localhost"), c->getDomain()); + CPPUNIT_ASSERT(c->getHostOnly()); + CPPUNIT_ASSERT_EQUAL(std::string("/"), c->getPath()); + CPPUNIT_ASSERT(c->getSecure()); c = cookies[1]; - CPPUNIT_ASSERT_EQUAL(std::string("foo"), c.getName()); - CPPUNIT_ASSERT_EQUAL(std::string("bar"), c.getValue()); - CPPUNIT_ASSERT((time_t)INT32_MAX <= c.getExpiryTime()); - CPPUNIT_ASSERT(c.getPersistent()); - CPPUNIT_ASSERT_EQUAL(std::string("overflow.time_t.org"), c.getDomain()); - CPPUNIT_ASSERT(!c.getHostOnly()); - CPPUNIT_ASSERT_EQUAL(std::string("/path/to"), c.getPath()); - CPPUNIT_ASSERT(!c.getSecure()); + CPPUNIT_ASSERT_EQUAL(std::string("foo"), c->getName()); + CPPUNIT_ASSERT_EQUAL(std::string("bar"), c->getValue()); + CPPUNIT_ASSERT((time_t)INT32_MAX <= c->getExpiryTime()); + CPPUNIT_ASSERT(c->getPersistent()); + CPPUNIT_ASSERT_EQUAL(std::string("overflow.time_t.org"), c->getDomain()); + CPPUNIT_ASSERT(!c->getHostOnly()); + CPPUNIT_ASSERT_EQUAL(std::string("/path/to"), c->getPath()); + CPPUNIT_ASSERT(!c->getSecure()); #else // !HAVE_SQLITE3 CPPUNIT_ASSERT(!st.load(A2_TEST_DIR"/cookies.sqlite", 1000)); @@ -345,7 +368,7 @@ void CookieStorageTest::testLoad_sqlite3() void CookieStorageTest::testLoad_fileNotfound() { - CookieStorage st; + auto st = CookieStorage{}; CPPUNIT_ASSERT(!st.load("./aria2_CookieStorageTest_testLoad_fileNotfound",0)); } @@ -354,22 +377,22 @@ void CookieStorageTest::testSaveNsFormat() // TODO add cookie with default domain std::string filename = A2_TEST_OUT_DIR"/aria2_CookieStorageTest_testSaveNsFormat"; File(filename).remove(); - CookieStorage st; + auto st = CookieStorage{}; time_t now = 1000; - st.store(Cookie(createCookie("favorite", "classic", "domain.org", false, - "/config",true)), now); - st.store(Cookie(createCookie("uid", "tujikawa", now, "domain.org", true, - "/",false)), now); + st.store(createCookie("favorite", "classic", "domain.org", false, + "/config",true), now); + st.store(createCookie("uid", "tujikawa", now, "domain.org", true, + "/",false), now); CPPUNIT_ASSERT(st.saveNsFormat(filename)); - CookieStorage loadst; + auto loadst = CookieStorage{}; loadst.load(filename, now); CPPUNIT_ASSERT_EQUAL((size_t)2, loadst.size()); - std::vector<Cookie> cookies; - dumpCookie(cookies, loadst); + auto cookies = dumpCookie(loadst); - CPPUNIT_ASSERT_EQUAL(std::string("favorite"), cookies[0].getName()); - CPPUNIT_ASSERT_EQUAL(std::string("uid"), cookies[1].getName()); + CPPUNIT_ASSERT_EQUAL((size_t)2, cookies.size()); + CPPUNIT_ASSERT_EQUAL(std::string("favorite"), cookies[0]->getName()); + CPPUNIT_ASSERT_EQUAL(std::string("uid"), cookies[1]->getName()); } void CookieStorageTest::testSaveNsFormat_fail() @@ -380,17 +403,16 @@ void CookieStorageTest::testSaveNsFormat_fail() f.remove(); f.mkdirs(); CPPUNIT_ASSERT(f.isDir()); - CookieStorage st; + auto st = CookieStorage{}; CPPUNIT_ASSERT(!st.saveNsFormat(filename)); } void CookieStorageTest::testCookieIsFull() { - CookieStorage st; + auto st = CookieStorage{}; for(size_t i = 0; i < CookieStorage::MAX_COOKIE_PER_DOMAIN+1; ++i) { - Cookie c(createCookie("k"+util::itos(i), "v", "aria2.org", false, - "/", false)); - st.store(c, 0); + st.store(createCookie("k"+util::itos(i), "v", "aria2.org", false, + "/", false), 0); } CPPUNIT_ASSERT_EQUAL((size_t)CookieStorage::MAX_COOKIE_PER_DOMAIN, st.size()); } @@ -399,13 +421,58 @@ void CookieStorageTest::testDomainIsFull() { // See DOMAIN_EVICTION_TRIGGER and DOMAIN_EVICTION_RATE in // CookieStorage.cc - CookieStorage st; + auto st = CookieStorage{}; for(size_t i = 0; i < 2001; ++i) { - Cookie c(createCookie("k", "v", "domain"+util::itos(i), true, - "/", false)); - st.store(c, 0); + st.store(createCookie("k", "v", "domain"+util::itos(i), true, + "/", false), 0); } CPPUNIT_ASSERT_EQUAL((size_t)1801, st.size()); } +void CookieStorageTest::testEviction() +{ + auto st = CookieStorage{}; + auto alpha = []() { + return createCookie("a", "alpha", "aria2.sf.net", false, "/", false); + }; + auto bravo = []() { + return createCookie("b", "bravo", "d.aria2.sf.net", false, "/", false); + }; + auto charlie = []() { + return createCookie("c", "charlie", "a2.github.com", false, "/", false); + }; + auto delta = []() { + return createCookie("d", "delta", "aria2.sf.net", false, "/", false); + }; + st.store(alpha(), 0); + CPPUNIT_ASSERT_EQUAL((size_t)1, st.getLruTrackerSize()); + st.store(bravo(), 1); + CPPUNIT_ASSERT_EQUAL((size_t)2, st.getLruTrackerSize()); + st.store(charlie(), 2); + CPPUNIT_ASSERT_EQUAL((size_t)3, st.getLruTrackerSize()); + st.store(delta(), 0); + CPPUNIT_ASSERT_EQUAL((size_t)3, st.getLruTrackerSize()); + + // aria2.sf.net will be evicted + st.evictNode(1); + CPPUNIT_ASSERT_EQUAL((size_t)2, st.getLruTrackerSize()); + CPPUNIT_ASSERT(!st.contains(*alpha())); + CPPUNIT_ASSERT(st.contains(*bravo())); + CPPUNIT_ASSERT(st.contains(*charlie())); + CPPUNIT_ASSERT(!st.contains(*delta())); + + // d.aria2.sf.net will be evicted + st.evictNode(1); + CPPUNIT_ASSERT_EQUAL((size_t)1, st.getLruTrackerSize()); + CPPUNIT_ASSERT(!st.contains(*bravo())); + CPPUNIT_ASSERT(st.contains(*charlie())); + + // a2.github.com will be evicted + st.evictNode(1); + CPPUNIT_ASSERT_EQUAL((size_t)0, st.getLruTrackerSize()); + CPPUNIT_ASSERT(!st.contains(*charlie())); + CPPUNIT_ASSERT_EQUAL((size_t)0, st.size()); + CPPUNIT_ASSERT(!st.getRootNode()->hasNext()); +} + } // namespace aria2 |
