mirror of
https://github.com/overleaf/overleaf.git
synced 2024-12-03 05:49:32 -05:00
59 lines
1.5 KiB
JavaScript
59 lines
1.5 KiB
JavaScript
|
const chai = require('chai')
|
||
|
const chaiAsPromised = require('chai-as-promised')
|
||
|
const config = require('config')
|
||
|
const fetch = require('node-fetch')
|
||
|
const { knex, mongodb } = require('../storage')
|
||
|
|
||
|
chai.use(chaiAsPromised)
|
||
|
|
||
|
async function setupPostgresDatabase() {
|
||
|
await knex.migrate.latest()
|
||
|
}
|
||
|
|
||
|
async function setupMongoDatabase() {
|
||
|
await mongodb.db.collection('projectHistoryChunks').createIndexes([
|
||
|
{
|
||
|
key: { projectId: 1, startVersion: 1 },
|
||
|
name: 'projectId_1_startVersion_1',
|
||
|
partialFilterExpression: { state: 'active' },
|
||
|
unique: true,
|
||
|
},
|
||
|
{
|
||
|
key: { state: 1 },
|
||
|
name: 'state_1',
|
||
|
partialFilterExpression: { state: 'deleted' },
|
||
|
},
|
||
|
])
|
||
|
}
|
||
|
|
||
|
async function createGcsBuckets() {
|
||
|
for (const bucket of [
|
||
|
config.get('blobStore.globalBucket'),
|
||
|
config.get('blobStore.projectBucket'),
|
||
|
config.get('chunkStore.bucket'),
|
||
|
config.get('zipStore.bucket'),
|
||
|
]) {
|
||
|
await fetch('http://gcs:9090/storage/v1/b', {
|
||
|
method: 'POST',
|
||
|
body: JSON.stringify({ name: bucket }),
|
||
|
headers: { 'Content-Type': 'application/json' },
|
||
|
})
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Tear down the connection pool after all the tests have run, so the process
|
||
|
// can exit.
|
||
|
async function tearDownConnectionPool() {
|
||
|
await knex.destroy()
|
||
|
}
|
||
|
|
||
|
module.exports = {
|
||
|
setupPostgresDatabase,
|
||
|
createGcsBuckets,
|
||
|
tearDownConnectionPool,
|
||
|
mochaHooks: {
|
||
|
beforeAll: [setupPostgresDatabase, setupMongoDatabase, createGcsBuckets],
|
||
|
afterAll: [tearDownConnectionPool],
|
||
|
},
|
||
|
}
|