mirror of
https://github.com/overleaf/overleaf.git
synced 2024-10-24 21:12:38 -04:00
e0dba75b81
GitOrigin-RevId: ab45010ec557d62576c470d2e024549e67261c66
256 lines
7.7 KiB
JavaScript
256 lines
7.7 KiB
JavaScript
import { db, ObjectId } from '../../../app/src/infrastructure/mongodb.js'
|
|
import { promisify } from 'util'
|
|
import { exec } from 'child_process'
|
|
import logger from '@overleaf/logger'
|
|
import { expect } from 'chai'
|
|
|
|
describe('RemoveDeletedUsersFromTokenAccessRefsTests', function () {
|
|
const userId1 = new ObjectId()
|
|
const userId2 = new ObjectId()
|
|
const userId3 = new ObjectId()
|
|
|
|
let insertedUsersCount
|
|
beforeEach('insert users', async function () {
|
|
const users = await db.users.insertMany([
|
|
{ _id: userId1, email: 'user1@example.com' },
|
|
])
|
|
insertedUsersCount = users.insertedCount
|
|
})
|
|
|
|
const projectId1 = new ObjectId('65d726e807c024c8db43be22')
|
|
const projectId2 = new ObjectId('65d726e807c024c8db43be23')
|
|
const projectId3 = new ObjectId('65d726e807c024c8db43be24')
|
|
const projectId4 = new ObjectId('65d726e807c024c8db43be25')
|
|
|
|
let insertedProjects
|
|
beforeEach('insert projects', async function () {
|
|
insertedProjects = await db.projects.insertMany([
|
|
{
|
|
_id: projectId1,
|
|
tokenAccessReadAndWrite_refs: [userId1],
|
|
tokenAccessReadOnly_refs: [],
|
|
},
|
|
{
|
|
_id: projectId2,
|
|
tokenAccessReadAndWrite_refs: [userId2],
|
|
tokenAccessReadOnly_refs: [],
|
|
},
|
|
{
|
|
_id: projectId3,
|
|
tokenAccessReadAndWrite_refs: [userId3],
|
|
},
|
|
{
|
|
_id: projectId4,
|
|
},
|
|
])
|
|
})
|
|
|
|
let stdOut
|
|
|
|
const runScript = async (dryRun, projectsList) => {
|
|
let result
|
|
try {
|
|
result = await promisify(exec)(
|
|
[
|
|
'VERBOSE_LOGGING=true',
|
|
'node',
|
|
'scripts/remove_deleted_users_from_token_access_refs',
|
|
dryRun,
|
|
projectsList,
|
|
].join(' ')
|
|
)
|
|
} catch (error) {
|
|
// dump details like exit code, stdErr and stdOut
|
|
logger.error({ error }, 'script failed')
|
|
throw error
|
|
}
|
|
const { stdout } = result
|
|
stdOut = stdout
|
|
|
|
expect(stdOut).to.match(new RegExp(`User ids count: ${insertedUsersCount}`))
|
|
}
|
|
|
|
describe('dry-run=true', function () {
|
|
beforeEach('run script', async function () {
|
|
await runScript('--dry-run=true')
|
|
expect(stdOut).to.match(/doing dry run/i)
|
|
})
|
|
|
|
it('should show current user id to be removed', function () {
|
|
expect(stdOut).to.match(
|
|
new RegExp(
|
|
`Found deleted user id: ${userId2.toString()} in project: ${projectId2.toString()}`
|
|
)
|
|
)
|
|
expect(stdOut).to.match(
|
|
new RegExp(
|
|
`DRY RUN - would remove deleted ${userId2.toString()} from all projects \\(found in project ${projectId2.toString()}\\)`
|
|
)
|
|
)
|
|
expect(stdOut).to.match(
|
|
new RegExp(
|
|
`Found deleted user id: ${userId3.toString()} in project: ${projectId3.toString()}`
|
|
)
|
|
)
|
|
expect(stdOut).to.match(
|
|
new RegExp(
|
|
`DRY RUN - would remove deleted ${userId3.toString()} from all projects \\(found in project ${projectId3.toString()}\\)`
|
|
)
|
|
)
|
|
})
|
|
|
|
it('should show projects with non-existing token access fields', function () {
|
|
expect(stdOut)
|
|
.to.match(
|
|
new RegExp(
|
|
`DRY RUN - would fix non-existing token access fields in project ${projectId3.toString()}`
|
|
)
|
|
)
|
|
.and.match(
|
|
new RegExp(
|
|
`DRY RUN - would fix non-existing token access fields in project ${projectId4.toString()}`
|
|
)
|
|
)
|
|
})
|
|
|
|
it('should show the user ids (and their count) to be deleted', function () {
|
|
expect(stdOut).to.match(
|
|
new RegExp(
|
|
`DRY RUN - would delete user ids \\(2\\)\\n${userId2.toString()}\\n${userId3.toString()}`
|
|
)
|
|
)
|
|
})
|
|
|
|
it('should show the project ids (and their count) that needs fixing', function () {
|
|
expect(stdOut).to.match(
|
|
new RegExp(
|
|
`Projects with deleted user ids \\(2\\)\\n${projectId2.toString()}\\n${projectId3.toString()}`
|
|
)
|
|
)
|
|
})
|
|
|
|
it('should not fix the token access fields of projects', async function () {
|
|
const projects = await db.projects
|
|
.find({}, { $sort: { _id: 1 } })
|
|
.toArray()
|
|
expect(projects).to.deep.equal([
|
|
{
|
|
_id: projectId1,
|
|
tokenAccessReadAndWrite_refs: [userId1],
|
|
tokenAccessReadOnly_refs: [],
|
|
},
|
|
{
|
|
_id: projectId2,
|
|
tokenAccessReadAndWrite_refs: [userId2],
|
|
tokenAccessReadOnly_refs: [],
|
|
},
|
|
{
|
|
_id: projectId3,
|
|
tokenAccessReadAndWrite_refs: [userId3],
|
|
},
|
|
{
|
|
_id: projectId4,
|
|
},
|
|
])
|
|
})
|
|
})
|
|
|
|
describe('dry-run=false', function () {
|
|
beforeEach('run script', async function () {
|
|
await runScript('--dry-run=false')
|
|
expect(stdOut).to.not.match(/dry run/i)
|
|
})
|
|
|
|
it('should show current user id to be removed', function () {
|
|
expect(stdOut).to.match(
|
|
new RegExp(
|
|
`Found deleted user id: ${userId2.toString()} in project: ${projectId2.toString()}`
|
|
)
|
|
)
|
|
expect(stdOut).to.match(
|
|
new RegExp(
|
|
`Removing deleted ${userId2.toString()} from all projects \\(found in project ${projectId2.toString()}\\)`
|
|
)
|
|
)
|
|
expect(stdOut).to.match(
|
|
new RegExp(
|
|
`Found deleted user id: ${userId3.toString()} in project: ${projectId3.toString()}`
|
|
)
|
|
)
|
|
expect(stdOut).to.match(
|
|
new RegExp(
|
|
`Removing deleted ${userId3.toString()} from all projects \\(found in project ${projectId3.toString()}\\)`
|
|
)
|
|
)
|
|
})
|
|
|
|
it('should show fixed projects with non-existing token access fields', function () {
|
|
expect(stdOut)
|
|
.to.match(
|
|
new RegExp(
|
|
`Fixed non-existing token access fields in project ${projectId3.toString()}`
|
|
)
|
|
)
|
|
.and.match(
|
|
new RegExp(
|
|
`Fixed non-existing token access fields in project ${projectId4.toString()}`
|
|
)
|
|
)
|
|
})
|
|
|
|
it('should show the deleted user ids (and their count) that were removed', function () {
|
|
expect(stdOut).to.match(
|
|
new RegExp(
|
|
`Deleted user ids \\(2\\)\\n${userId2.toString()}\\n${userId3.toString()}`
|
|
)
|
|
)
|
|
})
|
|
|
|
it('should show the project ids (and their count) that were fixed', function () {
|
|
expect(stdOut).to.match(
|
|
new RegExp(
|
|
`Projects with deleted user ids \\(2\\)\\n${projectId2.toString()}\\n${projectId3.toString()}`
|
|
)
|
|
)
|
|
})
|
|
|
|
it('should fix the token access fields of projects', async function () {
|
|
const [, ...fixedProjects] = await db.projects
|
|
.find({}, { $sort: { _id: 1 } })
|
|
.toArray()
|
|
expect(fixedProjects).to.deep.equal([
|
|
{
|
|
_id: projectId2,
|
|
tokenAccessReadAndWrite_refs: [],
|
|
tokenAccessReadOnly_refs: [],
|
|
},
|
|
{
|
|
_id: projectId3,
|
|
tokenAccessReadAndWrite_refs: [],
|
|
tokenAccessReadOnly_refs: [],
|
|
},
|
|
{
|
|
_id: projectId4,
|
|
tokenAccessReadOnly_refs: [],
|
|
tokenAccessReadAndWrite_refs: [],
|
|
},
|
|
])
|
|
})
|
|
})
|
|
|
|
describe('projects=projectId2', function () {
|
|
beforeEach('run script', async function () {
|
|
const projectId2 = insertedProjects.insertedIds[1]
|
|
await runScript('--dry-run=false', `--projects=${projectId2.toString()}`)
|
|
})
|
|
|
|
it('should fix only the projects provided', async function () {
|
|
const [project1, project2, project3] = await db.projects
|
|
.find({}, { $sort: { _id: 1 } })
|
|
.toArray()
|
|
expect(project1.tokenAccessReadAndWrite_refs.length).to.be.gt(0)
|
|
expect(project2.tokenAccessReadAndWrite_refs.length).to.eq(0) // deleted user removed
|
|
expect(project3.tokenAccessReadAndWrite_refs.length).to.be.gt(0)
|
|
})
|
|
})
|
|
})
|