/* * Copyright (C) 1996-2015 The Squid Software Foundation and contributors * * Squid software is distributed under GPLv2+ license and includes * contributions from numerous individuals and organizations. * Please see the COPYING and CONTRIBUTORS files for details. */ #include "squid.h" #include "Mem.h" #include "MemObject.h" #include "SquidConfig.h" #include "SquidTime.h" #include "Store.h" #include "StoreSearch.h" #include "SwapDir.h" #include "testStoreController.h" #include "TestSwapDir.h" CPPUNIT_TEST_SUITE_REGISTRATION( testStoreController ); static void addSwapDir(TestSwapDirPointer aStore) { allocate_new_swapdir(&Config.cacheSwap); Config.cacheSwap.swapDirs[Config.cacheSwap.n_configured] = aStore.getRaw(); ++Config.cacheSwap.n_configured; } void testStoreController::testStats() { StoreEntry *logEntry = new StoreEntry; logEntry->makeMemObject(); logEntry->mem_obj->setUris("dummy_storeId", NULL, HttpRequestMethod()); logEntry->store_status = STORE_PENDING; StorePointer aRoot (new StoreController); Store::Root(aRoot); TestSwapDirPointer aStore (new TestSwapDir); TestSwapDirPointer aStore2 (new TestSwapDir); addSwapDir(aStore); addSwapDir(aStore2); CPPUNIT_ASSERT_EQUAL(false, aStore->statsCalled); CPPUNIT_ASSERT_EQUAL(false, aStore2->statsCalled); Store::Stats(logEntry); free_cachedir(&Config.cacheSwap); CPPUNIT_ASSERT_EQUAL(true, aStore->statsCalled); CPPUNIT_ASSERT_EQUAL(true, aStore2->statsCalled); Store::Root(NULL); } static void commonInit() { static bool inited = false; if (inited) return; Mem::Init(); Config.Store.avgObjectSize = 1024; Config.Store.objectsPerBucket = 20; Config.Store.maxObjectSize = 2048; Config.store_dir_select_algorithm = xstrdup("round-robin"); } void testStoreController::testMaxSize() { commonInit(); StoreEntry *logEntry = new StoreEntry; logEntry->makeMemObject(); logEntry->mem_obj->setUris("dummy_storeId", NULL, HttpRequestMethod()); logEntry->store_status = STORE_PENDING; StorePointer aRoot (new StoreController); Store::Root(aRoot); TestSwapDirPointer aStore (new TestSwapDir); TestSwapDirPointer aStore2 (new TestSwapDir); addSwapDir(aStore); addSwapDir(aStore2); CPPUNIT_ASSERT_EQUAL(static_cast(6), Store::Root().maxSize()); free_cachedir(&Config.cacheSwap); Store::Root(NULL); } static StoreEntry * addedEntry(StorePointer hashStore, StorePointer aStore, String name, String varySpec, String varyKey ) { StoreEntry *e = new StoreEntry(); e->store_status = STORE_OK; e->setMemStatus(NOT_IN_MEMORY); e->swap_status = SWAPOUT_DONE; /* bogus haha */ e->swap_filen = 0; /* garh - lower level*/ e->swap_dirn = -1; for (int i=0; i < Config.cacheSwap.n_configured; ++i) { if (INDEXSD (i) == aStore.getRaw()) e->swap_dirn = i; } CPPUNIT_ASSERT (e->swap_dirn != -1); e->swap_file_sz = 0; /* garh lower level */ e->lastref = squid_curtime; e->timestamp = squid_curtime; e->expires = squid_curtime; e->lastmod = squid_curtime; e->refcount = 1; EBIT_CLR(e->flags, RELEASE_REQUEST); EBIT_CLR(e->flags, KEY_PRIVATE); e->ping_status = PING_NONE; EBIT_CLR(e->flags, ENTRY_VALIDATED); e->hashInsert((const cache_key *)name.termedBuf()); /* do it after we clear KEY_PRIVATE */ return e; } /* TODO make this a cbdata class */ static bool cbcalled; static void searchCallback(void *cbdata) { cbcalled = true; } void testStoreController::testSearch() { commonInit(); StorePointer aRoot (new StoreController()); Store::Root(aRoot); TestSwapDirPointer aStore (new TestSwapDir); TestSwapDirPointer aStore2 (new TestSwapDir); addSwapDir(aStore); addSwapDir(aStore2); Store::Root().init(); StoreEntry * entry1 = addedEntry (&Store::Root(), aStore.getRaw(), "name", NULL, NULL); StoreEntry * entry2 = addedEntry (&Store::Root(), aStore2.getRaw(), "name2", NULL, NULL); StoreSearchPointer search = aRoot->search (NULL, NULL); /* search for everything in the store */ /* nothing should be immediately available */ CPPUNIT_ASSERT_EQUAL(false, search->error()); CPPUNIT_ASSERT_EQUAL(false, search->isDone()); CPPUNIT_ASSERT_EQUAL(static_cast(NULL), search->currentItem()); #if 0 CPPUNIT_ASSERT_EQUAL(false, search->next()); #endif /* trigger a callback */ cbcalled = false; search->next(searchCallback, NULL); CPPUNIT_ASSERT_EQUAL(true, cbcalled); /* we should have access to a entry now, that matches the entry we had before */ CPPUNIT_ASSERT_EQUAL(false, search->error()); CPPUNIT_ASSERT_EQUAL(false, search->isDone()); /* note the hash order is random - the test happens to be in a nice order */ CPPUNIT_ASSERT_EQUAL(entry1, search->currentItem()); //CPPUNIT_ASSERT_EQUAL(false, search->next()); /* trigger another callback */ cbcalled = false; search->next(searchCallback, NULL); CPPUNIT_ASSERT_EQUAL(true, cbcalled); /* we should have access to a entry now, that matches the entry we had before */ CPPUNIT_ASSERT_EQUAL(false, search->error()); CPPUNIT_ASSERT_EQUAL(false, search->isDone()); CPPUNIT_ASSERT_EQUAL(entry2, search->currentItem()); //CPPUNIT_ASSERT_EQUAL(false, search->next()); /* trigger another callback */ cbcalled = false; search->next(searchCallback, NULL); CPPUNIT_ASSERT_EQUAL(true, cbcalled); /* now we should have no error, we should have finished and have no current item */ CPPUNIT_ASSERT_EQUAL(false, search->error()); CPPUNIT_ASSERT_EQUAL(true, search->isDone()); CPPUNIT_ASSERT_EQUAL(static_cast(NULL), search->currentItem()); //CPPUNIT_ASSERT_EQUAL(false, search->next()); Store::Root(NULL); }