mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-21 20:47:08 -05:00
Merge pull request #3678 from overleaf/jpa-back-fill-deleted-files-handle-duplicates
[scripts] back_fill_deleted_files: handle duplicate deleted files GitOrigin-RevId: 3f2f1b662727d61d2da2800ad0d635b65562164b
This commit is contained in:
parent
450d744f9a
commit
30cf22c8a9
2 changed files with 27 additions and 5 deletions
|
@ -46,12 +46,23 @@ async function processProject(project) {
|
|||
|
||||
async function backFillFiles(project) {
|
||||
const projectId = project._id
|
||||
filterDuplicatesInPlace(project)
|
||||
project.deletedFiles.forEach(file => {
|
||||
file.projectId = projectId
|
||||
})
|
||||
await db.deletedFiles.insertMany(project.deletedFiles)
|
||||
}
|
||||
|
||||
function filterDuplicatesInPlace(project) {
|
||||
const fileIds = new Set()
|
||||
project.deletedFiles = project.deletedFiles.filter(file => {
|
||||
const id = file._id.toString()
|
||||
if (fileIds.has(id)) return false
|
||||
fileIds.add(id)
|
||||
return true
|
||||
})
|
||||
}
|
||||
|
||||
async function cleanupProject(project) {
|
||||
await db.projects.updateOne(
|
||||
{ _id: project._id },
|
||||
|
|
|
@ -21,7 +21,7 @@ async function unsetDeletedFiles(projectId) {
|
|||
}
|
||||
|
||||
describe('BackFillDeletedFiles', function() {
|
||||
let user, projectId1, projectId2, projectId3, projectId4
|
||||
let user, projectId1, projectId2, projectId3, projectId4, projectId5
|
||||
|
||||
beforeEach('create projects', async function() {
|
||||
user = new User()
|
||||
|
@ -31,14 +31,16 @@ describe('BackFillDeletedFiles', function() {
|
|||
projectId2 = ObjectId(await user.createProject('project2'))
|
||||
projectId3 = ObjectId(await user.createProject('project3'))
|
||||
projectId4 = ObjectId(await user.createProject('project4'))
|
||||
projectId5 = ObjectId(await user.createProject('project5'))
|
||||
})
|
||||
|
||||
let fileId1, fileId2, fileId3
|
||||
let fileId1, fileId2, fileId3, fileId4
|
||||
beforeEach('create files', function() {
|
||||
// take a short cut and just allocate file ids
|
||||
fileId1 = ObjectId()
|
||||
fileId2 = ObjectId()
|
||||
fileId3 = ObjectId()
|
||||
fileId4 = ObjectId()
|
||||
})
|
||||
const otherFileDetails = {
|
||||
name: 'universe.jpg',
|
||||
|
@ -47,7 +49,7 @@ describe('BackFillDeletedFiles', function() {
|
|||
deletedAt: new Date(),
|
||||
__v: 0
|
||||
}
|
||||
let deletedFiles1, deletedFiles2
|
||||
let deletedFiles1, deletedFiles2, deletedFiles3
|
||||
beforeEach('set deletedFiles details', async function() {
|
||||
deletedFiles1 = [
|
||||
{ _id: fileId1, ...otherFileDetails },
|
||||
|
@ -61,6 +63,12 @@ describe('BackFillDeletedFiles', function() {
|
|||
await setDeletedFiles(projectId3, [])
|
||||
// a project without deletedFiles array
|
||||
await unsetDeletedFiles(projectId4)
|
||||
// duplicate entry
|
||||
deletedFiles3 = [
|
||||
{ _id: fileId4, ...otherFileDetails },
|
||||
{ _id: fileId4, ...otherFileDetails }
|
||||
]
|
||||
await setDeletedFiles(projectId5, deletedFiles3)
|
||||
})
|
||||
|
||||
async function runScript(args = []) {
|
||||
|
@ -81,7 +89,7 @@ describe('BackFillDeletedFiles', function() {
|
|||
expect(stdOut).to.include(projectId1.toString())
|
||||
expect(stdOut).to.include(projectId2.toString())
|
||||
|
||||
expect(stdErr).to.include(`Completed batch ending ${projectId2}`)
|
||||
expect(stdErr).to.include(`Completed batch ending ${projectId5}`)
|
||||
}
|
||||
|
||||
function checkAreFilesBackFilled() {
|
||||
|
@ -92,7 +100,8 @@ describe('BackFillDeletedFiles', function() {
|
|||
expect(docs).to.deep.equal([
|
||||
{ _id: fileId1, projectId: projectId1, ...otherFileDetails },
|
||||
{ _id: fileId2, projectId: projectId1, ...otherFileDetails },
|
||||
{ _id: fileId3, projectId: projectId2, ...otherFileDetails }
|
||||
{ _id: fileId3, projectId: projectId2, ...otherFileDetails },
|
||||
{ _id: fileId4, projectId: projectId5, ...otherFileDetails }
|
||||
])
|
||||
})
|
||||
}
|
||||
|
@ -105,6 +114,7 @@ describe('BackFillDeletedFiles', function() {
|
|||
it('should leave the deletedFiles as is', async function() {
|
||||
expect(await getDeletedFiles(projectId1)).to.deep.equal(deletedFiles1)
|
||||
expect(await getDeletedFiles(projectId2)).to.deep.equal(deletedFiles2)
|
||||
expect(await getDeletedFiles(projectId5)).to.deep.equal(deletedFiles3)
|
||||
})
|
||||
})
|
||||
|
||||
|
@ -118,6 +128,7 @@ describe('BackFillDeletedFiles', function() {
|
|||
it('should cleanup the deletedFiles', async function() {
|
||||
expect(await getDeletedFiles(projectId1)).to.deep.equal([])
|
||||
expect(await getDeletedFiles(projectId2)).to.deep.equal([])
|
||||
expect(await getDeletedFiles(projectId5)).to.deep.equal([])
|
||||
})
|
||||
})
|
||||
})
|
||||
|
|
Loading…
Reference in a new issue