overleaf/services/track-changes/test/unit/js/DocArchive/MongoAWS.js

109 lines
3.1 KiB
JavaScript
Raw Normal View History

/* eslint-disable
handle-callback-err,
no-return-assign,
*/
// 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 modulePath = '../../../../app/js/MongoAWS.js'
const SandboxedModule = require('sandboxed-module')
const { ObjectId } = require('mongodb')
const MemoryStream = require('memorystream')
const zlib = require('zlib')
2015-08-31 17:13:18 -04:00
2020-06-04 04:24:21 -04:00
describe('MongoAWS', function () {
beforeEach(function () {
this.MongoAWS = SandboxedModule.require(modulePath, {
2020-03-23 06:34:57 -04:00
singleOnly: true,
requires: {
'@overleaf/settings': (this.settings = {
trackchanges: {
s3: {
secret: 's3-secret',
2021-07-13 07:04:43 -04:00
key: 's3-key',
},
stores: {
2021-07-13 07:04:43 -04:00
doc_history: 's3-bucket',
},
},
}),
child_process: (this.child_process = {}),
'mongo-uri': (this.mongouri = {}),
'aws-sdk': (this.awssdk = {}),
fs: (this.fs = {}),
's3-streams': (this.S3S = {}),
'./mongodb': { db: (this.db = {}), ObjectId },
JSONStream: (this.JSONStream = {}),
'readline-stream': (this.readline = sinon.stub()),
2021-07-13 07:04:43 -04:00
'@overleaf/metrics': { inc() {} },
},
})
2015-08-31 17:13:18 -04:00
this.project_id = ObjectId().toString()
this.doc_id = ObjectId().toString()
this.pack_id = ObjectId()
this.update = { v: 123 }
return (this.callback = sinon.stub())
})
2015-08-31 17:13:18 -04:00
2020-06-04 04:24:21 -04:00
describe('archivePack', function () {
beforeEach(function (done) {
this.awssdk.config = { update: sinon.stub() }
this.awssdk.S3 = sinon.stub()
this.S3S.WriteStream = () => MemoryStream.createWriteStream()
this.db.docHistory = {}
this.db.docHistory.findOne = sinon
.stub()
.callsArgWith(1, null, { pack: 'hello' })
2015-08-31 17:13:18 -04:00
return this.MongoAWS.archivePack(
this.project_id,
this.doc_id,
this.pack_id,
(err, result) => {
this.callback()
return done()
}
)
})
2015-08-31 17:13:18 -04:00
2020-06-04 04:24:21 -04:00
return it('should call the callback', function () {
return this.callback.called.should.equal(true)
})
})
2015-08-31 17:13:18 -04:00
2020-06-04 04:24:21 -04:00
return describe('unArchivePack', function () {
beforeEach(function (done) {
return zlib.gzip('{"pack":"123"}', (err, zbuf) => {
this.awssdk.config = { update: sinon.stub() }
this.awssdk.S3 = sinon.stub()
this.S3S.ReadStream = () =>
MemoryStream.createReadStream(zbuf, { readable: true })
this.db.docHistory = {}
this.db.docHistory.insertOne = sinon
.stub()
.yields(null, { insertedId: ObjectId() })
2015-08-31 17:13:18 -04:00
return this.MongoAWS.unArchivePack(
this.project_id,
this.doc_id,
this.pack_id,
(err, result) => {
this.callback()
return done()
}
)
})
})
2015-08-31 17:13:18 -04:00
2020-06-04 04:24:21 -04:00
return it('should call db.docHistory.insert', function () {
return this.db.docHistory.insertOne.called.should.equal(true)
})
})
})