serverless/test/mocha-patch.js
Max Marze e0d6a8acbb
fix: remove bluebird and set concurrency limits for packaging (#12658)
* fix: remove bluebird from zip-service

* fix: remove bluebird and set concurrency limits for packaging
2024-07-02 14:26:28 -04:00

79 lines
2.6 KiB
JavaScript

'use strict'
const fs = require('fs')
const Module = require('module')
// Temporary patch to help tackle peekaboo error, by revelaing fuller stack trace for "fs" errors
// https://github.com/serverless/serverless/runs/1740873363
const patchPromised = (name) => {
const original = fs.promises[name]
fs.promises[name] = Object.defineProperties(
function (...args) {
const stack = new Error().stack
return original.apply(this, args).catch((error) => {
error.message += ` (initiated at: ${stack}\n)`
throw error
})
},
{
length: { value: original.length },
name: { value: original.name },
},
)
}
const patchCallback = (name) => {
const original = fs[name]
fs[name] = Object.defineProperties(
function (...args) {
const stack = new Error().stack
const callback = args[args.length - 1]
original.call(this, ...args.slice(0, -1), (error, ...result) => {
if (error) error.message += ` (initiated at: ${stack}\n)`
return callback.call(this, error, ...result)
})
},
{
length: { value: original.length },
name: { value: original.name },
},
)
}
patchPromised('mkdir')
patchPromised('open')
patchPromised('readFile')
patchCallback('mkdir')
patchCallback('open')
patchCallback('readFile')
const ensureArtifact = require('../lib/utils/ensure-artifact')
const localServerlessPathModulePath = require.resolve(
'../lib/cli/local-serverless-path',
)
const resolveInput = require('../lib/cli/resolve-input')
const { runnerEmitter } = require('@serverless/test/setup/patch')
runnerEmitter.on('runner', (runner) => {
runner.on('suite end', (suite) => {
if (!suite.parent || !suite.parent.root) return
resolveInput.clear()
// Ensure to reset cache for local serverless installation resolution
// Leaking it across test files may introduce wrong assumptions when result is used for testing
if (require.cache[localServerlessPathModulePath]) {
// As we rely on native require.resolve, we need to ensure that it's cache related to
// tmp homedir is cleared
const homedir = require('os').homedir()
for (const key of Object.keys(Module._pathCache)) {
if (key.includes(homedir)) delete Module._pathCache[key]
}
delete require.cache[localServerlessPathModulePath]
}
// Ensure to reset memoization on artifacts generation after each test file run.
// Reason is that homedir is automatically cleaned for each test,
// therefore eventually built custom resource file is also removed
ensureArtifact.clear()
})
})