index.js

'use strict'

const fastifyPlugin = require('fastify-plugin')
const MongoInMemory = require('mongo-in-memory')
const mongodb = require('mongodb')

function fastifyMongoMemory (fastify, opts, next) {
  const server = new MongoInMemory(opts.port || 8000)

  server.start(function (error) {
    if (error) {
      next(error)
    } else {
      const uri = server.getMongouri(opts.dbname || 'test')

      mongodb.connect(uri, function (error, db) {
        if (error) {
          next(error)
        } else {
          fastify.decorate('mongo', { db, ObjectId: mongodb.ObjectID })
          next()
        }
      })
    }
  })

  fastify.addHook('onClose', function (instance, done) {
    server.stop(function (error) {
      if (error) {
        done(error)
      } else {
        fastify.mongo.db.close(done)
      }
    })
  })
}

module.exports = fastifyPlugin(fastifyMongoMemory, '>=0.30.0')