Files
speckle-server/packages/server/test/hooks.js
T
2022-03-29 16:30:49 +03:00

80 lines
2.1 KiB
JavaScript

/* istanbul ignore file */
require('../bootstrap')
const chai = require('chai')
const chaiHttp = require('chai-http')
const deepEqualInAnyOrder = require('deep-equal-in-any-order')
const knex = require(`@/db/knex`)
const { init, startHttp } = require(`@/app`)
chai.use(chaiHttp)
chai.use(deepEqualInAnyOrder)
const unlock = async () => {
const exists = await knex.schema.hasTable('knex_migrations_lock')
if (exists) {
await knex('knex_migrations_lock').update('is_locked', '0')
}
}
exports.truncateTables = async (tableNames) => {
if (!tableNames?.length) {
//why is server config only created once!????
const protectedTables = ['server_config']
// const protectedTables = [ 'server_config', 'user_roles', 'scopes', 'server_acl' ]
tableNames = (
await knex('pg_tables')
.select('tablename')
.where({ schemaname: 'public' })
.whereRaw("tablename not like '%knex%'")
.whereNotIn('tablename', protectedTables)
).map((table) => table.tablename)
}
await knex.raw(`truncate table ${tableNames.join(',')} cascade`)
}
const initializeTestServer = async (app) => {
let serverAddress
let wsAddress
const { server } = await startHttp(app, 0)
app.on('appStarted', () => {
const port = server.address().port
serverAddress = `http://localhost:${port}`
wsAddress = `ws://localhost:${port}`
})
while (!serverAddress) {
await new Promise((resolve) => setTimeout(resolve, 100))
}
return {
server,
serverAddress,
wsAddress,
sendRequest(auth, obj) {
return chai.request(serverAddress).post('/graphql').set('Authorization', auth).send(obj)
}
}
}
exports.mochaHooks = {
beforeAll: async () => {
console.log('running before all')
await unlock()
await knex.migrate.rollback()
await knex.migrate.latest()
await init()
},
afterAll: async () => {
console.log('running after all')
await unlock()
}
}
exports.beforeEachContext = async () => {
await exports.truncateTables()
const { app, graphqlServer } = await init()
return { app, graphqlServer }
}
exports.initializeTestServer = initializeTestServer