Very Popular
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778import { MongoInvalidArgumentError, MongoServerError,} from "../../src/error.ts";import { CreateCollectionOptions } from "../../src/types.ts";import { testWithClient, testWithTestDBClient } from "../common.ts";import { assert, assertEquals, assertRejects, semver } from "../test.deps.ts";
interface IUser { username?: string; password?: string; _id: string; uid?: number; date?: Date;}
const dateNow = Date.now();
testWithClient("testListCollectionNames", async (client) => { const db = client.database("local"); const names = await db.listCollectionNames(); assertEquals(names, ["startup_log"]);});
testWithTestDBClient("testInsertOne", async (db) => { const users = db.collection<IUser>("mongo_test_users"); const insertId = await users.insertOne({ username: "user1", password: "pass1", date: new Date(dateNow), });
assertEquals(insertId.toString().length, 24);
const user1 = await users.findOne({ _id: insertId, });
assertEquals(user1, { _id: insertId, username: "user1", password: "pass1", date: new Date(dateNow), });});
testWithTestDBClient("testUpsertOne", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertOne({ username: "user1", password: "pass1", date: new Date(dateNow), }); const { upsertedId } = await users.updateOne( { _id: "aaaaaaaaaaaaaaaaaaaaaaaa" }, { $set: { username: "user2", password: "pass2", date: new Date(dateNow), }, }, { upsert: true }, );
assert(upsertedId);
const user2 = await users.findOne({ _id: upsertedId, });
assertEquals(user2, { _id: upsertedId, username: "user2", password: "pass2", date: new Date(dateNow), });});
testWithTestDBClient("testInsertOneTwice", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertOne({ _id: "aaaaaaaaaaaaaaaaaaaaaaaa", username: "user1", });
await assertRejects( () => users.insertOne({ _id: "aaaaaaaaaaaaaaaaaaaaaaaa", username: "user1", // deno-lint-ignore no-explicit-any }) as any, undefined, "E11000", );});
testWithTestDBClient("testFindOne", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertOne({ username: "user1", password: "pass1", date: new Date(dateNow), }); const user1 = await users.findOne(); assertEquals(Object.keys(user1!), ["_id", "username", "password", "date"]);
const query = { test: 1 }; const findNull = await users.findOne(query); assertEquals(findNull, undefined); const projectionUser = await users.findOne( {}, { projection: { _id: 0, username: 1 } }, ); assertEquals(Object.keys(projectionUser!), ["username"]); const projectionUserWithId = await users.findOne( {}, { projection: { username: 1 } }, ); assertEquals(Object.keys(projectionUserWithId!), ["_id", "username"]);});
testWithTestDBClient("testInsertMany", async (db) => { const users = db.collection<IUser>("mongo_test_users"); const { insertedCount, insertedIds } = await users.insertMany([ { username: "many", password: "pass1", }, { username: "many", password: "pass2", }, ]);
assertEquals(insertedCount, 2); assertEquals(insertedIds.length, 2);});
testWithTestDBClient("testFindAndModify-update", async (db) => { const users = db.collection<{ username: string; counter: number }>( "mongo_test_users", ); await users.insertOne({ username: "counter", counter: 5 }); const updated = await users.findAndModify({ username: "counter" }, { update: { $inc: { counter: 1 } }, new: true, });
assert(updated !== undefined); assertEquals(updated.counter, 6); assertEquals(updated.username, "counter");});
testWithTestDBClient("testFindAndModify-delete", async (db) => { const users = db.collection<{ username: string; counter: number }>( "mongo_test_users", ); await users.insertOne({ username: "delete", counter: 10 }); const updated = await users.findAndModify({ username: "delete" }, { remove: true, });
assert(updated !== undefined); assertEquals(updated.counter, 10); assertEquals(updated.username, "delete");
const tryFind = await users.findOne({ username: "delete" }); assertEquals(tryFind, undefined);});
testWithTestDBClient("test chain call for Find", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertMany([ { username: "user1", password: "pass1", }, { username: "user2", password: "pass2", }, { username: "user3", password: "pass3", }, ]); const user = await users.find().skip(1).limit(1).toArray(); assertEquals(user!.length > 0, true);});
testWithTestDBClient("testUpdateOne", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertOne({ username: "user1", password: "pass1", }); const result = await users.updateOne({}, { $set: { username: "USER1" } }); assertEquals(result, { matchedCount: 1, modifiedCount: 1, upsertedCount: 0, upsertedId: undefined, });});
testWithTestDBClient("testUpdateOne Error", async (db) => { // TODO: move tesr errors to a new file const users = db.collection<IUser>("mongo_test_users"); await users.insertOne({ username: "user1", password: "pass1", }); try { await users.updateOne({}, { username: "USER1" }); assert(false); } catch (e) { assert(e instanceof MongoInvalidArgumentError); }});
testWithTestDBClient("testUpdateOneWithUpsert", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertOne({ username: "user1", password: "pass1", }); const result = await users.updateOne( { username: "user2" }, { $set: { username: "USER2" } }, { upsert: true }, ); assertEquals(result.matchedCount, 1); assertEquals(result.modifiedCount, 0); assertEquals(result.upsertedCount, 1);});
testWithTestDBClient("testReplaceOne", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertOne({ username: "user1", password: "pass1", }); const result = await users.replaceOne({ username: "user1" }, { username: "user2", });
assertEquals(result, { matchedCount: 1, modifiedCount: 1, upsertedCount: 0, upsertedId: undefined, });});
testWithTestDBClient("testDeleteOne", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertOne({ username: "user1", password: "pass1", }); const deleteCount = await users.deleteOne({}); assertEquals(deleteCount, 1);});
testWithTestDBClient("testFindOr", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertMany([ { username: "user1", password: "pass1", }, { username: "user2", password: "pass1", }, { username: "user3", password: "pass2", }, ]); const user1 = await users .find({ $or: [{ password: "pass1" }, { password: "pass2" }], }) .toArray(); assert(user1 instanceof Array); assertEquals(user1.length, 3);});
testWithTestDBClient("testFind", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertMany([ { username: "user", password: "pass1", }, { username: "user", password: "pass2", }, { username: "user", password: "pass3", }, ]); const findUsers = await users .find({ username: "user" }, { skip: 1, limit: 1 }) .toArray(); assert(findUsers instanceof Array); assertEquals(findUsers.length, 1);
const notFound = await users.find({ test: 1 }).toArray(); assertEquals(notFound, []);});
testWithTestDBClient("test multiple queries at the same time", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertOne({ _id: "aaaaaaaaaaaaaaaaaaaaaaaa", username: "user1", password: "pass1", }); const result = await Promise.all([ users.findOne({}, { projection: { username: 1 } }), users.findOne({}, { projection: { username: 1 } }), users.findOne({}, { projection: { username: 1 } }), ]);
assertEquals(result, [ { _id: "aaaaaaaaaaaaaaaaaaaaaaaa", username: "user1" }, { _id: "aaaaaaaaaaaaaaaaaaaaaaaa", username: "user1" }, { _id: "aaaaaaaaaaaaaaaaaaaaaaaa", username: "user1" }, ]);});
testWithTestDBClient("testCount", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertMany([ { username: "user", password: "pass1", }, { username: "many", password: "pass2", }, { username: "many", password: "pass3", }, ]); const count = await users.count({ username: "many" }); assertEquals(count, 2);});
testWithTestDBClient("testCountDocuments", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertMany([ { username: "user1", password: "pass1", }, { username: "user2", password: "pass2", }, { username: "many", password: "pass3", }, { username: "many", password: "pass4", }, ]); const countAll = await users.countDocuments(); assertEquals(countAll, 4);
const count = await users.countDocuments({ username: "many" }); assertEquals(count, 2);});
testWithTestDBClient("testEstimatedDocumentCount", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertMany([ { username: "user1", password: "pass1", }, { username: "user2", password: "pass2", }, { username: "many", password: "pass3", }, { username: "many", password: "pass4", }, ]); const count = await users.estimatedDocumentCount(); assertEquals(count, 4);});
testWithTestDBClient("testAggregation", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertMany([ { username: "user1", password: "pass1", }, { username: "user2", password: "pass2", }, { username: "many", password: "pass3", }, { username: "many", password: "pass4", }, ]); const docs = await users .aggregate([ { $match: { username: "many" } }, { $group: { _id: "$username", total: { $sum: 1 } } }, ]) .toArray(); assertEquals(docs, [{ _id: "many", total: 2 }]);});
testWithTestDBClient("testUpdateMany", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertMany([ { username: "user1", password: "pass1", }, { username: "user2", password: "pass2", }, { username: "many", password: "pass3", }, { username: "many", password: "pass4", }, ]); const result = await users.updateMany( { username: "many" }, { $set: { username: "MANY" } }, ); assertEquals(result, { matchedCount: 2, modifiedCount: 2, upsertedCount: 0, upsertedIds: undefined, });});
testWithTestDBClient("testDeleteMany", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertMany([ { username: "user1", password: "pass1", }, { username: "user2", password: "pass2", }, { username: "many", password: "pass3", }, { username: "many", password: "pass4", }, ]); const deleteCount = await users.deleteMany({ username: "many" }); assertEquals(deleteCount, 2);});
testWithTestDBClient("testDistinct", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertMany([ { username: "user1", password: "pass1", }, { username: "user2", password: "pass2", }, { username: "many", password: "pass3", }, { username: "many", password: "pass4", }, ]); const user1 = await users.distinct("username"); assertEquals(user1, ["many", "user1", "user2"]);});
testWithTestDBClient("testDropConnection", async (db) => { const users = db.collection<IUser>("mongo_test_users"); await users.insertOne({ _id: "aaaaaaaaaaaaaaaaaaaaaaaa", username: "user1", password: "pass1", });
await db.collection("mongo_test_users").drop();});
testWithTestDBClient("testFindWithSort", async (db) => { const users = db.collection<IUser>("mongo_test_users");
const condition = { uid: { $exists: true } };
// prepare data for (let i = 0; i < 10; i++) { await users.insertOne({ username: "testFindWithSort", password: "pass1", uid: i, }); } const all = await users.find().toArray();
// test sorting const acceding = await users .find(condition, { sort: { uid: 1 } }) .toArray(); const descending = await users .find(condition, { sort: { uid: -1 } }) .toArray();
assertEquals( acceding, all.sort((lhs, rhs) => { return lhs.uid! - rhs.uid!; }), ); assertEquals( descending, all.sort((lhs, rhs) => { return -lhs.uid! - rhs.uid!; }), );
await db.collection("mongo_test_users").drop();});
testWithTestDBClient("testFindEmptyAsyncIteration", async (db) => { const users = db.collection<IUser>("mongo_test_users"); for (let i = 0; i < 10; i++) { await users.insertOne({ username: "testFindWithSort", password: "pass1", uid: i, }); } const cursor = users.find({ nonexistent: "foo" }); const docs = []; for await (const doc of cursor) { docs.push(doc); } assertEquals(docs, []);
await db.collection("mongo_test_users").drop();});
testWithClient("testFindWithMaxTimeMS", async (client) => { const db = client.database("local");
const supportsMaxTimeMSInFindOne = semver.gte( client.buildInfo!.version, "4.2.0", );
const users = db.collection<IUser>("mongo_test_users"); for (let i = 0; i < 10; i++) { await users.insertOne({ username: "testFindWithMaxTimeMS", password: "pass1", uid: i, }); } const users1 = await users.find({ uid: 0, }, { maxTimeMS: 100 }).toArray();
assertEquals(users1.length, 1);
const user1 = await users.findOne({ uid: 0, }, { maxTimeMS: 100 });
assertEquals(user1!.uid, 0);
try { await users.find({ uid: 0, $where: "sleep(10) || true", }, { maxTimeMS: 1 }).toArray(); assert(false); } catch (e) { assertEquals(e.ok, 0); assertEquals(e.codeName, "MaxTimeMSExpired"); assertEquals(e.errmsg, "operation exceeded time limit"); }
if (supportsMaxTimeMSInFindOne) { try { await users.findOne({ uid: 0, $where: "sleep(10) || true", }, { maxTimeMS: 1 }); assert(false); } catch (e) { assertEquals(e.ok, 0); assertEquals(e.codeName, "MaxTimeMSExpired"); assertEquals(e.errmsg, "operation exceeded time limit"); } }
await db.collection("mongo_test_users").drop();});
interface IStudents { _id: string; name: string; year: number; major: string; gpa?: number; address: { city: string; street: string; };}
testWithClient( "createCollection should create a collection without options", async (client) => { const db = client.database("test");
const testCollectionName = "test_collection_for_createCollection_0";
const createdCollection = await db .createCollection<{ _id: string; name: string }>( testCollectionName, );
assert(createdCollection);
await db.collection(testCollectionName).drop(); },);
testWithClient( "createCollection should create a collection with options", async (client) => { // Note that database is 'test' const db = client.database("test");
const testCollectionName = "test_collection_for_createCollection_1";
// Example from https://www.mongodb.com/docs/manual/reference/operator/query/jsonSchema/#mongodb-query-op.-jsonSchema const options: CreateCollectionOptions = { validator: { $jsonSchema: { bsonType: "object", required: ["name", "year", "major", "address"], properties: { name: { bsonType: "string", description: "must be a string and is required", }, year: { bsonType: "int", minimum: 2017, maximum: 3017, description: "must be an integer in [ 2017, 3017 ] and is required", }, major: { enum: ["Math", "English", "Computer Science", "History", null], description: "can only be one of the enum values and is required", }, gpa: { bsonType: ["double"], description: "must be a double if the field exists", }, address: { bsonType: "object", required: ["city"], properties: { street: { bsonType: "string", description: "must be a string if the field exists", }, city: { bsonType: "string", "description": "must be a string and is required", }, }, }, }, }, }, };
const createdCollection = await db .createCollection<IStudents>( testCollectionName, options, );
assert(createdCollection);
// sanity test to check whether the speicified validator from options works // error with message: "Document failed validation" await assertRejects( () => createdCollection.insertOne({ name: "Alice", year: 2019, major: "History", gpa: 3, address: { city: "NYC", street: "33rd Street", }, }), );
// TODO: refactor to clean up the test collection properly. // It should clean up the collection when above insertion succeeds in any case, which is unwanted result. // Refactor when test utility is more provided. await db.collection(testCollectionName).drop(); },);
testWithClient( "createCollection should throw an error with invalid options", async (client) => { const db = client.database("test");
const testCollectionName = "test_collection_for_createCollection_2"; const invalidOptions: CreateCollectionOptions = { capped: true, };
await assertRejects( () => db.createCollection<{ _id: string; name: string }>( testCollectionName, invalidOptions, ), // error with the message "the 'size' field is required when 'capped' is true" MongoServerError, ); },);