overleaf/services/web/test/acceptance/src/helpers/Publisher.js
Jakob Ackermann 018a44eeb5 Merge pull request #3185 from overleaf/jpa-normalize-mongo-imports
[misc] normalize mongo imports

GitOrigin-RevId: ac653d9982e0d36736b90f4c03d4c00be88ea76a
2020-09-25 02:04:20 +00:00

32 lines
815 B
JavaScript

const { ObjectId } = require('mongodb')
const PublisherModel = require('../../../../app/src/models/Publisher').Publisher
let count = parseInt(Math.random() * 999999)
class Publisher {
constructor(options = {}) {
this.slug = options.slug || `publisher-slug-${count}`
this.managerIds = []
count += 1
}
ensureExists(callback) {
const filter = { slug: this.slug }
const options = { upsert: true, new: true, setDefaultsOnInsert: true }
PublisherModel.findOneAndUpdate(filter, {}, options, (error, publisher) => {
this._id = publisher._id
callback(error)
})
}
setManagerIds(managerIds, callback) {
return PublisherModel.findOneAndUpdate(
{ _id: ObjectId(this._id) },
{ managerIds: managerIds },
callback
)
}
}
module.exports = Publisher