overleaf/services/spelling/test/unit/js/LearnedWordsManagerTests.js
2020-02-18 10:58:03 -05:00

165 lines
4.8 KiB
JavaScript

const sinon = require('sinon')
const chai = require('chai')
const { expect } = chai
const SandboxedModule = require('sandboxed-module')
const modulePath = require('path').join(
__dirname,
'../../../app/js/LearnedWordsManager'
)
const { assert } = require('chai')
describe('LearnedWordsManager', function() {
beforeEach(function() {
this.token = 'a6b3cd919ge'
this.callback = sinon.stub()
this.db = {
spellingPreferences: {
update: sinon.stub().yields()
}
}
this.cache = {
get: sinon.stub(),
set: sinon.stub(),
del: sinon.stub()
}
this.LearnedWordsManager = SandboxedModule.require(modulePath, {
globals: {
console: console
},
requires: {
'./DB': this.db,
'./MongoCache': this.cache,
'logger-sharelatex': {
log() {},
err() {},
info() {}
},
'metrics-sharelatex': {
timeAsyncMethod: sinon.stub(),
inc: sinon.stub()
}
}
})
})
describe('learnWord', function() {
beforeEach(function() {
this.word = 'instanton'
this.LearnedWordsManager.learnWord(this.token, this.word, this.callback)
})
it('should insert the word in the word list in the database', function() {
expect(
this.db.spellingPreferences.update.calledWith(
{
token: this.token
},
{
$addToSet: { learnedWords: this.word }
},
{
upsert: true
}
)
).to.equal(true)
})
it('should call the callback', function() {
expect(this.callback.called).to.equal(true)
})
})
describe('unlearnWord', function() {
beforeEach(function() {
this.word = 'instanton'
this.LearnedWordsManager.unlearnWord(this.token, this.word, this.callback)
})
it('should remove the word from the word list in the database', function() {
expect(
this.db.spellingPreferences.update.calledWith(
{
token: this.token
},
{
$pull: { learnedWords: this.word }
}
)
).to.equal(true)
})
it('should call the callback', function() {
expect(this.callback.called).to.equal(true)
})
})
describe('getLearnedWords', function() {
beforeEach(function() {
this.wordList = ['apples', 'bananas', 'pears']
this.wordListWithDuplicates = this.wordList.slice()
this.wordListWithDuplicates.push('bananas')
this.db.spellingPreferences.findOne = (conditions, callback) => {
callback(null, { learnedWords: this.wordListWithDuplicates })
}
sinon.spy(this.db.spellingPreferences, 'findOne')
this.LearnedWordsManager.getLearnedWords(this.token, this.callback)
})
it('should get the word list for the given user', function() {
expect(
this.db.spellingPreferences.findOne.calledWith({ token: this.token })
).to.equal(true)
})
it('should return the word list in the callback without duplicates', function() {
expect(this.callback.calledWith(null, this.wordList)).to.equal(true)
})
})
describe('caching the result', function() {
it('should use the cache first if it is primed', function(done) {
this.wordList = ['apples', 'bananas', 'pears']
this.cache.get.returns(this.wordList)
this.db.spellingPreferences.findOne = sinon.stub()
this.LearnedWordsManager.getLearnedWords(this.token, (err, spellings) => {
expect(err).not.to.exist
this.db.spellingPreferences.findOne.called.should.equal(false)
assert.deepEqual(this.wordList, spellings)
done()
})
})
it('should set the cache after hitting the db', function(done) {
this.wordList = ['apples', 'bananas', 'pears']
this.db.spellingPreferences.findOne = sinon
.stub()
.callsArgWith(1, null, { learnedWords: this.wordList })
this.LearnedWordsManager.getLearnedWords(this.token, () => {
this.cache.set.calledWith(this.token, this.wordList).should.equal(true)
done()
})
})
it('should break cache when update is called', function(done) {
this.word = 'instanton'
this.LearnedWordsManager.learnWord(this.token, this.word, () => {
this.cache.del.calledWith(this.token).should.equal(true)
done()
})
})
})
describe('deleteUsersLearnedWords', function() {
beforeEach(function() {
this.db.spellingPreferences.remove = sinon.stub().callsArgWith(1)
})
it('should get the word list for the given user', function(done) {
this.LearnedWordsManager.deleteUsersLearnedWords(this.token, () => {
this.db.spellingPreferences.remove
.calledWith({ token: this.token })
.should.equal(true)
done()
})
})
})
})