]>
Commit | Line | Data |
---|---|---|
ec94e362 AJ |
1 | #define SQUID_UNIT_TEST 1 |
2 | ||
582c2af2 | 3 | #include "squid.h" |
c8f4eac4 | 4 | #include "testStoreController.h" |
c8f4eac4 | 5 | #include "Store.h" |
6 | #include "SwapDir.h" | |
7 | #include "TestSwapDir.h" | |
8 | #include "Mem.h" | |
985c86bc | 9 | #include "SquidTime.h" |
c8f4eac4 | 10 | #include "StoreSearch.h" |
11 | ||
12 | CPPUNIT_TEST_SUITE_REGISTRATION( testStoreController ); | |
13 | ||
14 | static void | |
15 | addSwapDir(TestSwapDirPointer aStore) | |
16 | { | |
17 | allocate_new_swapdir(&Config.cacheSwap); | |
18 | Config.cacheSwap.swapDirs[Config.cacheSwap.n_configured] = aStore.getRaw(); | |
19 | ++Config.cacheSwap.n_configured; | |
20 | } | |
21 | ||
c8f4eac4 | 22 | void |
23 | testStoreController::testStats() | |
24 | { | |
25 | StoreEntry * logEntry = new StoreEntry("dummy_url", "dummy_log_url"); | |
26 | logEntry->store_status = STORE_PENDING; | |
27 | StorePointer aRoot (new StoreController); | |
28 | Store::Root(aRoot); | |
29 | TestSwapDirPointer aStore (new TestSwapDir); | |
30 | TestSwapDirPointer aStore2 (new TestSwapDir); | |
31 | addSwapDir(aStore); | |
32 | addSwapDir(aStore2); | |
33 | CPPUNIT_ASSERT(aStore->statsCalled == false); | |
34 | CPPUNIT_ASSERT(aStore2->statsCalled == false); | |
35 | Store::Stats(logEntry); | |
36 | free_cachedir(&Config.cacheSwap); | |
37 | CPPUNIT_ASSERT(aStore->statsCalled == true); | |
38 | CPPUNIT_ASSERT(aStore2->statsCalled == true); | |
39 | Store::Root(NULL); | |
40 | } | |
41 | ||
42 | static void | |
43 | commonInit() | |
44 | { | |
45 | static bool inited = false; | |
46 | ||
47 | if (inited) | |
48 | return; | |
49 | ||
50 | Mem::Init(); | |
51 | ||
c8f4eac4 | 52 | Config.Store.avgObjectSize = 1024; |
53 | ||
54 | Config.Store.objectsPerBucket = 20; | |
55 | ||
56 | Config.Store.maxObjectSize = 2048; | |
57 | ||
58 | Config.store_dir_select_algorithm = xstrdup("round-robin"); | |
59 | } | |
60 | ||
61 | void | |
62 | testStoreController::testMaxSize() | |
63 | { | |
64 | commonInit(); | |
65 | StoreEntry * logEntry = new StoreEntry("dummy_url", "dummy_log_url"); | |
66 | logEntry->store_status = STORE_PENDING; | |
67 | StorePointer aRoot (new StoreController); | |
68 | Store::Root(aRoot); | |
69 | TestSwapDirPointer aStore (new TestSwapDir); | |
70 | TestSwapDirPointer aStore2 (new TestSwapDir); | |
71 | addSwapDir(aStore); | |
72 | addSwapDir(aStore2); | |
73 | CPPUNIT_ASSERT(Store::Root().maxSize() == 6); | |
74 | free_cachedir(&Config.cacheSwap); | |
75 | Store::Root(NULL); | |
76 | } | |
77 | ||
78 | static StoreEntry * | |
79 | addedEntry(StorePointer hashStore, | |
80 | StorePointer aStore, | |
30abd221 | 81 | String name, |
82 | String varySpec, | |
83 | String varyKey | |
84 | ||
c8f4eac4 | 85 | ) |
86 | { | |
87 | StoreEntry *e = new StoreEntry(); | |
88 | e->store_status = STORE_OK; | |
3900307b | 89 | e->setMemStatus(NOT_IN_MEMORY); |
c8f4eac4 | 90 | e->swap_status = SWAPOUT_DONE; /* bogus haha */ |
91 | e->swap_filen = 0; /* garh - lower level*/ | |
92 | e->swap_dirn = -1; | |
93 | ||
d7ae3534 | 94 | for (int i=0; i < Config.cacheSwap.n_configured; ++i) { |
c8f4eac4 | 95 | if (INDEXSD (i) == aStore.getRaw()) |
96 | e->swap_dirn = i; | |
97 | } | |
98 | ||
99 | CPPUNIT_ASSERT (e->swap_dirn != -1); | |
100 | e->swap_file_sz = 0; /* garh lower level */ | |
101 | e->lock_count = 0; | |
102 | e->lastref = squid_curtime; | |
103 | e->timestamp = squid_curtime; | |
104 | e->expires = squid_curtime; | |
105 | e->lastmod = squid_curtime; | |
106 | e->refcount = 1; | |
107 | EBIT_SET(e->flags, ENTRY_CACHABLE); | |
108 | EBIT_CLR(e->flags, RELEASE_REQUEST); | |
109 | EBIT_CLR(e->flags, KEY_PRIVATE); | |
110 | e->ping_status = PING_NONE; | |
111 | EBIT_CLR(e->flags, ENTRY_VALIDATED); | |
5b4117d8 | 112 | e->hashInsert((const cache_key *)name.termedBuf()); /* do it after we clear KEY_PRIVATE */ |
c8f4eac4 | 113 | return e; |
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 | testStoreController::testSearch() | |
128 | { | |
129 | commonInit(); | |
130 | StorePointer aRoot (new StoreController()); | |
131 | Store::Root(aRoot); | |
132 | TestSwapDirPointer aStore (new TestSwapDir); | |
133 | TestSwapDirPointer aStore2 (new TestSwapDir); | |
134 | addSwapDir(aStore); | |
135 | addSwapDir(aStore2); | |
136 | Store::Root().init(); | |
137 | StoreEntry * entry1 = addedEntry (&Store::Root(), aStore.getRaw(), "name", NULL, NULL); | |
138 | StoreEntry * entry2 = addedEntry (&Store::Root(), aStore2.getRaw(), "name2", NULL, NULL); | |
139 | StoreSearchPointer search = aRoot->search (NULL, NULL); /* search for everything in the store */ | |
140 | ||
141 | /* nothing should be immediately available */ | |
142 | CPPUNIT_ASSERT(search->error() == false); | |
143 | CPPUNIT_ASSERT(search->isDone() == false); | |
144 | CPPUNIT_ASSERT(search->currentItem() == NULL); | |
145 | #if 0 | |
146 | ||
147 | CPPUNIT_ASSERT(search->next() == false); | |
148 | #endif | |
149 | ||
150 | /* trigger a callback */ | |
151 | cbcalled = false; | |
152 | search->next(searchCallback, NULL); | |
153 | CPPUNIT_ASSERT(cbcalled == true); | |
154 | ||
155 | /* we should have access to a entry now, that matches the entry we had before */ | |
156 | CPPUNIT_ASSERT(search->error() == false); | |
157 | CPPUNIT_ASSERT(search->isDone() == false); | |
158 | /* note the hash order is random - the test happens to be in a nice order */ | |
159 | CPPUNIT_ASSERT(search->currentItem() == entry1); | |
160 | //CPPUNIT_ASSERT(search->next() == false); | |
161 | ||
162 | /* trigger another callback */ | |
163 | cbcalled = false; | |
164 | search->next(searchCallback, NULL); | |
165 | CPPUNIT_ASSERT(cbcalled == true); | |
166 | ||
167 | /* we should have access to a entry now, that matches the entry we had before */ | |
168 | CPPUNIT_ASSERT(search->error() == false); | |
169 | CPPUNIT_ASSERT(search->isDone() == false); | |
170 | CPPUNIT_ASSERT(search->currentItem() == entry2); | |
171 | //CPPUNIT_ASSERT(search->next() == false); | |
172 | ||
173 | /* trigger another callback */ | |
174 | cbcalled = false; | |
175 | search->next(searchCallback, NULL); | |
176 | CPPUNIT_ASSERT(cbcalled == true); | |
177 | ||
178 | /* now we should have no error, we should have finished and have no current item */ | |
179 | CPPUNIT_ASSERT(search->error() == false); | |
180 | CPPUNIT_ASSERT(search->isDone() == true); | |
181 | CPPUNIT_ASSERT(search->currentItem() == NULL); | |
182 | //CPPUNIT_ASSERT(search->next() == false); | |
183 | ||
184 | Store::Root(NULL); | |
185 | } |