overleaf/services/document-updater/test/unit/js/DocumentManager/DocumentManagerTests.js

1067 lines
30 KiB
JavaScript
Raw Normal View History

const sinon = require('sinon')
const modulePath = '../../../../app/js/DocumentManager.js'
const SandboxedModule = require('sandboxed-module')
const Errors = require('../../../../app/js/Errors')
const tk = require('timekeeper')
describe('DocumentManager', function () {
beforeEach(function () {
tk.freeze(new Date())
this.Metrics = {
Timer: class Timer {},
}
this.Metrics.Timer.prototype.done = sinon.stub()
this.DocumentManager = SandboxedModule.require(modulePath, {
requires: {
'./RedisManager': (this.RedisManager = {}),
'./ProjectHistoryRedisManager': (this.ProjectHistoryRedisManager = {}),
'./PersistenceManager': (this.PersistenceManager = {}),
'./HistoryManager': (this.HistoryManager = {
flushDocChangesAsync: sinon.stub(),
2021-07-13 07:04:42 -04:00
flushProjectChangesAsync: sinon.stub(),
}),
'./Metrics': this.Metrics,
'./RealTimeRedisManager': (this.RealTimeRedisManager = {}),
'./DiffCodec': (this.DiffCodec = {}),
'./UpdateManager': (this.UpdateManager = {}),
2020-05-15 14:29:49 -04:00
'./RangesManager': (this.RangesManager = {}),
2021-07-13 07:04:42 -04:00
'./Errors': Errors,
},
})
this.project_id = 'project-id-123'
this.projectHistoryId = 'history-id-123'
this.projectHistoryType = 'project-history'
this.doc_id = 'doc-id-123'
this.user_id = 1234
this.callback = sinon.stub()
this.lines = ['one', 'two', 'three']
this.version = 42
this.ranges = { comments: 'mock', entries: 'mock' }
this.pathname = '/a/b/c.tex'
this.unflushedTime = Date.now()
this.lastUpdatedAt = Date.now()
this.lastUpdatedBy = 'last-author-id'
})
afterEach(function () {
tk.reset()
})
describe('flushAndDeleteDoc', function () {
describe('successfully', function () {
beforeEach(function () {
this.RedisManager.removeDocFromMemory = sinon.stub().callsArg(2)
this.DocumentManager.flushDocIfLoaded = sinon.stub().callsArgWith(2)
this.DocumentManager.flushAndDeleteDoc(
this.project_id,
this.doc_id,
{},
this.callback
)
})
it('should flush the doc', function () {
this.DocumentManager.flushDocIfLoaded
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should remove the doc from redis', function () {
this.RedisManager.removeDocFromMemory
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should call the callback without error', function () {
this.callback.calledWith(null).should.equal(true)
})
it('should time the execution', function () {
this.Metrics.Timer.prototype.done.called.should.equal(true)
})
it('should flush to the history api', function () {
this.HistoryManager.flushDocChangesAsync
.calledWithExactly(this.project_id, this.doc_id)
.should.equal(true)
})
})
describe('when a flush error occurs', function () {
beforeEach(function () {
this.DocumentManager.flushDocIfLoaded = sinon
.stub()
.callsArgWith(2, new Error('boom!'))
this.RedisManager.removeDocFromMemory = sinon.stub().callsArg(2)
})
it('should not remove the doc from redis', function (done) {
this.DocumentManager.flushAndDeleteDoc(
this.project_id,
this.doc_id,
{},
2021-07-13 07:04:42 -04:00
error => {
error.should.exist
this.RedisManager.removeDocFromMemory.called.should.equal(false)
done()
}
)
})
describe('when ignoring flush errors', function () {
it('should remove the doc from redis', function (done) {
this.DocumentManager.flushAndDeleteDoc(
this.project_id,
this.doc_id,
{ ignoreFlushErrors: true },
2021-07-13 07:04:42 -04:00
error => {
if (error) {
return done(error)
}
this.RedisManager.removeDocFromMemory.called.should.equal(true)
done()
}
)
})
})
})
})
describe('flushDocIfLoaded', function () {
describe('when the doc is in Redis', function () {
beforeEach(function () {
this.RedisManager.getDoc = sinon
.stub()
.callsArgWith(
2,
null,
this.lines,
this.version,
this.ranges,
this.pathname,
this.projectHistoryId,
this.unflushedTime,
this.lastUpdatedAt,
this.lastUpdatedBy
)
this.RedisManager.clearUnflushedTime = sinon
.stub()
.callsArgWith(1, null)
this.PersistenceManager.setDoc = sinon.stub().yields()
this.DocumentManager.flushDocIfLoaded(
this.project_id,
this.doc_id,
this.callback
)
})
it('should get the doc from redis', function () {
this.RedisManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should write the doc lines to the persistence layer', function () {
this.PersistenceManager.setDoc
.calledWith(
this.project_id,
this.doc_id,
this.lines,
this.version,
this.ranges,
this.lastUpdatedAt,
this.lastUpdatedBy
)
.should.equal(true)
})
it('should call the callback without error', function () {
this.callback.calledWith(null).should.equal(true)
})
it('should time the execution', function () {
this.Metrics.Timer.prototype.done.called.should.equal(true)
})
})
describe('when the document is not in Redis', function () {
beforeEach(function () {
this.RedisManager.getDoc = sinon
.stub()
.callsArgWith(2, null, null, null, null)
this.PersistenceManager.setDoc = sinon.stub().yields()
this.DocumentManager.flushDocIfLoaded(
this.project_id,
this.doc_id,
this.callback
)
})
it('should get the doc from redis', function () {
this.RedisManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should not write anything to the persistence layer', function () {
this.PersistenceManager.setDoc.called.should.equal(false)
})
it('should call the callback without error', function () {
this.callback.calledWith(null).should.equal(true)
})
it('should time the execution', function () {
this.Metrics.Timer.prototype.done.called.should.equal(true)
})
})
})
describe('getDocAndRecentOps', function () {
describe('with a previous version specified', function () {
beforeEach(function () {
this.DocumentManager.getDoc = sinon
.stub()
.callsArgWith(
2,
null,
this.lines,
this.version,
this.ranges,
this.pathname,
this.projectHistoryId
)
this.RedisManager.getPreviousDocOps = sinon
.stub()
.callsArgWith(3, null, this.ops)
this.DocumentManager.getDocAndRecentOps(
this.project_id,
this.doc_id,
this.fromVersion,
this.callback
)
})
it('should get the doc', function () {
this.DocumentManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should get the doc ops', function () {
this.RedisManager.getPreviousDocOps
.calledWith(this.doc_id, this.fromVersion, this.version)
.should.equal(true)
})
it('should call the callback with the doc info', function () {
this.callback
.calledWith(
null,
this.lines,
this.version,
this.ops,
this.ranges,
this.pathname,
this.projectHistoryId
)
.should.equal(true)
})
it('should time the execution', function () {
this.Metrics.Timer.prototype.done.called.should.equal(true)
})
})
describe('with no previous version specified', function () {
beforeEach(function () {
this.DocumentManager.getDoc = sinon
.stub()
.callsArgWith(
2,
null,
this.lines,
this.version,
this.ranges,
this.pathname,
this.projectHistoryId
)
this.RedisManager.getPreviousDocOps = sinon
.stub()
.callsArgWith(3, null, this.ops)
this.DocumentManager.getDocAndRecentOps(
this.project_id,
this.doc_id,
-1,
this.callback
)
})
it('should get the doc', function () {
this.DocumentManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should not need to get the doc ops', function () {
this.RedisManager.getPreviousDocOps.called.should.equal(false)
})
it('should call the callback with the doc info', function () {
this.callback
.calledWith(
null,
this.lines,
this.version,
[],
this.ranges,
this.pathname,
this.projectHistoryId
)
.should.equal(true)
})
it('should time the execution', function () {
this.Metrics.Timer.prototype.done.called.should.equal(true)
})
})
})
describe('getDoc', function () {
describe('when the doc exists in Redis', function () {
beforeEach(function () {
this.RedisManager.getDoc = sinon
.stub()
.callsArgWith(
2,
null,
this.lines,
this.version,
this.ranges,
this.pathname,
this.projectHistoryId,
this.unflushedTime
)
this.DocumentManager.getDoc(this.project_id, this.doc_id, this.callback)
})
it('should get the doc from Redis', function () {
this.RedisManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should call the callback with the doc info', function () {
this.callback
.calledWith(
null,
this.lines,
this.version,
this.ranges,
this.pathname,
this.projectHistoryId,
this.unflushedTime,
true
)
.should.equal(true)
})
it('should time the execution', function () {
this.Metrics.Timer.prototype.done.called.should.equal(true)
})
})
describe('when the doc does not exist in Redis', function () {
beforeEach(function () {
this.RedisManager.getDoc = sinon
.stub()
.callsArgWith(2, null, null, null, null, null, null)
this.PersistenceManager.getDoc = sinon
.stub()
.callsArgWith(
2,
null,
this.lines,
this.version,
this.ranges,
this.pathname,
this.projectHistoryId,
this.projectHistoryType
)
this.RedisManager.putDocInMemory = sinon.stub().yields()
this.RedisManager.setHistoryType = sinon.stub().yields()
this.DocumentManager.getDoc(this.project_id, this.doc_id, this.callback)
})
it('should try to get the doc from Redis', function () {
this.RedisManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should get the doc from the PersistenceManager', function () {
this.PersistenceManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should set the doc in Redis', function () {
this.RedisManager.putDocInMemory
.calledWith(
this.project_id,
this.doc_id,
this.lines,
this.version,
this.ranges,
this.pathname,
this.projectHistoryId
)
.should.equal(true)
})
it('should set the history type in Redis', function () {
this.RedisManager.setHistoryType
.calledWith(this.doc_id, this.projectHistoryType)
.should.equal(true)
})
it('should call the callback with the doc info', function () {
this.callback
.calledWith(
null,
this.lines,
this.version,
this.ranges,
this.pathname,
this.projectHistoryId,
null,
false
)
.should.equal(true)
})
it('should time the execution', function () {
this.Metrics.Timer.prototype.done.called.should.equal(true)
})
})
})
describe('setDoc', function () {
describe('with plain tex lines', function () {
beforeEach(function () {
this.beforeLines = ['before', 'lines']
this.afterLines = ['after', 'lines']
this.ops = [
{ i: 'foo', p: 4 },
2021-07-13 07:04:42 -04:00
{ d: 'bar', p: 42 },
]
this.DocumentManager.getDoc = sinon
.stub()
.callsArgWith(
2,
null,
this.beforeLines,
this.version,
this.ranges,
this.pathname,
this.projectHistoryId,
this.unflushedTime,
true
)
this.DiffCodec.diffAsShareJsOp = sinon
.stub()
.callsArgWith(2, null, this.ops)
this.UpdateManager.applyUpdate = sinon.stub().callsArgWith(3, null)
this.DocumentManager.flushDocIfLoaded = sinon.stub().callsArg(2)
this.DocumentManager.flushAndDeleteDoc = sinon.stub().callsArg(3)
})
describe('when already loaded', function () {
beforeEach(function () {
this.DocumentManager.setDoc(
this.project_id,
this.doc_id,
this.afterLines,
this.source,
this.user_id,
false,
this.callback
)
})
it('should get the current doc lines', function () {
this.DocumentManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should return a diff of the old and new lines', function () {
this.DiffCodec.diffAsShareJsOp
.calledWith(this.beforeLines, this.afterLines)
.should.equal(true)
})
it('should apply the diff as a ShareJS op', function () {
this.UpdateManager.applyUpdate
.calledWith(this.project_id, this.doc_id, {
doc: this.doc_id,
v: this.version,
op: this.ops,
meta: {
type: 'external',
source: this.source,
2021-07-13 07:04:42 -04:00
user_id: this.user_id,
},
})
.should.equal(true)
})
it('should flush the doc to Mongo', function () {
this.DocumentManager.flushDocIfLoaded
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should not flush the project history', function () {
this.HistoryManager.flushProjectChangesAsync.called.should.equal(
false
)
})
it('should call the callback', function () {
this.callback.calledWith(null).should.equal(true)
})
it('should time the execution', function () {
this.Metrics.Timer.prototype.done.called.should.equal(true)
})
})
describe('when not already loaded', function () {
beforeEach(function () {
this.DocumentManager.getDoc = sinon
.stub()
.callsArgWith(
2,
null,
this.beforeLines,
this.version,
this.pathname,
null,
false
)
this.DocumentManager.setDoc(
this.project_id,
this.doc_id,
this.afterLines,
this.source,
this.user_id,
false,
this.callback
)
})
it('should flush and delete the doc from the doc updater', function () {
this.DocumentManager.flushAndDeleteDoc
.calledWith(this.project_id, this.doc_id, {})
.should.equal(true)
})
it('should not flush the project history', function () {
this.HistoryManager.flushProjectChangesAsync
.calledWithExactly(this.project_id)
.should.equal(true)
})
})
describe('without new lines', function () {
beforeEach(function () {
this.DocumentManager.setDoc(
this.project_id,
this.doc_id,
null,
this.source,
this.user_id,
false,
this.callback
)
})
it('should return the callback with an error', function () {
this.callback.calledWith(new Error('No lines were passed to setDoc'))
})
it('should not try to get the doc lines', function () {
this.DocumentManager.getDoc.called.should.equal(false)
})
})
describe('with the undoing flag', function () {
beforeEach(function () {
// Copy ops so we don't interfere with other tests
this.ops = [
{ i: 'foo', p: 4 },
2021-07-13 07:04:42 -04:00
{ d: 'bar', p: 42 },
]
this.DiffCodec.diffAsShareJsOp = sinon
.stub()
.callsArgWith(2, null, this.ops)
this.DocumentManager.setDoc(
this.project_id,
this.doc_id,
this.afterLines,
this.source,
this.user_id,
true,
this.callback
)
})
it('should set the undo flag on each op', function () {
this.ops.map(op => op.u.should.equal(true))
})
})
})
})
describe('acceptChanges', function () {
beforeEach(function () {
this.change_id = 'mock-change-id'
this.change_ids = [
'mock-change-id-1',
'mock-change-id-2',
'mock-change-id-3',
2021-07-13 07:04:42 -04:00
'mock-change-id-4',
]
this.version = 34
this.lines = ['original', 'lines']
this.ranges = { entries: 'mock', comments: 'mock' }
this.updated_ranges = { entries: 'updated', comments: 'updated' }
this.DocumentManager.getDoc = sinon
.stub()
.yields(null, this.lines, this.version, this.ranges)
this.RangesManager.acceptChanges = sinon
.stub()
.yields(null, this.updated_ranges)
this.RedisManager.updateDocument = sinon.stub().yields()
})
describe('successfully with a single change', function () {
beforeEach(function () {
this.DocumentManager.acceptChanges(
this.project_id,
this.doc_id,
[this.change_id],
this.callback
)
})
it("should get the document's current ranges", function () {
this.DocumentManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should apply the accept change to the ranges', function () {
this.RangesManager.acceptChanges
.calledWith([this.change_id], this.ranges)
.should.equal(true)
})
it('should save the updated ranges', function () {
this.RedisManager.updateDocument
.calledWith(
this.project_id,
this.doc_id,
this.lines,
this.version,
[],
this.updated_ranges,
{}
)
.should.equal(true)
})
it('should call the callback', function () {
this.callback.called.should.equal(true)
})
})
describe('successfully with multiple changes', function () {
beforeEach(function () {
this.DocumentManager.acceptChanges(
this.project_id,
this.doc_id,
this.change_ids,
this.callback
)
})
it('should apply the accept change to the ranges', function () {
this.RangesManager.acceptChanges
.calledWith(this.change_ids, this.ranges)
.should.equal(true)
})
})
describe('when the doc is not found', function () {
beforeEach(function () {
this.DocumentManager.getDoc = sinon
.stub()
.yields(null, null, null, null)
this.DocumentManager.acceptChanges(
this.project_id,
this.doc_id,
[this.change_id],
this.callback
)
})
it('should not save anything', function () {
this.RedisManager.updateDocument.called.should.equal(false)
})
it('should call the callback with a not found error', function () {
this.callback
2020-05-15 14:29:49 -04:00
.calledWith(sinon.match.instanceOf(Errors.NotFoundError))
.should.equal(true)
})
})
})
describe('deleteComment', function () {
beforeEach(function () {
this.comment_id = 'mock-comment-id'
this.version = 34
this.lines = ['original', 'lines']
this.ranges = { comments: ['one', 'two', 'three'] }
this.updated_ranges = { comments: ['one', 'three'] }
this.DocumentManager.getDoc = sinon
.stub()
.yields(null, this.lines, this.version, this.ranges)
this.RangesManager.deleteComment = sinon
.stub()
.yields(null, this.updated_ranges)
this.RedisManager.updateDocument = sinon.stub().yields()
})
describe('successfully', function () {
beforeEach(function () {
this.DocumentManager.deleteComment(
this.project_id,
this.doc_id,
this.comment_id,
this.callback
)
})
it("should get the document's current ranges", function () {
this.DocumentManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should delete the comment from the ranges', function () {
this.RangesManager.deleteComment
.calledWith(this.comment_id, this.ranges)
.should.equal(true)
})
it('should save the updated ranges', function () {
this.RedisManager.updateDocument
.calledWith(
this.project_id,
this.doc_id,
this.lines,
this.version,
[],
this.updated_ranges,
{}
)
.should.equal(true)
})
it('should call the callback', function () {
this.callback.called.should.equal(true)
})
})
describe('when the doc is not found', function () {
beforeEach(function () {
this.DocumentManager.getDoc = sinon
.stub()
.yields(null, null, null, null)
this.DocumentManager.acceptChanges(
this.project_id,
this.doc_id,
[this.comment_id],
this.callback
)
})
it('should not save anything', function () {
this.RedisManager.updateDocument.called.should.equal(false)
})
it('should call the callback with a not found error', function () {
this.callback
2020-05-15 14:29:49 -04:00
.calledWith(sinon.match.instanceOf(Errors.NotFoundError))
.should.equal(true)
})
})
})
describe('getDocAndFlushIfOld', function () {
beforeEach(function () {
this.DocumentManager.flushDocIfLoaded = sinon.stub().callsArg(2)
})
describe('when the doc is in Redis', function () {
describe('and has changes to be flushed', function () {
beforeEach(function () {
this.DocumentManager.getDoc = sinon
.stub()
.callsArgWith(
2,
null,
this.lines,
this.version,
this.ranges,
this.projectHistoryId,
this.pathname,
Date.now() - 1e9,
true
)
this.DocumentManager.getDocAndFlushIfOld(
this.project_id,
this.doc_id,
this.callback
)
})
it('should get the doc', function () {
this.DocumentManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should flush the doc', function () {
this.DocumentManager.flushDocIfLoaded
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should call the callback with the lines and versions', function () {
this.callback
.calledWith(null, this.lines, this.version)
.should.equal(true)
})
})
describe("and has only changes that don't need to be flushed", function () {
beforeEach(function () {
this.DocumentManager.getDoc = sinon
.stub()
.callsArgWith(
2,
null,
this.lines,
this.version,
this.ranges,
this.pathname,
Date.now() - 100,
true
)
this.DocumentManager.getDocAndFlushIfOld(
this.project_id,
this.doc_id,
this.callback
)
})
it('should get the doc', function () {
this.DocumentManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should not flush the doc', function () {
this.DocumentManager.flushDocIfLoaded.called.should.equal(false)
})
it('should call the callback with the lines and versions', function () {
this.callback
.calledWith(null, this.lines, this.version)
.should.equal(true)
})
})
})
describe('when the doc is not in Redis', function () {
beforeEach(function () {
this.DocumentManager.getDoc = sinon
.stub()
.callsArgWith(
2,
null,
this.lines,
this.version,
this.ranges,
null,
false
)
this.DocumentManager.getDocAndFlushIfOld(
this.project_id,
this.doc_id,
this.callback
)
})
it('should get the doc', function () {
this.DocumentManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('should not flush the doc', function () {
this.DocumentManager.flushDocIfLoaded.called.should.equal(false)
})
it('should call the callback with the lines and versions', function () {
this.callback
.calledWith(null, this.lines, this.version)
.should.equal(true)
})
})
})
describe('renameDoc', function () {
beforeEach(function () {
this.update = 'some-update'
this.RedisManager.renameDoc = sinon.stub().yields()
})
describe('successfully', function () {
beforeEach(function () {
this.DocumentManager.renameDoc(
this.project_id,
this.doc_id,
this.user_id,
this.update,
this.projectHistoryId,
this.callback
)
})
it('should rename the document', function () {
this.RedisManager.renameDoc
.calledWith(
this.project_id,
this.doc_id,
this.user_id,
this.update,
this.projectHistoryId
)
.should.equal(true)
})
it('should call the callback', function () {
this.callback.called.should.equal(true)
})
})
})
describe('resyncDocContents', function () {
describe('when doc is loaded in redis', function () {
beforeEach(function () {
this.RedisManager.getDoc = sinon
.stub()
.callsArgWith(
2,
null,
this.lines,
this.version,
this.ranges,
this.pathname,
this.projectHistoryId
)
this.ProjectHistoryRedisManager.queueResyncDocContent = sinon.stub()
this.DocumentManager.resyncDocContents(
this.project_id,
this.doc_id,
this.callback
)
})
it('gets the doc contents from redis', function () {
this.RedisManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('queues a resync doc content update', function () {
this.ProjectHistoryRedisManager.queueResyncDocContent
.calledWith(
this.project_id,
this.projectHistoryId,
this.doc_id,
this.lines,
this.version,
this.pathname,
this.callback
)
.should.equal(true)
})
})
describe('when doc is not loaded in redis', function () {
beforeEach(function () {
this.RedisManager.getDoc = sinon.stub().callsArgWith(2, null)
this.PersistenceManager.getDoc = sinon
.stub()
.yields(
null,
this.lines,
this.version,
this.ranges,
this.pathname,
this.projectHistoryId
)
this.ProjectHistoryRedisManager.queueResyncDocContent = sinon.stub()
this.DocumentManager.resyncDocContents(
this.project_id,
this.doc_id,
this.callback
)
})
it('tries to get the doc contents from redis', function () {
this.RedisManager.getDoc
.calledWith(this.project_id, this.doc_id)
.should.equal(true)
})
it('gets the doc contents from web', function () {
this.PersistenceManager.getDoc
.calledWith(this.project_id, this.doc_id, { peek: true })
.should.equal(true)
})
it('queues a resync doc content update', function () {
this.ProjectHistoryRedisManager.queueResyncDocContent
.calledWith(
this.project_id,
this.projectHistoryId,
this.doc_id,
this.lines,
this.version,
this.pathname,
this.callback
)
.should.equal(true)
})
})
})
})