mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-07 20:31:06 -05:00
bdc5360bc0
Use the default Mongoose connection pool for all models GitOrigin-RevId: d227b7eb36f130085c9eb1480dc07bd50ba57768
39 lines
1.2 KiB
JavaScript
39 lines
1.2 KiB
JavaScript
const SandboxedModule = require('sandboxed-module')
|
|
const mongoose = require('mongoose')
|
|
|
|
/**
|
|
* Imports a model as we would usually do with e.g. `require('models/User')`
|
|
* an returns a model an schema, but without connecting to Mongo. This allows
|
|
* us to use model classes in tests, and to use them with `sinon-mongoose`
|
|
*
|
|
* @param modelName the name of the model - e.g. 'User'
|
|
* @param requires additional `requires` to be passed to SanboxedModule in
|
|
* the event that these also need to be stubbed. For example,
|
|
* additional dependent models to be included
|
|
*
|
|
* @return model and schema pair - e.g. { User, UserSchema }
|
|
*/
|
|
|
|
module.exports = (modelName, requires = {}) => {
|
|
let model = {}
|
|
|
|
requires['../infrastructure/Mongoose'] = {
|
|
createConnection: () => {
|
|
return {
|
|
model: () => {}
|
|
}
|
|
},
|
|
model: (modelName, schema) => {
|
|
model[modelName + 'Schema'] = schema
|
|
model[modelName] = mongoose.model(modelName, schema)
|
|
},
|
|
Schema: mongoose.Schema,
|
|
Types: mongoose.Types
|
|
}
|
|
|
|
SandboxedModule.require('../../../../app/src/models/' + modelName, {
|
|
requires: requires
|
|
})
|
|
|
|
return model
|
|
}
|