Source Format Enforcement (#532)

[thirdparty/squid.git] / src / tests / testStore.cc

diff --git a/src/tests/testStore.cc b/src/tests/testStore.cc
index 0382776fe67f75a92db8303ce79ae8cc0cd96e2b..46d91c7e2b4bcbb2fff000f01760a0d1566daa53 100644 (file)
--- a/src/tests/testStore.cc
+++ b/src/tests/testStore.cc
@@ -1,8 +1,15 @@
-#define SQUID_UNIT_TEST 1
+/*
+ * Copyright (C) 1996-2020 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 "testStore.h"
 #include "Store.h"
+#include "testStore.h"
+#include "unitTestMain.h"
 
 CPPUNIT_TEST_SUITE_REGISTRATION( testStore );
 
@@ -23,7 +30,6 @@ TestStore::get(String, void (*)(StoreEntry*, void*), void*)
 {}
 
 void
-
 TestStore::init()
 {}
 
@@ -57,6 +63,11 @@ TestStore::maxObjectSize() const
     return 1;
 }
 
+void
+TestStore::getStats(StoreInfoStats &) const
+{
+}
+
 void
 TestStore::stat(StoreEntry &) const
 {
@@ -64,7 +75,7 @@ TestStore::stat(StoreEntry &) const
 }
 
 StoreSearch *
-TestStore::search(String const url, HttpRequest *)
+TestStore::search()
 {
     return NULL;
 }
@@ -72,40 +83,43 @@ TestStore::search(String const url, HttpRequest *)
 void
 testStore::testSetRoot()
 {
-    StorePointer aStore(new TestStore);
-    Store::Root(aStore);
+    Store::Controller *aStore(new TestStore);
+    Store::Init(aStore);
 
-    CPPUNIT_ASSERT(&Store::Root() == aStore.getRaw());
-    Store::Root(NULL);
+    CPPUNIT_ASSERT_EQUAL(&Store::Root(), aStore);
+    Store::FreeMemory();
 }
 
 void
 testStore::testUnsetRoot()
 {
-    StorePointer aStore(new TestStore);
-    StorePointer aStore2(new TestStore);
-    Store::Root(aStore);
-    Store::Root(aStore2);
-    CPPUNIT_ASSERT(&Store::Root() == aStore2.getRaw());
-    Store::Root(NULL);
+    Store::Controller *aStore(new TestStore);
+    Store::Controller *aStore2(new TestStore);
+    Store::Init(aStore);
+    Store::FreeMemory();
+    Store::Init(aStore2);
+    CPPUNIT_ASSERT_EQUAL(&Store::Root(),aStore2);
+    Store::FreeMemory();
 }
 
 void
 testStore::testStats()
 {
-    TestStorePointer aStore(new TestStore);
-    Store::Root(aStore.getRaw());
-    CPPUNIT_ASSERT(aStore->statsCalled == false);
-    Store::Stats(NullStoreEntry::getInstance());
-    CPPUNIT_ASSERT(aStore->statsCalled == true);
-    Store::Root(NULL);
+    TestStore *aStore(new TestStore);
+    Store::Init(aStore);
+    CPPUNIT_ASSERT_EQUAL(false, aStore->statsCalled);
+    StoreEntry entry;
+    Store::Stats(&entry);
+    CPPUNIT_ASSERT_EQUAL(true, aStore->statsCalled);
+    Store::FreeMemory();
 }
 
 void
 testStore::testMaxSize()
 {
-    StorePointer aStore(new TestStore);
-    Store::Root(aStore.getRaw());
-    CPPUNIT_ASSERT(aStore->maxSize() == 3);
-    Store::Root(NULL);
+    Store::Controller *aStore(new TestStore);
+    Store::Init(aStore);
+    CPPUNIT_ASSERT_EQUAL(static_cast<uint64_t>(3), aStore->maxSize());
+    Store::FreeMemory();
 }
+