138 lines
3.5 KiB
JavaScript
138 lines
3.5 KiB
JavaScript
const rowController = require("../../../controllers/row")
|
|
const appController = require("../../../controllers/application")
|
|
const { AppStatus } = require("../../../../db/utils")
|
|
const { BUILTIN_ROLE_IDS } = require("@budibase/backend-core/roles")
|
|
const { TENANT_ID } = require("../../../../tests/utilities/structures")
|
|
const { getAppDB, doInAppContext } = require("@budibase/backend-core/context")
|
|
const env = require("../../../../environment")
|
|
|
|
function Request(appId, params) {
|
|
this.appId = appId
|
|
this.params = params
|
|
this.request = {}
|
|
}
|
|
|
|
function runRequest(appId, controlFunc, request) {
|
|
return doInAppContext(appId, async () => {
|
|
return controlFunc(request)
|
|
})
|
|
}
|
|
|
|
exports.getAllTableRows = async config => {
|
|
const req = new Request(config.appId, { tableId: config.table._id })
|
|
await runRequest(config.appId, rowController.fetch, req)
|
|
return req.body
|
|
}
|
|
|
|
exports.clearAllApps = async (tenantId = TENANT_ID) => {
|
|
const req = { query: { status: AppStatus.DEV }, user: { tenantId } }
|
|
await appController.fetch(req)
|
|
const apps = req.body
|
|
if (!apps || apps.length <= 0) {
|
|
return
|
|
}
|
|
for (let app of apps) {
|
|
const { appId } = app
|
|
const req = new Request(null, { appId })
|
|
await runRequest(appId, appController.delete, req)
|
|
}
|
|
}
|
|
|
|
exports.clearAllAutomations = async config => {
|
|
const automations = await config.getAllAutomations()
|
|
for (let auto of automations) {
|
|
await doInAppContext(config.appId, async () => {
|
|
await config.deleteAutomation(auto)
|
|
})
|
|
}
|
|
}
|
|
|
|
exports.createRequest = (request, method, url, body) => {
|
|
let req
|
|
|
|
if (method === "POST") req = request.post(url).send(body)
|
|
else if (method === "GET") req = request.get(url)
|
|
else if (method === "DELETE") req = request.delete(url)
|
|
else if (method === "PATCH") req = request.patch(url).send(body)
|
|
else if (method === "PUT") req = request.put(url).send(body)
|
|
|
|
return req
|
|
}
|
|
|
|
exports.checkBuilderEndpoint = async ({ config, method, url, body }) => {
|
|
const headers = await config.login({
|
|
userId: "us_fail",
|
|
builder: false,
|
|
prodApp: true,
|
|
})
|
|
await exports
|
|
.createRequest(config.request, method, url, body)
|
|
.set(headers)
|
|
.expect(403)
|
|
}
|
|
|
|
exports.checkPermissionsEndpoint = async ({
|
|
config,
|
|
method,
|
|
url,
|
|
body,
|
|
passRole,
|
|
failRole,
|
|
}) => {
|
|
const passHeader = await config.login({
|
|
roleId: passRole,
|
|
prodApp: true,
|
|
})
|
|
|
|
await exports
|
|
.createRequest(config.request, method, url, body)
|
|
.set(passHeader)
|
|
.expect(200)
|
|
|
|
let failHeader
|
|
if (failRole === BUILTIN_ROLE_IDS.PUBLIC) {
|
|
failHeader = config.publicHeaders({ prodApp: true })
|
|
} else {
|
|
failHeader = await config.login({
|
|
roleId: failRole,
|
|
builder: false,
|
|
prodApp: true,
|
|
})
|
|
}
|
|
|
|
await exports
|
|
.createRequest(config.request, method, url, body)
|
|
.set(failHeader)
|
|
.expect(403)
|
|
}
|
|
|
|
exports.getDB = () => {
|
|
return getAppDB()
|
|
}
|
|
|
|
exports.testAutomation = async (config, automation) => {
|
|
return runRequest(automation.appId, async () => {
|
|
return await config.request
|
|
.post(`/api/automations/${automation._id}/test`)
|
|
.send({
|
|
row: {
|
|
name: "Test",
|
|
description: "TEST",
|
|
},
|
|
})
|
|
.set(config.defaultHeaders())
|
|
.expect("Content-Type", /json/)
|
|
.expect(200)
|
|
})
|
|
}
|
|
|
|
exports.runInProd = async func => {
|
|
const nodeEnv = env.NODE_ENV
|
|
const workerId = env.JEST_WORKER_ID
|
|
env._set("NODE_ENV", "PRODUCTION")
|
|
env._set("JEST_WORKER_ID", null)
|
|
await func()
|
|
env._set("NODE_ENV", nodeEnv)
|
|
env._set("JEST_WORKER_ID", workerId)
|
|
}
|