overleaf/services/track-changes/test/unit/coffee/DocArchive/MongoAWS.coffee
2017-08-25 14:47:07 +01:00

70 lines
2 KiB
CoffeeScript

chai = require('chai')
chai.should()
sinon = require("sinon")
modulePath = "../../../../app/js/MongoAWS.js"
SandboxedModule = require('sandboxed-module')
{ObjectId} = require("mongojs")
MemoryStream = require('memorystream')
zlib = require "zlib"
describe "MongoAWS", ->
beforeEach ->
@MongoAWS = SandboxedModule.require modulePath, requires:
"settings-sharelatex": @settings =
trackchanges:
s3:
secret: "s3-secret"
key: "s3-key"
stores:
doc_history: "s3-bucket"
"child_process": @child_process = {}
"mongo-uri": @mongouri = {}
"logger-sharelatex": @logger = {log: sinon.stub(), error: sinon.stub(), err:->}
"aws-sdk": @awssdk = {}
"fs": @fs = {}
"s3-streams": @S3S = {}
"./mongojs" : { db: @db = {}, ObjectId: ObjectId }
"JSONStream": @JSONStream = {}
"readline-stream": @readline = sinon.stub()
'metrics-sharelatex': {inc: ()->}
@project_id = ObjectId().toString()
@doc_id = ObjectId().toString()
@pack_id = ObjectId()
@update = { v:123 }
@callback = sinon.stub()
describe "archivePack", ->
beforeEach (done) ->
@awssdk.config = { update: sinon.stub() }
@awssdk.S3 = sinon.stub()
@S3S.WriteStream = () ->
MemoryStream.createWriteStream()
@db.docHistory = {}
@db.docHistory.findOne = sinon.stub().callsArgWith(1, null, {"pack":"hello"})
@MongoAWS.archivePack @project_id, @doc_id, @pack_id, (err, result) =>
@callback()
done()
it "should call the callback", ->
@callback.called.should.equal true
describe "unArchivePack", ->
beforeEach (done) ->
zlib.gzip '{"pack":"123"}', (err, zbuf) =>
@awssdk.config = { update: sinon.stub() }
@awssdk.S3 = sinon.stub()
@S3S.ReadStream = () ->
MemoryStream.createReadStream(zbuf, {readable:true})
@db.docHistory = {}
@db.docHistory.insert = sinon.stub().callsArgWith(1, null, "pack")
@MongoAWS.unArchivePack @project_id, @doc_id, @pack_id, (err, result) =>
@callback()
done()
it "should call db.docHistory.insert", ->
@db.docHistory.insert.called.should.equal true