SourceFormat Enforcement

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

diff --git a/src/tests/testStore.h b/src/tests/testStore.h
index dcf4a38d60a5e71ebdb9832e45f77cb48c0c1331..3229c897f7fd32f95a139fa39affe039a05dbd30 100644 (file)
--- a/src/tests/testStore.h
+++ b/src/tests/testStore.h
@@ -1,7 +1,17 @@
+/*
+ * Copyright (C) 1996-2017 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.
+ */
 
 #ifndef SQUID_SRC_TEST_STORE_H
 #define SQUID_SRC_TEST_STORE_H
 
+#include "Store.h"
+#include "store/Controlled.h"
+
 #include <cppunit/extensions/HelperMacros.h>
 
 /*
@@ -26,5 +36,47 @@ protected:
     void testMaxSize();
 };
 
+/// allows testing of methods without having all the other components live
+class TestStore : public Store::Controller
+{
+
+public:
+    TestStore() : statsCalled (false) {}
+
+    bool statsCalled;
+
+    virtual int callback();
+
+    virtual StoreEntry* get(const cache_key*);
+
+    virtual void get(String, void (*)(StoreEntry*, void*), void*);
+
+    virtual void init();
+
+    virtual void maintain() {};
+
+    virtual uint64_t maxSize() const;
+
+    virtual uint64_t minSize() const;
+
+    virtual uint64_t currentSize() const;
+
+    virtual uint64_t currentCount() const;
+
+    virtual int64_t maxObjectSize() const;
+
+    virtual void getStats(StoreInfoStats &) const;
+
+    virtual void stat(StoreEntry &) const; /* output stats to the provided store entry */
+
+    virtual void reference(StoreEntry &) {} /* Reference this object */
+
+    virtual bool dereference(StoreEntry &) { return true; }
+
+    virtual StoreSearch *search();
+};
+
+typedef RefCount<TestStore> TestStorePointer;
+
 #endif