SERVER-41842 Collapse unit tests under storage

1 files changed, 16 insertions, 0 deletions

diff --git a/src/mongo/db/storage/recovery_unit_test_harness.cpp b/src/mongo/db/storage/recovery_unit_test_harness.cpp
index fd1fe3bab35..86382c10785 100644
--- a/src/mongo/db/storage/recovery_unit_test_harness.cpp
+++ b/src/mongo/db/storage/recovery_unit_test_harness.cpp
@@ -37,6 +37,22 @@
 
 namespace mongo {
 namespace {
+std::function<std::unique_ptr<RecoveryUnitHarnessHelper>()> recoveryUnitHarnessFactory;
+}
+}  // namespace mongo
+
+void mongo::registerRecoveryUnitHarnessHelperFactory(
+    std::function<std::unique_ptr<RecoveryUnitHarnessHelper>()> factory) {
+    recoveryUnitHarnessFactory = std::move(factory);
+}
+
+namespace mongo {
+
+auto newRecoveryUnitHarnessHelper() -> std::unique_ptr<RecoveryUnitHarnessHelper> {
+    return recoveryUnitHarnessFactory();
+}
+
+namespace {
 
 class RecoveryUnitTestHarness : public unittest::Test {
 public: