Overleaf/services/track-changes/test/unit/coffee/DocArchive/MongoAWS.coffee

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()

		@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