SERVER-23267 Add currentOp tests for ns, command & planSummary

1 files changed, 232 insertions, 0 deletions

diff --git a/jstests/noPassthrough/currentop_query.js b/jstests/noPassthrough/currentop_query.js
new file mode 100644
index 00000000000..c9820149791
--- /dev/null
+++ b/jstests/noPassthrough/currentop_query.js
@@ -0,0 +1,232 @@
+/**
+ * Confirms inclusion of query, command object and planSummary in currentOp() for CRUD operations.
+ * This test should not be run in the parallel suite as it sets fail points.
+ */
+(function() {
+    "use strict";
+
+    var conn = MongoRunner.runMongod({smallfiles: "", nojournal: ""});
+    assert.neq(null, conn, "mongod was unable to start up");
+
+    var testDB = conn.getDB("test");
+    assert.commandWorked(testDB.dropDatabase());
+
+    var coll = testDB.currentop_query;
+
+    // Force yield to occur on every PlanExecutor iteration.
+    assert.commandWorked(
+        testDB.adminCommand({setParameter: 1, internalQueryExecYieldIterations: 1}));
+
+    /**
+     * Captures currentOp() for a given test command/operation and confirms that namespace,
+     * operation type and planSummary are correct.
+     *
+     *  @param {Object} testObj - Contains test arguments.
+     *  @param {function} testObj.test - A function that runs the desired test op/cmd.
+     *  @param {string} testObj.planSummary - A string containing the expected planSummary.
+     *  @param {Object} testObj.currentOpFilter - A filter to be used to narrow currentOp() output
+     *  to only the relevant operation or command.
+     *  @param {string} [testObj.command] - The command to test against. Will look for this to be
+     *  a key in the currentOp().query object.
+     *  @param {string} [testObj.operation] - The operation to test against. Will look for this
+     *  to be the value of the currentOp().op field.
+     */
+    function confirmCurrentOpContents(testObj) {
+        // Force queries to hang on yield to allow for currentOp capture.
+        assert.commandWorked(
+            testDB.adminCommand({configureFailPoint: "setYieldAllLocksHang", mode: "alwaysOn"}));
+
+        // Run query.
+        var awaitShell = startParallelShell(testObj.test, conn.port);
+
+        // Capture currentOp record for the query and confirm that the 'query' and 'planSummary'
+        // fields contain the content expected. We are indirectly testing the 'ns' field as well
+        // with the currentOp query argument.
+        assert.soon(
+            function() {
+                testObj.currentOpFilter.ns = coll.getFullName();
+                testObj.currentOpFilter.planSummary = testObj.planSummary;
+                if (testObj.hasOwnProperty("command")) {
+                    testObj.currentOpFilter["query." + testObj.command] = {
+                        $exists: true
+                    };
+                } else if (testObj.hasOwnProperty("operation")) {
+                    testObj.currentOpFilter.op = testObj.operation;
+                }
+
+                var result = testDB.currentOp(testObj.currentOpFilter);
+                assert.commandWorked(result);
+
+                if (result.inprog.length === 1) {
+                    return true;
+                }
+
+                return false;
+            },
+            function() {
+                return "Failed to find operation from " + tojson(testObj) +
+                    " in currentOp() output: " + tojson(testDB.currentOp());
+            });
+
+        // Allow the query to complete.
+        assert.commandWorked(
+            testDB.adminCommand({configureFailPoint: "setYieldAllLocksHang", mode: "off"}));
+
+        awaitShell();
+    }
+
+    //
+    // Confirm currentOp content for commands defined in 'testList'.
+    //
+    var testList = [
+        {
+          test: function() {
+              assert.eq(
+                  db.currentop_query.aggregate([{$match: {a: 1, $comment: "currentop_query"}}])
+                      .itcount(),
+                  1);
+          },
+          command: "aggregate",
+          planSummary: "COLLSCAN",
+          currentOpFilter: {"query.pipeline.0.$match.$comment": "currentop_query"}
+        },
+        {
+          test: function() {
+              assert.eq(db.currentop_query.find({a: 1, $comment: "currentop_query"}).count(), 1);
+          },
+          command: "count",
+          planSummary: "COLLSCAN",
+          currentOpFilter: {"query.query.$comment": "currentop_query"}
+        },
+        {
+          test: function() {
+              assert.eq(db.currentop_query.distinct("a", {a: 1, $comment: "currentop_query"}), [1]);
+          },
+          command: "distinct",
+          planSummary: "COLLSCAN",
+          currentOpFilter: {"query.query.$comment": "currentop_query"}
+        },
+        {
+          test: function() {
+              assert.eq(db.currentop_query.find({a: 1}).comment("currentop_query").itcount(), 1);
+          },
+          command: "find",
+          planSummary: "COLLSCAN",
+          currentOpFilter: {"query.comment": "currentop_query"}
+        },
+        {
+          test: function() {
+              assert.eq(db.currentop_query.findAndModify(
+                            {query: {a: 1, $comment: "currentop_query"}, update: {$inc: {b: 1}}}),
+                        {"_id": 1, "a": 1});
+          },
+          command: "findandmodify",
+          planSummary: "COLLSCAN",
+          currentOpFilter: {"query.query.$comment": "currentop_query"}
+        },
+        {
+          test: function() {
+              assert.eq(db.currentop_query.group({
+                  key: {a: 1},
+                  cond: {a: 1, $comment: "currentop_query"},
+                  reduce: function() {},
+                  initial: {}
+              }),
+                            [{"a": 1}]);
+          },
+          command: "group",
+          planSummary: "COLLSCAN",
+          currentOpFilter: {"query.group.cond.$comment": "currentop_query"}
+        },
+        {
+          test: function() {
+              assert.commandWorked(db.currentop_query.mapReduce(
+                  function() {
+                      emit(this.a, this.b);
+                  },
+                  function(a, b) {
+                      return Array.sum(b);
+                  },
+                  {query: {a: 1, $comment: "currentop_query"}, out: {inline: 1}}));
+          },
+          command: "mapreduce",
+          planSummary: "COLLSCAN",
+          currentOpFilter: {"query.query.$comment": "currentop_query"}
+        },
+        {
+          test: function() {
+              assert.writeOK(db.currentop_query.remove({a: 2, $comment: "currentop_query"}));
+          },
+          operation: "remove",
+          planSummary: "COLLSCAN",
+          currentOpFilter: {"query.$comment": "currentop_query"}
+        },
+        {
+          test: function() {
+              assert.writeOK(
+                  db.currentop_query.update({a: 1, $comment: "currentop_query"}, {$inc: {b: 1}}));
+          },
+          operation: "update",
+          planSummary: "COLLSCAN",
+          currentOpFilter: {"query.$comment": "currentop_query"}
+        }
+    ];
+
+    coll.drop();
+    var i;
+    for (i = 0; i < 5; ++i) {
+        assert.writeOK(coll.insert({_id: i, a: i}));
+    }
+
+    testList.forEach(confirmCurrentOpContents);
+
+    //
+    // Confirm currentOp content for geoNear.
+    //
+    coll.drop();
+    for (i = 0; i < 10; ++i) {
+        assert.writeOK(coll.insert({a: i, loc: {type: "Point", coordinates: [i, i]}}));
+    }
+    assert.commandWorked(coll.createIndex({loc: "2dsphere"}));
+
+    confirmCurrentOpContents({
+        test: function() {
+            assert.commandWorked(db.runCommand({
+                geoNear: "currentop_query",
+                near: {type: "Point", coordinates: [1, 1]},
+                spherical: true,
+                query: {$comment: "currentop_query"}
+            }));
+        },
+        command: "geoNear",
+        planSummary: "GEO_NEAR_2DSPHERE { loc: \"2dsphere\" }",
+        currentOpFilter: {"query.query.$comment": "currentop_query"}
+    });
+
+    //
+    // Confirm currentOp content for getMore.
+    //
+    coll.drop();
+    for (i = 0; i < 10; ++i) {
+        assert.writeOK(coll.insert({a: i}));
+    }
+
+    var cmdRes = testDB.runCommand(
+        {find: "currentop_query", filter: {$comment: "currentop_query"}, batchSize: 2});
+    assert.commandWorked(cmdRes);
+
+    // Adding cursorId to TestData to make available within the context of a parallel shell.
+    TestData.cursorId = cmdRes.cursor.id;
+
+    confirmCurrentOpContents({
+        test: function() {
+            assert.commandWorked(
+                db.runCommand({getMore: TestData.cursorId, collection: "currentop_query"}));
+        },
+        planSummary: "COLLSCAN",
+        currentOpFilter: {"query.getMore": TestData.cursorId}
+    });
+
+    delete TestData.cursorId;
+
+})();