SERVER-68490 Robustify ce_histogram.js

1 files changed, 17 insertions, 4 deletions

diff --git a/jstests/cqf/ce_histogram.js b/jstests/cqf/analyze/ce_histogram.js
index 58d9c43b829..7667b641655 100644
--- a/jstests/cqf/ce_histogram.js
+++ b/jstests/cqf/analyze/ce_histogram.js
@@ -274,8 +274,21 @@ if (checkSBEEnabled(db, ["featureFlagSbeFull"], true)) {
     return;
 }
 
-verifyCEForNDV(1);
-verifyCEForNDV(2);
-verifyCEForNDV(3);
-verifyCEForNDV(10);
+// We will be updating some query knobs, so store the old state and restore it after the test.
+const {internalQueryCardinalityEstimatorMode, internalQueryFrameworkControl} = db.adminCommand({
+    getParameter: 1,
+    internalQueryCardinalityEstimatorMode: 1,
+    internalQueryFrameworkControl: 1,
+});
+
+try {
+    verifyCEForNDV(1);
+    verifyCEForNDV(2);
+    verifyCEForNDV(3);
+    verifyCEForNDV(10);
+} finally {
+    // Reset query knobs to their original state.
+    assert.commandWorked(db.adminCommand(
+        {setParameter: 1, internalQueryCardinalityEstimatorMode, internalQueryFrameworkControl}));
+}
 }());