SERVER-30212 moved test RAII class ForceSupportsDocLocking from d_concurrency_test.cpp to lock_manager_test_help.h

1 files changed, 19 insertions, 0 deletions

diff --git a/src/mongo/db/concurrency/lock_manager_test_help.h b/src/mongo/db/concurrency/lock_manager_test_help.h
index 79de7dcba50..d7c70515fab 100644
--- a/src/mongo/db/concurrency/lock_manager_test_help.h
+++ b/src/mongo/db/concurrency/lock_manager_test_help.h
@@ -70,4 +70,23 @@ public:
     }
 };
 
+/**
+ * A RAII object that temporarily forces setting of the _supportsDocLocking global variable (defined
+ * in db/service_context.cpp and returned by mongo::supportsDocLocking()) for testing purposes.
+ */
+extern bool _supportsDocLocking;
+class ForceSupportsDocLocking {
+public:
+    explicit ForceSupportsDocLocking(bool supported) : _oldSupportsDocLocking(_supportsDocLocking) {
+        _supportsDocLocking = supported;
+    }
+
+    ~ForceSupportsDocLocking() {
+        _supportsDocLocking = _oldSupportsDocLocking;
+    }
+
+private:
+    const bool _oldSupportsDocLocking;
+};
+
 }  // namespace mongo