summaryrefslogtreecommitdiff
path: root/src/mongo/dbtests
diff options
context:
space:
mode:
authorMaria van Keulen <maria@mongodb.com>2019-05-02 15:15:17 -0400
committerMaria van Keulen <maria@mongodb.com>2019-05-08 15:00:26 -0400
commit4cfe7bd17640acd296fad4d916aecc27ba8b5f4f (patch)
tree946923d34eb76d45b95b55556809940679577dff /src/mongo/dbtests
parentd3ee35d6e3ac7a42cd5ad106c3ecb9fb554900c7 (diff)
downloadmongo-4cfe7bd17640acd296fad4d916aecc27ba8b5f4f.tar.gz
SERVER-40823 Enable Flow Control in testing
Diffstat (limited to 'src/mongo/dbtests')
-rw-r--r--src/mongo/dbtests/SConscript1
-rw-r--r--src/mongo/dbtests/framework_options.cpp6
-rw-r--r--src/mongo/dbtests/framework_options.idl5
3 files changed, 12 insertions, 0 deletions
diff --git a/src/mongo/dbtests/SConscript b/src/mongo/dbtests/SConscript
index ef20fb6a986..8f9f1286a5a 100644
--- a/src/mongo/dbtests/SConscript
+++ b/src/mongo/dbtests/SConscript
@@ -16,6 +16,7 @@ env.Library(
],
LIBDEPS=[
'$BUILD_DIR/mongo/db/server_options_core',
+ '$BUILD_DIR/mongo/db/storage/flow_control_parameters',
'$BUILD_DIR/mongo/db/storage/storage_options',
'$BUILD_DIR/mongo/util/options_parser/options_parser_init',
'$BUILD_DIR/mongo/unittest/unittest',
diff --git a/src/mongo/dbtests/framework_options.cpp b/src/mongo/dbtests/framework_options.cpp
index c8940e3df68..ea4f54b65d8 100644
--- a/src/mongo/dbtests/framework_options.cpp
+++ b/src/mongo/dbtests/framework_options.cpp
@@ -40,6 +40,7 @@
#include "mongo/base/status.h"
#include "mongo/bson/util/builder.h"
#include "mongo/db/query/find.h"
+#include "mongo/db/storage/flow_control_parameters_gen.h"
#include "mongo/db/storage/storage_options.h"
#include "mongo/dbtests/dbtests.h"
#include "mongo/unittest/unittest.h"
@@ -123,6 +124,11 @@ Status storeTestFrameworkOptions(const moe::Environment& params,
storageGlobalParams.dbpath = dbpathString.c_str();
storageGlobalParams.engine = params["storage.engine"].as<string>();
+ gFlowControlEnabled.store(params["enableFlowControl"].as<bool>());
+
+ if (gFlowControlEnabled.load()) {
+ log() << "Flow Control enabled" << endl;
+ }
if (storageGlobalParams.engine == "wiredTiger" &&
params.count("replication.enableMajorityReadConcern")) {
diff --git a/src/mongo/dbtests/framework_options.idl b/src/mongo/dbtests/framework_options.idl
index 29311cae79a..a4be3951ee7 100644
--- a/src/mongo/dbtests/framework_options.idl
+++ b/src/mongo/dbtests/framework_options.idl
@@ -93,6 +93,11 @@ configs:
arg_vartype: String
short_name: storageEngine
default: wiredTiger
+ enableFlowControl:
+ description: 'Whether Flow Control is enabled'
+ arg_vartype: Bool
+ short_name: flowControl
+ default: true
suites:
description: 'Test suites to run'
arg_vartype: StringVector