overleaf/services/notifications/test/unit/js/NotificationsTests.js
Christopher Hoskin e86ef1139e Merge pull request #15350 from overleaf/csh-issue-11625-mongo-ug-5-notifications
Use mongodb-legacy in notifications for callback support

GitOrigin-RevId: 90f05f747e221c1bd61564e2e28e505d0b125467
2023-10-24 08:03:15 +00:00

303 lines
9.2 KiB
JavaScript

/* eslint-disable
no-dupe-keys,
no-return-assign,
no-unused-vars,
*/
// TODO: This file was created by bulk-decaffeinate.
// Fix any style issues and re-enable lint.
/*
* decaffeinate suggestions:
* DS102: Remove unnecessary code created because of implicit returns
* Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
*/
const sinon = require('sinon')
const { expect } = require('chai')
const modulePath = '../../../app/js/Notifications.js'
const SandboxedModule = require('sandboxed-module')
const assert = require('assert')
const { ObjectId } = require('mongodb')
const userId = '51dc93e6fb625a261300003b'
const notificationId = '574ee8d6f40c3a244e704249'
const notificationKey = 'notification-key'
describe('Notifications Tests', function () {
beforeEach(function () {
this.findToArrayStub = sinon.stub()
this.findStub = sinon.stub().returns({ toArray: this.findToArrayStub })
this.countStub = sinon.stub()
this.updateOneStub = sinon.stub()
this.deleteOneStub = sinon.stub()
this.db = {
notifications: {
find: this.findStub,
count: this.countStub,
updateOne: this.updateOneStub,
deleteOne: this.deleteOneStub,
},
}
this.notifications = SandboxedModule.require(modulePath, {
requires: {
'@overleaf/settings': {},
'./mongodb': { db: this.db, ObjectId },
'@overleaf/metrics': { timeAsyncMethod: sinon.stub() },
},
})
this.stubbedNotification = {
user_id: new ObjectId(userId),
key: 'notification-key',
messageOpts: 'some info',
templateKey: 'template-key',
}
return (this.stubbedNotificationArray = [this.stubbedNotification])
})
describe('getUserNotifications', function () {
return it('should find all notifications and return i', function (done) {
this.findToArrayStub.callsArgWith(0, null, this.stubbedNotificationArray)
return this.notifications.getUserNotifications(
userId,
(err, notifications) => {
if (err) return done(err)
notifications.should.equal(this.stubbedNotificationArray)
assert.deepEqual(this.findStub.args[0][0], {
user_id: new ObjectId(userId),
templateKey: { $exists: true },
})
return done()
}
)
})
})
describe('addNotification', function () {
beforeEach(function () {
this.stubbedNotification = {
user_id: new ObjectId(userId),
key: 'notification-key',
messageOpts: 'some info',
templateKey: 'template-key',
}
this.expectedDocument = {
user_id: this.stubbedNotification.user_id,
key: 'notification-key',
messageOpts: 'some info',
templateKey: 'template-key',
}
this.expectedQuery = {
user_id: this.stubbedNotification.user_id,
key: 'notification-key',
}
this.updateOneStub.yields()
return this.countStub.yields(null, 0)
})
it('should insert the notification into the collection', function (done) {
return this.notifications.addNotification(
userId,
this.stubbedNotification,
err => {
expect(err).not.to.exist
sinon.assert.calledWith(
this.updateOneStub,
this.expectedQuery,
{ $set: this.expectedDocument },
{ upsert: true }
)
return done()
}
)
})
describe('when there is an existing notification', function (done) {
beforeEach(function () {
return this.countStub.yields(null, 1)
})
it('should fail to insert', function (done) {
return this.notifications.addNotification(
userId,
this.stubbedNotification,
err => {
expect(err).not.to.exist
sinon.assert.notCalled(this.updateOneStub)
return done()
}
)
})
return it('should update the key if forceCreate is true', function (done) {
this.stubbedNotification.forceCreate = true
return this.notifications.addNotification(
userId,
this.stubbedNotification,
err => {
expect(err).not.to.exist
sinon.assert.calledWith(
this.updateOneStub,
this.expectedQuery,
{ $set: this.expectedDocument },
{ upsert: true }
)
return done()
}
)
})
})
describe('when the notification is set to expire', function () {
beforeEach(function () {
this.stubbedNotification = {
user_id: new ObjectId(userId),
key: 'notification-key',
messageOpts: 'some info',
templateKey: 'template-key',
expires: '2922-02-13T09:32:56.289Z',
}
this.expectedDocument = {
user_id: this.stubbedNotification.user_id,
key: 'notification-key',
messageOpts: 'some info',
templateKey: 'template-key',
expires: new Date(this.stubbedNotification.expires),
}
return (this.expectedQuery = {
user_id: this.stubbedNotification.user_id,
key: 'notification-key',
})
})
return it('should add an `expires` Date field to the document', function (done) {
return this.notifications.addNotification(
userId,
this.stubbedNotification,
err => {
expect(err).not.to.exist
sinon.assert.calledWith(
this.updateOneStub,
this.expectedQuery,
{ $set: this.expectedDocument },
{ upsert: true }
)
return done()
}
)
})
})
return describe('when the notification has a nonsensical expires field', function () {
beforeEach(function () {
this.stubbedNotification = {
user_id: new ObjectId(userId),
key: 'notification-key',
messageOpts: 'some info',
templateKey: 'template-key',
expires: 'WAT',
}
return (this.expectedDocument = {
user_id: this.stubbedNotification.user_id,
key: 'notification-key',
messageOpts: 'some info',
templateKey: 'template-key',
expires: new Date(this.stubbedNotification.expires),
})
})
return it('should produce an error', function (done) {
return this.notifications.addNotification(
userId,
this.stubbedNotification,
err => {
;(err instanceof Error).should.equal(true)
sinon.assert.notCalled(this.updateOneStub)
return done()
}
)
})
})
})
describe('removeNotificationId', function () {
return it('should mark the notification id as read', function (done) {
this.updateOneStub.callsArgWith(2, null)
return this.notifications.removeNotificationId(
userId,
notificationId,
err => {
if (err) return done(err)
const searchOps = {
user_id: new ObjectId(userId),
_id: new ObjectId(notificationId),
}
const updateOperation = {
$unset: { templateKey: true, messageOpts: true },
}
assert.deepEqual(this.updateOneStub.args[0][0], searchOps)
assert.deepEqual(this.updateOneStub.args[0][1], updateOperation)
return done()
}
)
})
})
describe('removeNotificationKey', function () {
return it('should mark the notification key as read', function (done) {
this.updateOneStub.callsArgWith(2, null)
return this.notifications.removeNotificationKey(
userId,
notificationKey,
err => {
if (err) return done(err)
const searchOps = {
user_id: new ObjectId(userId),
key: notificationKey,
}
const updateOperation = {
$unset: { templateKey: true },
}
assert.deepEqual(this.updateOneStub.args[0][0], searchOps)
assert.deepEqual(this.updateOneStub.args[0][1], updateOperation)
return done()
}
)
})
})
describe('removeNotificationByKeyOnly', function () {
return it('should mark the notification key as read', function (done) {
this.updateOneStub.callsArgWith(2, null)
return this.notifications.removeNotificationByKeyOnly(
notificationKey,
err => {
if (err) return done(err)
const searchOps = { key: notificationKey }
const updateOperation = { $unset: { templateKey: true } }
assert.deepEqual(this.updateOneStub.args[0][0], searchOps)
assert.deepEqual(this.updateOneStub.args[0][1], updateOperation)
return done()
}
)
})
})
return describe('deleteNotificationByKeyOnly', function () {
return it('should completely remove the notification', function (done) {
this.deleteOneStub.callsArgWith(1, null)
return this.notifications.deleteNotificationByKeyOnly(
notificationKey,
err => {
if (err) return done(err)
const searchOps = { key: notificationKey }
assert.deepEqual(this.deleteOneStub.args[0][0], searchOps)
return done()
}
)
})
})
})