mongo/jstests/core/update_metrics.js

79 lines
3.8 KiB
JavaScript

/**
* Tests "metrics.commands.update.pipeline" and "metrics.commands.update.arrayFilters" counters of
* the update command.
*
* @tags: [
* requires_fcv_44,
* # The test relies on the precise number of executions of commands.
* requires_non_retryable_writes,
* # The test is designed to work with an unsharded collection.
* assumes_unsharded_collection,
* # The coll.update command does not work with $set operator in compatibility write mode.
* requires_find_command,
* ]
*/
(function() {
"use strict";
const testDB = db.getSiblingDB(jsTestName());
assert.commandWorked(testDB.dropDatabase());
const coll = testDB.update_metrics;
assert.commandWorked(testDB.createCollection(coll.getName()));
assert.commandWorked(coll.insert([{key: 1, value: 1, array: [5, 10]}]));
// "Initialize" the counters for the update command.
assert.commandWorked(coll.update({key: 1}, {$set: {value: 0}}));
let serverStatusBeforeTest = testDB.serverStatus();
// Verify that the metrics.commands.update.pipeline counter is present.
assert.gte(
serverStatusBeforeTest.metrics.commands.update.pipeline, 0, tojson(serverStatusBeforeTest));
// Verify that that update command without aggregation pipeline-style update does not increment the
// counter.
assert.commandWorked(coll.update({key: 1}, {$set: {value: 5}}));
let serverStatusAfterTest = testDB.serverStatus();
assert.eq(serverStatusBeforeTest.metrics.commands.update.pipeline,
serverStatusAfterTest.metrics.commands.update.pipeline,
`Before: ${tojson(serverStatusBeforeTest)}, after: ${tojson(serverStatusAfterTest)}`);
// Verify that that update command with aggregation pipeline-style update increments the counter.
assert.commandWorked(coll.update({key: 1}, [{$set: {value: 10}}]));
serverStatusAfterTest = testDB.serverStatus();
assert.eq(serverStatusBeforeTest.metrics.commands.update.pipeline + 1,
serverStatusAfterTest.metrics.commands.update.pipeline,
`Before: ${tojson(serverStatusBeforeTest)}, after: ${tojson(serverStatusAfterTest)}`);
serverStatusBeforeTest = testDB.serverStatus();
// Verify that the metrics.commands.update.arrayFilters counter is present.
assert.gte(
serverStatusBeforeTest.metrics.commands.update.arrayFilters, 0, tojson(serverStatusBeforeTest));
// Verify that that update command without arrayFilters does not increment the counter.
assert.commandWorked(coll.update({key: 1}, {$set: {value: 5}}));
serverStatusAfterTest = testDB.serverStatus();
assert.eq(serverStatusBeforeTest.metrics.commands.update.arrayFilters,
serverStatusAfterTest.metrics.commands.update.arrayFilters,
`Before: ${tojson(serverStatusBeforeTest)}, after: ${tojson(serverStatusAfterTest)}`);
// Verify that that update command with arrayFilters increments the counter.
assert.commandWorked(coll.update(
{key: 1}, {$set: {"array.$[element]": 20}}, {arrayFilters: [{"element": {$gt: 6}}]}));
serverStatusAfterTest = testDB.serverStatus();
assert.eq(serverStatusBeforeTest.metrics.commands.update.arrayFilters + 1,
serverStatusAfterTest.metrics.commands.update.arrayFilters,
`Before: ${tojson(serverStatusBeforeTest)}, after: ${tojson(serverStatusAfterTest)}`);
// Verify that that a multi-document update command with arrayFilters increments the counter.
assert.commandWorked(
coll.insert([{key: 2, value: 1, array: [7, 0]}, {key: 3, value: 1, array: [7, 0]}]));
assert.commandWorked(coll.update(
{}, {$set: {"array.$[element]": 20}}, {multi: true, arrayFilters: [{"element": {$gt: 6}}]}));
serverStatusAfterTest = testDB.serverStatus();
assert.eq(serverStatusBeforeTest.metrics.commands.update.arrayFilters + 2,
serverStatusAfterTest.metrics.commands.update.arrayFilters,
`Before: ${tojson(serverStatusBeforeTest)}, after: ${tojson(serverStatusAfterTest)}`);
})();