123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263
import { cargoBuild } from "./build.ts";import {  assert,  assertEquals,  assertThrowsAsync,  exists,} from "./test.deps.ts";import { MongoClient } from "./ts/client.ts";import { ObjectId } from "./ts/types.ts";import { init } from "./ts/util.ts";interface IUser {  username: string;  password: string;  _id: { $oid: string };  date?: Date;}const { test } = Deno;const dateNow = Date.now();
let testClient: MongoClient | undefined;
function getClient(): MongoClient {  if (testClient) return testClient;  const client = new MongoClient();  client.connectWithUri("mongodb://localhost:27017");  testClient = client;  return client;}
test("testConnectWithUri", async () => {  const client = new MongoClient();  client.connectWithUri("mongodb://localhost:27017");  const names = await client.listDatabases();  assert(names instanceof Array);  assert(names.length > 0);});
test("testConnectWithOptions", async () => {  const client = new MongoClient();  client.connectWithOptions({    hosts: ["localhost:27017"],  });  const names = await client.listDatabases();  assert(names instanceof Array);  assert(names.length > 0);});
test("testListCollectionNames", async () => {  const db = getClient().database("local");  const names = await db.listCollectionNames();  assertEquals(names, ["startup_log"]);});
test("testInsertOne", async () => {  const db = getClient().database("test");  const users = db.collection<IUser>("mongo_test_users");  const insertId: ObjectId = await users.insertOne({    username: "user1",    password: "pass1",    date: new Date(dateNow),  });
  assertEquals(Object.keys(insertId), ["$oid"]);
  const user1 = await users.findOne({    _id: ObjectId(insertId.$oid),  });
  assertEquals(user1, {    _id: insertId,    username: "user1",    password: "pass1",    date: new Date(dateNow),  });});
test("testUpsertOne", async () => {  const db = getClient().database("test");  const users = db.collection<IUser>("mongo_test_users");  const { upsertedId } = await users.updateOne(    {      _id: ObjectId("aaaaaaaaaaaaaaaaaaaaaaaa"),    },    {      username: "user1",      password: "pass1",      date: new Date(dateNow),    },    { upsert: true },  );
  assert(upsertedId);  assertEquals(Object.keys(upsertedId), ["$oid"]);
  const user1 = await users.findOne({    _id: ObjectId(upsertedId.$oid),  });
  assertEquals(user1, {    _id: upsertedId,    username: "user1",    password: "pass1",    date: new Date(dateNow),  });});
test("testInsertOneTwice", async () => {  const db = getClient().database("test");  const users = db.collection<IUser>("mongo_test_users_2");  const insertId: ObjectId = await users.insertOne({    _id: ObjectId("aaaaaaaaaaaaaaaaaaaaaaaa"),    username: "user1",  });
  await assertThrowsAsync(    () =>      users.insertOne({        _id: ObjectId("aaaaaaaaaaaaaaaaaaaaaaaa"),        username: "user1",      }) as any,    undefined,    "E11000",  );});
test("testFindOne", async () => {  const db = getClient().database("test");  const users = db.collection<IUser>("mongo_test_users");  const user1 = await users.findOne();  assert(user1 instanceof Object);  assertEquals(Object.keys(user1), ["_id", "username", "password", "date"]);
  const query = { test: 1 } as any;  const findNull = await users.findOne(query);  assertEquals(findNull, null);});
test("testUpdateOne", async () => {  const db = getClient().database("test");  const users = db.collection("mongo_test_users");  const result = await users.updateOne({}, { username: "USER1" });  assertEquals(result, { matchedCount: 1, modifiedCount: 1, upsertedId: null });});
test("testDeleteOne", async () => {  const db = getClient().database("test");  const users = db.collection("mongo_test_users");  const deleteCount = await users.deleteOne({});  assertEquals(deleteCount, 1);});
test("testInsertMany", async () => {  const db = getClient().database("test");  const users = db.collection("mongo_test_users");  const insertIds = await users.insertMany([    {      username: "many",      password: "pass1",    },    {      username: "many",      password: "pass2",    },  ]);
  assertEquals(insertIds.length, 2);});
test("testFindOr", async () => {  const db = getClient().database("test");  const users = db.collection<IUser>("mongo_test_users");  const user1 = await users.find({    $or: [      {        password: "pass1",      },      {        password: "pass2",      },    ],  });
  assert(user1 instanceof Array);  assertEquals(user1.length, 3);});
test("testFind", async () => {  const db = getClient().database("test");  const users = db.collection("mongo_test_users");  const findUsers = await users.find(    { username: "many" },    { skip: 1, limit: 1 },  );  assert(findUsers instanceof Array);  assertEquals(findUsers.length, 1);
  const notFound = await users.find({ test: 1 });  assertEquals(notFound, []);});
test("testCount", async () => {  const db = getClient().database("test");  const users = db.collection("mongo_test_users");  const count = await users.count({ username: "many" });  assertEquals(count, 2);});
test("testAggregation", async () => {  const db = getClient().database("test");  const users = db.collection("mongo_test_users");  const docs = await users.aggregate([    { $match: { username: "many" } },    { $group: { _id: "$username", total: { $sum: 1 } } },  ]);  assertEquals(docs, [{ _id: "many", total: 2 }]);});
test("testUpdateMany", async () => {  const db = getClient().database("test");  const users = db.collection("mongo_test_users");  const result = await users.updateMany(    { username: "many" },    { $set: { username: "MANY" } },  );  assertEquals(result, { matchedCount: 2, modifiedCount: 2, upsertedId: null });});
test("testDeleteMany", async () => {  const db = getClient().database("test");  const users = db.collection("mongo_test_users");  const deleteCount = await users.deleteMany({ username: "MANY" });  assertEquals(deleteCount, 2);});
test("testDistinct", async () => {  const db = getClient().database("test");  const users = db.collection<IUser>("mongo_test_users");  const user1 = await users.distinct("username", {});  assertEquals(user1, ["user1"]);});
// TODO mongdb_rust official library has not implemented this feature// test("testCreateIndexes", async () => {//   const db = getClient().database("test");//   const collection = db.collection("mongo_indexes");//   const result = await collection.createIndexes([//     { keys: { created_at: 1 }, options: { expireAfterSeconds: 10000 } }//   ]);//   console.log(result);// });
test("testClose", async () => {  const result = getClient().close();  assertEquals(result, { success: true });  testClient = undefined;});
if (await exists(".deno_plugins")) {  await Deno.remove(".deno_plugins", { recursive: true });}await cargoBuild();await init("file://./target/release");