mirror of
https://github.com/serverless/serverless.git
synced 2026-01-18 14:58:43 +00:00
105 lines
3.2 KiB
JavaScript
105 lines
3.2 KiB
JavaScript
// Integration tests related utils
|
|
|
|
'use strict';
|
|
|
|
const path = require('path');
|
|
const fse = require('fs-extra');
|
|
const spawn = require('child-process-ext/spawn');
|
|
const resolveAwsEnv = require('@serverless/test/resolve-aws-env');
|
|
const { getServiceName, wait } = require('./misc');
|
|
const { readYamlFile, writeYamlFile } = require('./fs');
|
|
|
|
const serverlessExec = require('../serverless-binary');
|
|
|
|
const env = resolveAwsEnv();
|
|
|
|
async function createTestService(
|
|
tmpDir,
|
|
options = {
|
|
// Either templateName or templateDir have to be provided
|
|
templateName: null, // Generic template to use (e.g. 'aws-nodejs')
|
|
templateDir: null, // Path to custom pre-prepared service template
|
|
filesToAdd: [], // Array of additional files to add to the service directory
|
|
serverlessConfigHook: null, // Eventual hook that allows to customize serverless config
|
|
}
|
|
) {
|
|
const serviceName = getServiceName();
|
|
|
|
fse.mkdirsSync(tmpDir);
|
|
|
|
if (options.templateName) {
|
|
// create a new Serverless service
|
|
await spawn(serverlessExec, ['create', '--template', options.templateName], {
|
|
cwd: tmpDir,
|
|
env,
|
|
});
|
|
} else if (options.templateDir) {
|
|
fse.copySync(options.templateDir, tmpDir, { clobber: true, preserveTimestamps: true });
|
|
} else {
|
|
throw new Error("Either 'templateName' or 'templateDir' options have to be provided");
|
|
}
|
|
|
|
if (options.filesToAdd && options.filesToAdd.length) {
|
|
options.filesToAdd.forEach(filePath => {
|
|
fse.copySync(filePath, tmpDir, { preserveTimestamps: true });
|
|
});
|
|
}
|
|
|
|
const serverlessFilePath = path.join(tmpDir, 'serverless.yml');
|
|
const serverlessConfig = readYamlFile(serverlessFilePath);
|
|
// Ensure unique service name
|
|
serverlessConfig.service = serviceName;
|
|
if (options.serverlessConfigHook) options.serverlessConfigHook(serverlessConfig);
|
|
writeYamlFile(serverlessFilePath, serverlessConfig);
|
|
|
|
process.env.TOPIC_1 = `${serviceName}-1`;
|
|
process.env.TOPIC_2 = `${serviceName}-1`;
|
|
process.env.BUCKET_1 = `${serviceName}-1`;
|
|
process.env.BUCKET_2 = `${serviceName}-2`;
|
|
process.env.COGNITO_USER_POOL_1 = `${serviceName}-1`;
|
|
process.env.COGNITO_USER_POOL_2 = `${serviceName}-2`;
|
|
|
|
return serverlessConfig;
|
|
}
|
|
|
|
async function deployService(cwd) {
|
|
return spawn(serverlessExec, ['deploy'], { cwd, env });
|
|
}
|
|
|
|
async function removeService(cwd) {
|
|
return spawn(serverlessExec, ['remove'], { cwd, env });
|
|
}
|
|
|
|
async function getFunctionLogs(cwd, functionName) {
|
|
let logs;
|
|
try {
|
|
({ stdoutBuffer: logs } = await spawn(
|
|
serverlessExec,
|
|
['logs', '--function', functionName, '--noGreeting', 'true'],
|
|
{
|
|
cwd,
|
|
env,
|
|
}
|
|
));
|
|
} catch (_) {
|
|
// Attempting to read logs before first invocation will will result in a "No existing streams for the function" error
|
|
return null;
|
|
}
|
|
return String(logs);
|
|
}
|
|
|
|
async function waitForFunctionLogs(cwd, functionName, startMarker, endMarker) {
|
|
await wait(2000);
|
|
const logs = await getFunctionLogs(cwd, functionName);
|
|
if (logs && logs.includes(startMarker) && logs.includes(endMarker)) return logs;
|
|
return waitForFunctionLogs(cwd, functionName, startMarker, endMarker);
|
|
}
|
|
|
|
module.exports = {
|
|
createTestService,
|
|
deployService,
|
|
removeService,
|
|
waitForFunctionLogs,
|
|
env,
|
|
};
|