Overleaf/services/web/test/unit/coffee/Project/ProjectDeleterTests.coffee

should = require('chai').should()
modulePath = "../../../../app/js/Features/Project/ProjectDeleter"
SandboxedModule = require('sandboxed-module')
sinon = require('sinon')


describe 'ProjectDeleter', ->

	beforeEach ->
		@project_id = "12312"
		@project = 
			_id: @project_id
			rootFolder:[]
			collaberator_refs:["collab1", "collab2"]
			readOnly_refs:["readOnly1", "readOnly2"]
			owner_ref:"owner ref here"
			remove: sinon.stub().callsArg(0)

		@user_id = 1234

		@mongojs =
			db:
				deletedProjects:
					insert: sinon.stub().callsArg(1)
				projectsDeletedByMigration:
					insert: sinon.stub().callsArg(1)
	
		@Project = 
			update: sinon.stub().callsArgWith(3)
			remove: sinon.stub().callsArgWith(1)
			findOne: sinon.stub().callsArgWith(1, null, @project)
			find: sinon.stub().callsArgWith(1, null, [@project])
			applyToAllFilesRecursivly: sinon.stub()
		@documentUpdaterHandler = 
			flushProjectToMongoAndDelete:sinon.stub().callsArgWith(1)
		@editorController = notifyUsersProjectHasBeenDeletedOrRenamed : sinon.stub().callsArgWith(1)
		@TagsHandler = 
			removeProjectFromAllTags: sinon.stub().callsArgWith(2)
		@CollaboratorsHandler =
			removeUserFromAllProjets: sinon.stub().yields()
			getMemberIds: sinon.stub().withArgs(@project_id).yields(null, ["member-id-1", "member-id-2"])
		@deleter = SandboxedModule.require modulePath, requires:
			"../Editor/EditorController": @editorController
			'../../models/Project':{Project:@Project}
			'../DocumentUpdater/DocumentUpdaterHandler': @documentUpdaterHandler
			"../Tags/TagsHandler":@TagsHandler
			"../FileStore/FileStoreHandler": @FileStoreHandler = {}
			"../Collaborators/CollaboratorsHandler": @CollaboratorsHandler
			"../../infrastructure/mongojs": @mongojs
			'logger-sharelatex':
				log:->

	describe "mark as deleted by external source", ->
		project_id = 1234
		it 'should update the project with the flag set to true', (done)->
			@deleter.markAsDeletedByExternalSource project_id, =>
				conditions = {_id:project_id}
				update = {deletedByExternalDataSource:true}
				@Project.update.calledWith(conditions, update).should.equal true
				done()

		it 'should tell the editor controler so users are notified', (done)->
			@deleter.markAsDeletedByExternalSource project_id, =>
				@editorController.notifyUsersProjectHasBeenDeletedOrRenamed.calledWith(project_id).should.equal true
				done()
				
	describe "unmarkAsDeletedByExternalSource", ->
		beforeEach ->
			@Project.update = sinon.stub().callsArg(3)
			@callback = sinon.stub()
			@project = {
				_id: @project_id
			}
			@deleter.unmarkAsDeletedByExternalSource @project_id, @callback
		
		it "should remove the flag from the project", ->
			@Project.update
				.calledWith({_id: @project_id}, {deletedByExternalDataSource:false})
				.should.equal true

	describe "deleteUsersProjects", ->
		beforeEach ->
			@deleter.deleteProject = sinon.stub().callsArg(1)

		it "should find all the projects owned by the user_id", (done)->
			@deleter.deleteUsersProjects @user_id, =>
				sinon.assert.calledWith(@Project.find, owner_ref: @user_id)
				done()

		it "should call deleteProject on the found projects", (done)->
			@deleter.deleteUsersProjects @user_id, =>
				sinon.assert.calledWith(@deleter.deleteProject, @project._id)
				done()

		it "should call deleteProject once for each project", (done)->
			@Project.find.callsArgWith(1, null, [
				{_id: 'potato'}, {_id: 'wombat'}
			])
			@deleter.deleteUsersProjects @user_id, =>
				sinon.assert.calledTwice(@deleter.deleteProject)
				sinon.assert.calledWith(@deleter.deleteProject, 'wombat')
				sinon.assert.calledWith(@deleter.deleteProject, 'potato')
				done()

		it "should remove all the projects the user is a collaborator of", (done)->
			@deleter.deleteUsersProjects @user_id, =>
				@CollaboratorsHandler.removeUserFromAllProjets.calledWith(@user_id).should.equal true
				done()

	describe "softDeleteUsersProjectsForMigrationForMigration", ->
		beforeEach ->
			@deleter.softDeleteProjectForMigration = sinon.stub().callsArg(1)

		it "should find all the projects owned by the user_id", (done)->
			@deleter.softDeleteUsersProjectsForMigration @user_id, =>
				@Project.find.calledWith(owner_ref: @user_id).should.equal true
				done()

		it "should call deleteProject on the found projects", (done)->
			@deleter.softDeleteUsersProjectsForMigration @user_id, =>
				sinon.assert.calledWith(@deleter.softDeleteProjectForMigration, @project._id)
				done()

		it "should call deleteProject once for each project", (done)->
			@Project.find.callsArgWith(1, null, [
				{_id: 'potato'}, {_id: 'wombat'}
			])
			@deleter.softDeleteUsersProjectsForMigration @user_id, =>
				sinon.assert.calledTwice(@deleter.softDeleteProjectForMigration)
				sinon.assert.calledWith(@deleter.softDeleteProjectForMigration, 'wombat')
				sinon.assert.calledWith(@deleter.softDeleteProjectForMigration, 'potato')
				done()

		it "should remove all the projects the user is a collaborator of", (done)->
			@deleter.softDeleteUsersProjectsForMigration @user_id, =>
				@CollaboratorsHandler.removeUserFromAllProjets.calledWith(@user_id).should.equal true
				done()

	describe "deleteProject", ->
		beforeEach (done) ->
			@project_id = "mock-project-id-123"
			@Project.remove.callsArgWith(1)
			done()

		it "should flushProjectToMongoAndDelete in doc updater", (done)->
			@deleter.deleteProject @project_id, =>
				@documentUpdaterHandler.flushProjectToMongoAndDelete.calledWith(@project_id).should.equal true
				done()

		it "should removeProjectFromAllTags", (done)->
			@deleter.deleteProject @project_id, =>
				@TagsHandler.removeProjectFromAllTags.calledWith("member-id-1", @project_id).should.equal true
				@TagsHandler.removeProjectFromAllTags.calledWith("member-id-2", @project_id).should.equal true
				done()

		it "should remove the project from Mongo", (done) ->
			@deleter.deleteProject @project_id, =>
				@Project.remove.calledWith({
					_id: @project_id
				}).should.equal true
				done()

	describe "softDeleteProjectForMigration", ->
		beforeEach ->
			@deleter.deleteProject = sinon.stub().callsArg(1)

		it "should set the deletedAt time", (done)->
			@deleter.softDeleteProjectForMigration @project_id, =>
				@project.deletedAt.should.exist
				done()

		it "should insert the project into the deleted projects collection", (done)->
			@deleter.softDeleteProjectForMigration @project_id, =>
				sinon.assert.calledWith(@mongojs.db.projectsDeletedByMigration.insert, @project)
				done()

		it "should delete the project", (done)->
			@deleter.softDeleteProjectForMigration @project_id, =>
				sinon.assert.calledWith(@deleter.deleteProject, @project_id)
				done()

	describe "archiveProject", ->
		beforeEach ->
			@Project.update.callsArgWith(2)

		it "should update the project", (done)->
			@deleter.archiveProject @project_id, =>
				@Project.update.calledWith({
					_id:@project_id
				}, {
					$set: { archived: true }
				}).should.equal true
				done()

	describe "restoreProject", ->
		beforeEach ->
			@Project.update.callsArgWith(2)

		it "should unset the archive attribute", (done)->
			@deleter.restoreProject @project_id, =>
				@Project.update.calledWith({
					_id: @project_id
				}, {
					$unset: { archived: true }
				}).should.equal true
				done()