overleaf/services/web/test/UnitTests/coffee/Versioning/AutomaticSnapshotManagerTests.coffee

135 lines
4.7 KiB
CoffeeScript

should = require('chai').should()
sinon = require('sinon')
SandboxedModule = require('sandboxed-module')
assert = require('assert')
path = require("path")
modulePath = path.join __dirname, '../../../../app/js/Features/Versioning/AutomaticSnapshotManager'
tk = require 'timekeeper'
Settings = require "settings-sharelatex"
describe 'AutomaticSnapshotManager', ->
beforeEach ->
tk.freeze(Date.now())
@project_id = "project-id-1234"
@callback = sinon.stub()
@rclient =
auth:->
del: sinon.stub().callsArg(2)
srem: sinon.stub().callsArg(2)
sadd: sinon.stub().callsArg(2)
set: sinon.stub().callsArg(2)
@VersioningApiHandler =
takeSnapshot: sinon.stub().callsArg(2)
@AutomaticSnapshotManager = SandboxedModule.require modulePath,
requires:
'redis' : createClient: () => @rclient
'../../Features/Versioning/VersioningApiHandler' : @VersioningApiHandler
globals:
Date: Date
afterEach -> tk.reset()
describe "markProjectAsUpdated", ->
beforeEach ->
@AutomaticSnapshotManager.markProjectAsUpdated(@project_id, @callback)
it "should update the project's last updated date", ->
@rclient.set.calledWith("project_last_updated:#{@project_id}", Date.now())
it "should add the project to the set needing processing", ->
@rclient.sadd.calledWith("projects_to_snapshot", @project_id)
it "should call the callback", ->
@callback.calledWith().should.equal true
describe "takeSnapshotIfRequired", ->
beforeEach ->
@lastUpdated = Date.now() - Settings.automaticSnapshots.waitTimeAfterLastEdit + 1
@lastSnapshot = Date.now() - Settings.automaticSnapshots.maxTimeBetweenSnapshots + 1
@rclient.get = (key, callback) =>
if key == "project_last_updated:#{@project_id}"
callback(null, @lastUpdated)
else if key == "project_last_snapshot:#{@project_id}"
callback(null, @lastSnapshot)
else
throw new Error("unexpected key: #{key}")
describe "when updated longer than waitTimeAfterLastEdit ago", ->
beforeEach ->
@lastUpdated = Date.now() - Settings.automaticSnapshots.waitTimeAfterLastEdit - 1
@AutomaticSnapshotManager.takeSnapshotIfRequired(@project_id, @callback)
it "should take a snapshot", ->
@VersioningApiHandler.takeSnapshot.calledWith(@project_id, "Automatic snapshot")
.should.equal true
it "should return the callback", ->
@callback.calledWithExactly().should.equal true
describe "when last snapshot longer then maxTimeBetweenSnapshots ago", ->
beforeEach ->
@lastSnapshot = Date.now() - Settings.automaticSnapshots.maxTimeBetweenSnapshots - 1
@AutomaticSnapshotManager.takeSnapshotIfRequired(@project_id, @callback)
it "should take a snapshot", ->
@VersioningApiHandler.takeSnapshot.calledWith(@project_id, "Automatic snapshot")
.should.equal true
it "should return the callback", ->
@callback.calledWithExactly().should.equal true
describe "when no snapshot has been taken", ->
beforeEach ->
@lastSnapshot = null
@AutomaticSnapshotManager.takeSnapshotIfRequired(@project_id, @callback)
it "should take a snapshot", ->
@VersioningApiHandler.takeSnapshot.calledWith(@project_id, "Automatic snapshot")
.should.equal true
it "should return the callback", ->
@callback.calledWithExactly().should.equal true
describe "when no snapshot is needed", ->
beforeEach ->
@AutomaticSnapshotManager.takeSnapshotIfRequired(@project_id, @callback)
it "should take a snapshot", ->
@VersioningApiHandler.takeSnapshot.called.should.equal false
it "should return the callback", ->
@callback.calledWithExactly().should.equal true
describe "takeAutomaticSnapshots", ->
beforeEach ->
@project_ids = ["project-1", "project-2", "project-3"]
@rclient.smembers = (key, callback) =>
if key == "projects_to_snapshot"
callback(null, @project_ids)
else
throw new Error("unexpected key: #{key}")
sinon.stub(@AutomaticSnapshotManager, "takeSnapshotIfRequired").callsArgWith(1, null)
@AutomaticSnapshotManager.takeAutomaticSnapshots @callback
afterEach ->
@AutomaticSnapshotManager.takeSnapshotIfRequired.restore()
it "should call the callback", ->
@callback.calledWith(undefined).should.equal true
describe "removing project from marked set", ->
beforeEach ->
@AutomaticSnapshotManager.markProjectAsUpdated @project_id, =>
@AutomaticSnapshotManager.unmarkProjectAsUpdated @project_id, @callback
it "should update the project's last updated date", ->
@rclient.del.calledWith("project_last_updated:#{@project_id}", Date.now())
it "should add the project to the set needing processing", ->
@rclient.srem.calledWith("projects_to_snapshot", @project_id)
it "should call the callback", ->
@callback.calledWith().should.equal true