Files
mongo/jstests/core/query/api/apitest_dbcollection.js
Silvia Surroca 3d3328e613 SERVER-83211 Enable counting index tests on sharded passthrough suites (#43649)
GitOrigin-RevId: dcb676f1697884eb629906f0d8d8979f10ef4035
2025-11-12 14:45:27 +00:00

336 lines
11 KiB
JavaScript

// stats will return ok:1 for non-existant colls
import {IndexUtils} from "jstests/libs/index_utils.js";
/**
* Tests for the db collection
*
* @tags: [
* requires_collstats,
* requires_fastcount,
* # In-memory data structures are not causally consistent.
* does_not_support_causal_consistency,
* ]
*/
const testDb = db.getSiblingDB("test");
const collName = jsTestName();
const coll = testDb.getCollection(collName);
/*
* test drop
*/
coll.drop();
assert.eq(0, coll.find().length(), "1");
coll.save({a: 1});
assert.eq(1, coll.find().length(), "2");
coll.drop();
assert.eq(0, coll.find().length(), "3");
/*
* test count
*/
assert.eq(0, coll.count(), "4");
coll.save({a: 1});
assert.eq(1, coll.count(), "5");
for (var i = 0; i < 100; i++) {
coll.save({a: 1});
}
assert.eq(101, coll.count(), "6");
coll.drop();
assert.eq(0, coll.count(), "7");
/*
* test validate
*/
coll.drop();
assert.eq(0, coll.count(), "8");
for (var i = 0; i < 100; i++) {
coll.save({a: 1});
}
(function () {
const validateResult = assert.commandWorked(coll.validate());
// Extract validation results from mongos output if running in a sharded context.
const isShardedNS = validateResult.hasOwnProperty("raw");
if (isShardedNS) {
// Sample mongos format:
// {
// raw: {
// "localhost:30000": {
// "ns" : coll.getName(),
// ...
// "valid": true,
// ...
// "ok": 1
// }
// },
// "valid": true,
// ...
// "ok": 1
// }
let nrecords = 0;
for (let field in validateResult.raw) {
const rawValidateResult = validateResult.raw[field];
assert.neq(null, rawValidateResult);
assert.eq(
"test." + collName,
rawValidateResult.ns,
"incorrect namespace in testDb.collection.validate() result: " + tojson(rawValidateResult),
);
assert(rawValidateResult.valid, "collection validation failed");
nrecords += rawValidateResult.nrecords;
}
assert.eq(100, nrecords, "11");
} else {
assert.eq(
"test." + collName,
validateResult.ns,
"incorrect namespace in testDb.collection.validate() result: " + tojson(validateResult),
);
assert.eq(100, validateResult.nrecords, "11");
}
assert(validateResult.valid, "collection validation failed");
})();
/*
* test deleteIndex, deleteIndexes
*/
coll.drop();
IndexUtils.assertIndexes(coll, [], "12");
coll.dropIndexes();
IndexUtils.assertIndexes(coll, [], "13");
coll.save({a: 10});
IndexUtils.assertIndexes(coll, [{_id: 1}], "14");
coll.createIndex({a: 1});
coll.save({a: 10});
print(tojson(coll.getIndexes()));
IndexUtils.assertIndexes(coll, [{_id: 1}, {a: 1}], "15");
coll.dropIndex({a: 1});
IndexUtils.assertIndexes(coll, [{_id: 1}], "16");
coll.save({a: 10});
coll.createIndex({a: 1});
coll.save({a: 10});
IndexUtils.assertIndexes(coll, [{_id: 1}, {a: 1}], "17");
coll.dropIndex("a_1");
IndexUtils.assertIndexes(coll, [{_id: 1}], "18");
coll.save({a: 10, b: 11});
coll.createIndex({a: 1});
coll.createIndex({b: 1});
coll.save({a: 10, b: 12});
IndexUtils.assertIndexes(coll, [{_id: 1}, {a: 1}, {b: 1}], "19");
coll.dropIndex({b: 1});
IndexUtils.assertIndexes(coll, [{_id: 1}, {a: 1}], "20");
coll.dropIndex({a: 1});
IndexUtils.assertIndexes(coll, [{_id: 1}], "21");
coll.save({a: 10, b: 11});
coll.createIndex({a: 1});
coll.createIndex({b: 1});
coll.save({a: 10, b: 12});
IndexUtils.assertIndexes(coll, [{_id: 1}, {a: 1}, {b: 1}], "22");
coll.dropIndexes();
IndexUtils.assertIndexes(coll, [{_id: 1}], "23");
coll.find();
coll.drop();
IndexUtils.assertIndexes(coll, [], "24");
/*
* stats()
*/
(function () {
if (typeof globalThis.ImplicitlyShardAccessCollSettings !== "undefined") {
print("Skipping this test because collections are implicitly sharded and sharded collections can't be capped.");
return;
}
let t = testDb.apttest_dbcollection;
// Non-existent collection.
t.drop();
let noCollStats = assert.commandWorked(
t.stats(),
"testDb.collection.stats() should work on non-existent collection",
);
assert.eq(0, noCollStats.size, "All properties should be 0 on nonexistant collections");
assert.eq(0, noCollStats.count, "All properties should be 0 on nonexistant collections");
assert.eq(0, noCollStats.storageSize, "All properties should be 0 on nonexistant collections");
assert.eq(0, noCollStats.nindexes, "All properties should be 0 on nonexistant collections");
assert.eq(0, noCollStats.totalIndexSize, "All properties should be 0 on nonexistant collections");
assert.eq(0, noCollStats.totalSize, "All properties should be 0 on nonexistant collections");
// scale - passed to stats() as sole numerical argument or part of an options object.
t.drop();
assert.commandWorked(testDb.createCollection(t.getName(), {capped: true, size: 10 * 1024 * 1024}));
var collectionStats = assert.commandWorked(t.stats(1024 * 1024));
assert.eq(
10,
collectionStats.maxSize,
"testDb.collection.stats(scale) - capped collection size scaled incorrectly: " + tojson(collectionStats),
);
var collectionStats = assert.commandWorked(t.stats({scale: 1024 * 1024}));
assert.eq(
10,
collectionStats.maxSize,
"testDb.collection.stats({scale: N}) - capped collection size scaled incorrectly: " + tojson(collectionStats),
);
// Ensure that collStats can handle large values for 'scale'.
var collectionStats = assert.commandWorked(t.stats(Number.MAX_VALUE));
assert.eq(
0,
collectionStats.maxSize,
"testDb.collection.stats(Number.MAX_VALUE) - capped collection size scaled incorrectly: " +
tojson(collectionStats),
);
var collectionStats = assert.commandWorked(t.stats({scale: Number.MAX_VALUE}));
assert.eq(
0,
collectionStats.maxSize,
"testDb.collection.stats({scale: Number.MAX_VALUE}) - capped collection size scaled incorrectly: " +
tojson(collectionStats),
);
// indexDetails - If true, includes 'indexDetails' field in results. Default: false.
t.drop();
t.save({a: 1});
t.createIndex({a: 1});
collectionStats = assert.commandWorked(t.stats());
assert(
!collectionStats.hasOwnProperty("indexDetails"),
"unexpected indexDetails found in testDb.collection.stats() result: " + tojson(collectionStats),
);
collectionStats = assert.commandWorked(t.stats({indexDetails: false}));
assert(
!collectionStats.hasOwnProperty("indexDetails"),
"unexpected indexDetails found in testDb.collection.stats({indexDetails: true}) result: " +
tojson(collectionStats),
);
collectionStats = assert.commandWorked(t.stats({indexDetails: true}));
assert(
collectionStats.hasOwnProperty("indexDetails"),
"indexDetails missing from testDb.collection.stats({indexDetails: true}) result: " + tojson(collectionStats),
);
// Returns index name.
function getIndexName(indexKey) {
let indexes = t.getIndexes().filter(function (doc) {
return friendlyEqual(doc.key, indexKey);
});
assert.eq(1, indexes.length, tojson(indexKey) + " not found in getIndexes() result: " + tojson(t.getIndexes()));
return indexes[0].name;
}
function checkIndexDetails(options, indexName) {
let collectionStats = assert.commandWorked(t.stats(options));
assert(
collectionStats.hasOwnProperty("indexDetails"),
"indexDetails missing from " +
"testDb.collection.stats(" +
tojson(options) +
") result: " +
tojson(collectionStats),
);
// Currently, indexDetails is only supported with WiredTiger.
let storageEngine = jsTest.options().storageEngine;
if (storageEngine && storageEngine !== "wiredTiger") {
return;
}
assert.eq(1, Object.keys(collectionStats.indexDetails).length, "indexDetails must have exactly one entry");
assert(
collectionStats.indexDetails[indexName],
indexName + " missing from indexDetails: " + tojson(collectionStats.indexDetails),
);
assert.neq(
0,
Object.keys(collectionStats.indexDetails[indexName]).length,
indexName + " exists in indexDetails but contains no information: " + tojson(collectionStats),
);
}
// indexDetailsKey - show indexDetails results for this index key only.
let indexKey = {a: 1};
let indexName = getIndexName(indexKey);
checkIndexDetails({indexDetails: true, indexDetailsKey: indexKey}, indexName);
// indexDetailsName - show indexDetails results for this index name only.
checkIndexDetails({indexDetails: true, indexDetailsName: indexName}, indexName);
// Cannot specify both indexDetailsKey and indexDetailsName.
let error = assert.throws(
function () {
t.stats({indexDetails: true, indexDetailsKey: indexKey, indexDetailsName: indexName});
},
[],
"indexDetailsKey and indexDetailsName cannot be used at the same time",
);
assert.eq(
Error,
error.constructor,
"testDb.collection.stats() failed when both indexDetailsKey and indexDetailsName " +
"are used but with incorrect error type",
);
t.drop();
})();
/*
* test testDb.collection.totalSize()
*/
(function () {
let t = testDb.apitest_dbcollection;
if (typeof globalThis.ImplicitlyShardAccessCollSettings !== "undefined") {
print("Skipping this test for sharded collections.");
return;
}
t.drop();
let emptyStats = assert.commandWorked(t.stats());
assert.eq(emptyStats.storageSize, 0);
assert.eq(emptyStats.totalIndexSize, 0);
assert.eq(0, t.storageSize(), "testDb.collection.storageSize() on empty collection should return 0");
assert.eq(0, t.totalIndexSize(), "testDb.collection.totalIndexSize() on empty collection should return 0");
assert.eq(0, t.totalSize(), "testDb.collection.totalSize() on empty collection should return 0");
t.save({a: 1});
let stats = assert.commandWorked(t.stats());
assert.neq(
undefined,
t.storageSize(),
"testDb.collection.storageSize() cannot be undefined on a non-empty collection",
);
assert.neq(
undefined,
t.totalIndexSize(),
"testDb.collection.totalIndexSize() cannot be undefined on a non-empty collection",
);
t.drop();
})();