vitest-mms

vitest-mms

mongodb-memory-server integration for vitest

• mongodb driver support.
• mongoose support.
• clear database between each test

...other ORMs: Pull Requests are welcome!

You can also connect to the mongodb memory server directly by using a connection uri const connectionUri = inject("MONGO_URI");

Installation

npm install -D vitest-mms mongodb-memory-server
yarn add -D vitest-mms mongodb-memory-server
pnpm add -D vitest-mms mongodb-memory-server

Usage with mongodb

NOTE: You need to install mongodb separately.

Setup:

To make it available in the global context for every test you need to add a globalSetup and setupFile in your vitest config:

// vitest.config.mjs
import { defineConfig } from "vitest/config";

export default defineConfig({
  test: {
    globalSetup: ["vitest-mms/globalSetup"],
    setupFiles: ["vitest-mms/mongodb/setupFile"],
  },
});

// index.test.js
import { test } from "vitest";

test("my test", async ({ db, mongoClient }) => {
  const users = db.collection("users");
  users.insertOne({ name: "John" });
  expect(await users.countDocuments()).toBe(1);
});

• mongoClient is the connected MongoClient instance (see import("mongodb").MongoClient)
• db is a random database name connected to the mongodb-memory-server instance (see import("mongodb").Db)

For typescript support add the following to your tsconfig.json

// tsconfig.json
{
  "compilerOptions": {
    "types": ["vitest-mms/mongodb/setupFile"]
  }
}

Usage with mongoose

NOTE: You need to install mongoose separately.

// vitest.config.mjs
import { defineConfig } from "vitest/config";

export default defineConfig({
  test: {
    globalSetup: ["vitest-mms/globalSetup"],
    setupFile: ["vitest-mms/mongoose/setupFile"],
  },
});

// tsconfig.json
{
  "compilerOptions": {
    "types": ["vitest-mms/mongoose/setupFile"]
  }
}

// index.test.js
test("my test", async ({ mongoose }) => {
  mongoose.connection.db; // use db

  const User = mongoose.model("User", new mongoose.Schema({ name: String }));
  await User.create({ name: "John" });
  expect(await User.countDocuments()).toBe(1);
});

• mongoose is the mongoose instance returned by mongoose.connect

Alternative using extended test context

vitest.config.mjs:

import { defineConfig } from "vitest/config";

export default defineConfig({
  test: {
    globalSetup: ["vitest-mms/globalSetup"],
  },
});

index.test.js:

// using the extended test context
import { mssTest as test } from "vitest-mms/mongodb/test";

test("my test", async ({ db, mongoClient }) => {
  const users = db.collection("users");
  users.insertOne({ name: "John" });
  expect(await users.countDocuments()).toBe(1);
});

See https://vitest.dev/guide/test-context.html#extend-test-context for more information

Usage with unplugin-auto-import

vitest.config.mjs:

import { defineConfig } from "vitest/config";
import AutoImport from "unplugin-auto-import/vite";

export default defineConfig({
  plugins: [
    AutoImport({
      imports: [{ "vitest-mms/mongodb/test": [["mmsTest", "test"]] }],
    }),
  ],
  test: {
    globalSetup: ["vitest-mms/globalSetup"],
  },
});

Then use it directly without having to import mmsTest:

// index.test.js
test("my test", async ({ db, mongoClient }) => {
  const users = db.collection("users");
  users.insertOne({ name: "John" });
  expect(await users.countDocuments()).toBe(1);
});