1 files changed, 52 insertions, 0 deletions
diff --git a/src/test/objectstore/store_test_fixture.h b/src/test/objectstore/store_test_fixture.h
new file mode 100644
index 000000000..3b7971d7c
--- /dev/null
+++ b/src/test/objectstore/store_test_fixture.h
@@ -0,0 +1,52 @@
+#include <string>
+#include <stack>
+#include <boost/scoped_ptr.hpp>
+#include <gtest/gtest.h>
+#include "common/config_fwd.h"
+
+class ObjectStore;
+
+class StoreTestFixture : virtual public ::testing::Test {
+  const std::string type;
+  const std::string data_dir;
+
+  std::stack<std::pair<std::string, std::string>> saved_settings;
+  ConfigProxy* conf = nullptr;
+
+  std::string orig_death_test_style;
+
+public:
+  boost::scoped_ptr<ObjectStore> store;
+  ObjectStore::CollectionHandle ch;
+
+  explicit StoreTestFixture(const std::string& type)
+    : type(type), data_dir(type + ".test_temp_dir")
+  {}
+
+  void SetUp() override;
+  void TearDown() override;
+  void SetDeathTestStyle(const char* new_style) {
+    if (orig_death_test_style.empty()) {
+      orig_death_test_style = ::testing::FLAGS_gtest_death_test_style;
+    }
+    ::testing::FLAGS_gtest_death_test_style = new_style;
+  }
+
+  void SetVal(ConfigProxy& conf, const char* key, const char* val);
+  struct SettingsBookmark {
+    StoreTestFixture& s;
+    size_t pos;
+
+    SettingsBookmark(StoreTestFixture& _s, size_t p) : s(_s), pos(p)
+    {}
+
+    ~SettingsBookmark() {
+      s.PopSettings(pos);
+    }
+  };
+  SettingsBookmark BookmarkSettings() {
+    return SettingsBookmark(*this, saved_settings.size());
+  }
+  void PopSettings(size_t);
+  void CloseAndReopen();
+};