src/tests/testStoreHashIndex.cc

   1 /*
   2  * Copyright (C) 1996-2018 The Squid Software Foundation and contributors
   3  *
   4  * Squid software is distributed under GPLv2+ license and includes
   5  * contributions from numerous individuals and organizations.
   6  * Please see the COPYING and CONTRIBUTORS files for details.
   7  */
   8
   9 #include "squid.h"
  10 #include "MemObject.h"
  11 #include "SquidConfig.h"
  12 #include "SquidTime.h"
  13 #include "Store.h"
  14 #include "store/Disks.h"
  15 #include "StoreSearch.h"
  16 #include "testStoreHashIndex.h"
  17 #include "TestSwapDir.h"
  18
  19 CPPUNIT_TEST_SUITE_REGISTRATION( testStoreHashIndex );
  20
  21 static void
  22 addSwapDir(TestSwapDirPointer aStore)
  23 {
  24     allocate_new_swapdir(&Config.cacheSwap);
  25     Config.cacheSwap.swapDirs[Config.cacheSwap.n_configured] = aStore.getRaw();
  26     ++Config.cacheSwap.n_configured;
  27 }
  28
  29 void
  30 testStoreHashIndex::testStats()
  31 {
  32     StoreEntry *logEntry = new StoreEntry;
  33     logEntry->createMemObject("dummy_storeId", NULL, HttpRequestMethod());
  34     logEntry->store_status = STORE_PENDING;
  35     Store::Init();
  36     TestSwapDirPointer aStore (new TestSwapDir);
  37     TestSwapDirPointer aStore2 (new TestSwapDir);
  38     addSwapDir(aStore);
  39     addSwapDir(aStore2);
  40     CPPUNIT_ASSERT_EQUAL(false, aStore->statsCalled);
  41     CPPUNIT_ASSERT_EQUAL(false, aStore2->statsCalled);
  42     Store::Stats(logEntry);
  43     free_cachedir(&Config.cacheSwap);
  44     CPPUNIT_ASSERT_EQUAL(true, aStore->statsCalled);
  45     CPPUNIT_ASSERT_EQUAL(true, aStore2->statsCalled);
  46     Store::FreeMemory();
  47 }
  48
  49 void
  50 testStoreHashIndex::testMaxSize()
  51 {
  52     StoreEntry *logEntry = new StoreEntry;
  53     logEntry->createMemObject("dummy_storeId", NULL, HttpRequestMethod());
  54     logEntry->store_status = STORE_PENDING;
  55     Store::Init();
  56     TestSwapDirPointer aStore (new TestSwapDir);
  57     TestSwapDirPointer aStore2 (new TestSwapDir);
  58     addSwapDir(aStore);
  59     addSwapDir(aStore2);
  60     CPPUNIT_ASSERT_EQUAL(static_cast<uint64_t>(6), Store::Root().maxSize());
  61     free_cachedir(&Config.cacheSwap);
  62     Store::FreeMemory();
  63 }
  64
  65 StoreEntry *
  66 addedEntry(Store::Disk *aStore,
  67            String name,
  68            String varySpec,
  69            String varyKey
  70
  71           )
  72 {
  73     StoreEntry *e = new StoreEntry();
  74     e->store_status = STORE_OK;
  75     e->setMemStatus(NOT_IN_MEMORY);
  76     e->swap_status = SWAPOUT_DONE; /* bogus haha */
  77     e->swap_filen = 0; /* garh - lower level*/
  78     e->swap_dirn = -1;
  79
  80     for (int i=0; i < Config.cacheSwap.n_configured; ++i) {
  81         if (INDEXSD(i) == aStore)
  82             e->swap_dirn = i;
  83     }
  84
  85     CPPUNIT_ASSERT (e->swap_dirn != -1);
  86     e->swap_file_sz = 0; /* garh lower level */
  87     e->lastref = squid_curtime;
  88     e->timestamp = squid_curtime;
  89     e->expires = squid_curtime;
  90     e->lastModified(squid_curtime);
  91     e->refcount = 1;
  92     EBIT_CLR(e->flags, RELEASE_REQUEST);
  93     e->clearPrivate();
  94     e->ping_status = PING_NONE;
  95     EBIT_CLR(e->flags, ENTRY_VALIDATED);
  96     e->hashInsert((const cache_key *)name.termedBuf()); /* do it after we clear KEY_PRIVATE */
  97     return e;
  98 }
  99
 100 void commonInit()
 101 {
 102     static bool inited = false;
 103
 104     if (inited)
 105         return;
 106
 107     Mem::Init();
 108
 109     Config.Store.avgObjectSize = 1024;
 110
 111     Config.Store.objectsPerBucket = 20;
 112
 113     Config.Store.maxObjectSize = 2048;
 114 }
 115
 116 /* TODO make this a cbdata class */
 117
 118 static bool cbcalled;
 119
 120 static void
 121 searchCallback(void *cbdata)
 122 {
 123     cbcalled = true;
 124 }
 125
 126 void
 127 testStoreHashIndex::testSearch()
 128 {
 129     commonInit();
 130     Store::Init();
 131     TestSwapDirPointer aStore (new TestSwapDir);
 132     TestSwapDirPointer aStore2 (new TestSwapDir);
 133     addSwapDir(aStore);
 134     addSwapDir(aStore2);
 135     Store::Root().init();
 136     StoreEntry * entry1 = addedEntry(aStore.getRaw(), "name", NULL, NULL);
 137     StoreEntry * entry2 = addedEntry(aStore2.getRaw(), "name2", NULL, NULL);
 138     StoreSearchPointer search = Store::Root().search(); /* search for everything in the store */
 139
 140     /* nothing should be immediately available */
 141     CPPUNIT_ASSERT_EQUAL(false, search->error());
 142     CPPUNIT_ASSERT_EQUAL(false, search->isDone());
 143     CPPUNIT_ASSERT_EQUAL(static_cast<StoreEntry *>(NULL), search->currentItem());
 144 #if 0
 145
 146     CPPUNIT_ASSERT_EQUAL(false, search->next());
 147 #endif
 148
 149     /* trigger a callback */
 150     cbcalled = false;
 151     search->next(searchCallback, NULL);
 152     CPPUNIT_ASSERT_EQUAL(true, cbcalled);
 153
 154     /* we should have access to a entry now, that matches the entry we had before */
 155     CPPUNIT_ASSERT_EQUAL(false, search->error());
 156     CPPUNIT_ASSERT_EQUAL(false, search->isDone());
 157     /* note the hash order is random - the test happens to be in a nice order */
 158     CPPUNIT_ASSERT_EQUAL(entry1, search->currentItem());
 159     //CPPUNIT_ASSERT_EQUAL(false, search->next());
 160
 161     /* trigger another callback */
 162     cbcalled = false;
 163     search->next(searchCallback, NULL);
 164     CPPUNIT_ASSERT_EQUAL(true, cbcalled);
 165
 166     /* we should have access to a entry now, that matches the entry we had before */
 167     CPPUNIT_ASSERT_EQUAL(false, search->error());
 168     CPPUNIT_ASSERT_EQUAL(false, search->isDone());
 169     CPPUNIT_ASSERT_EQUAL(entry2, search->currentItem());
 170     //CPPUNIT_ASSERT_EQUAL(false, search->next());
 171
 172     /* trigger another callback */
 173     cbcalled = false;
 174     search->next(searchCallback, NULL);
 175     CPPUNIT_ASSERT_EQUAL(true, cbcalled);
 176
 177     /* now we should have no error, we should have finished and have no current item */
 178     CPPUNIT_ASSERT_EQUAL(false, search->error());
 179     CPPUNIT_ASSERT_EQUAL(true, search->isDone());
 180     CPPUNIT_ASSERT_EQUAL(static_cast<StoreEntry *>(NULL), search->currentItem());
 181     //CPPUNIT_ASSERT_EQUAL(false, search->next());
 182
 183     Store::FreeMemory();
 184 }
 185