2017-07-17 10:51:20 +02:00

425 lines
15 KiB
JavaScript

'use strict';
const expect = require('chai').expect;
const sinon = require('sinon');
const path = require('path');
const AwsInvokeLocal = require('./index');
const AwsProvider = require('../provider/awsProvider');
const Serverless = require('../../../Serverless');
const CLI = require('../../../classes/CLI');
const testUtils = require('../../../../tests/utils');
describe('AwsInvokeLocal', () => {
const serverless = new Serverless();
serverless.setProvider('aws', new AwsProvider(serverless));
const options = {
stage: 'dev',
region: 'us-east-1',
function: 'first',
};
const awsInvokeLocal = new AwsInvokeLocal(serverless, options);
describe('#constructor()', () => {
it('should have hooks', () => expect(awsInvokeLocal.hooks).to.be.not.empty);
it('should set the provider variable to an instance of AwsProvider', () =>
expect(awsInvokeLocal.provider).to.be.instanceof(AwsProvider));
it('should run promise chain in order', () => {
const validateStub = sinon
.stub(awsInvokeLocal, 'extendedValidate').resolves();
const loadEnvVarsStub = sinon
.stub(awsInvokeLocal, 'loadEnvVars').resolves();
const invokeLocalStub = sinon
.stub(awsInvokeLocal, 'invokeLocal').resolves();
return awsInvokeLocal.hooks['invoke:local:invoke']().then(() => {
expect(validateStub.calledOnce).to.be.equal(true);
expect(loadEnvVarsStub.calledAfter(validateStub)).to.be.equal(true);
expect(invokeLocalStub.calledAfter(loadEnvVarsStub)).to.be.equal(true);
awsInvokeLocal.extendedValidate.restore();
awsInvokeLocal.loadEnvVars.restore();
awsInvokeLocal.invokeLocal.restore();
});
});
it('should set an empty options object if no options are given', () => {
const awsInvokeWithEmptyOptions = new AwsInvokeLocal(serverless);
expect(awsInvokeWithEmptyOptions.options).to.deep.equal({});
});
});
describe('#extendedValidate()', () => {
beforeEach(() => {
serverless.config.servicePath = true;
serverless.service.environment = {
vars: {},
stages: {
dev: {
vars: {},
regions: {
'us-east-1': {
vars: {},
},
},
},
},
};
serverless.service.functions = {
first: {
handler: true,
},
};
awsInvokeLocal.options.data = null;
awsInvokeLocal.options.path = false;
});
it('should not throw error when there are no input data', () => {
awsInvokeLocal.options.data = undefined;
return awsInvokeLocal.extendedValidate().then(() => {
expect(awsInvokeLocal.options.data).to.equal('');
});
});
it('it should throw error if function is not provided', () => {
serverless.service.functions = null;
expect(() => awsInvokeLocal.extendedValidate()).to.throw(Error);
});
it('should keep data if it is a simple string', () => {
awsInvokeLocal.options.data = 'simple-string';
return awsInvokeLocal.extendedValidate().then(() => {
expect(awsInvokeLocal.options.data).to.equal('simple-string');
});
});
it('should parse data if it is a json string', () => {
awsInvokeLocal.options.data = '{"key": "value"}';
return awsInvokeLocal.extendedValidate().then(() => {
expect(awsInvokeLocal.options.data).to.deep.equal({ key: 'value' });
});
});
it('it should parse file if relative file path is provided', () => {
serverless.config.servicePath = testUtils.getTmpDirPath();
const data = {
testProp: 'testValue',
};
serverless.utils.writeFileSync(path
.join(serverless.config.servicePath, 'data.json'), JSON.stringify(data));
awsInvokeLocal.options.path = 'data.json';
return awsInvokeLocal.extendedValidate().then(() => {
expect(awsInvokeLocal.options.data).to.deep.equal(data);
});
});
it('it should parse file if absolute file path is provided', () => {
serverless.config.servicePath = testUtils.getTmpDirPath();
const data = {
event: {
testProp: 'testValue',
},
};
const dataFile = path.join(serverless.config.servicePath, 'data.json');
serverless.utils.writeFileSync(dataFile, JSON.stringify(data));
awsInvokeLocal.options.path = dataFile;
return awsInvokeLocal.extendedValidate().then(() => {
expect(awsInvokeLocal.options.data).to.deep.equal(data);
});
});
it('it should parse a yaml file if file path is provided', () => {
serverless.config.servicePath = testUtils.getTmpDirPath();
const yamlContent = 'event: data';
serverless.utils.writeFileSync(path
.join(serverless.config.servicePath, 'data.yml'), yamlContent);
awsInvokeLocal.options.path = 'data.yml';
return awsInvokeLocal.extendedValidate().then(() => {
expect(awsInvokeLocal.options.data).to.deep.equal({ event: 'data' });
});
});
it('it should require a js file if file path is provided', () => {
serverless.config.servicePath = testUtils.getTmpDirPath();
const jsContent = [
'module.exports = {',
' headers: { "Content-Type" : "application/json" },',
' body: JSON.stringify([100, 200]),',
'}',
].join('\n');
serverless.utils.writeFileSync(path
.join(serverless.config.servicePath, 'data.js'), jsContent);
awsInvokeLocal.options.path = 'data.js';
return awsInvokeLocal.extendedValidate().then(() => {
expect(awsInvokeLocal.options.data).to.deep.equal({
headers: { 'Content-Type': 'application/json' },
body: '[100,200]',
});
});
});
it('it should throw error if service path is not set', () => {
serverless.config.servicePath = false;
expect(() => awsInvokeLocal.extendedValidate()).to.throw(Error);
});
it('it should reject error if file path does not exist', () => {
serverless.config.servicePath = testUtils.getTmpDirPath();
awsInvokeLocal.options.path = 'some/path';
return awsInvokeLocal.extendedValidate().catch((err) => {
expect(err).to.be.instanceOf(Error);
});
});
it('should resolve if path is not given', (done) => {
awsInvokeLocal.options.path = false;
awsInvokeLocal.extendedValidate().then(() => done());
});
});
describe('#loadEnvVars()', () => {
beforeEach(() => {
serverless.config.servicePath = true;
serverless.service.provider = {
environment: {
providerVar: 'providerValue',
},
};
awsInvokeLocal.options = {
region: 'us-east-1',
functionObj: {
name: 'serviceName-dev-hello',
environment: {
functionVar: 'functionValue',
},
},
};
});
it('it should load provider env vars', () => awsInvokeLocal
.loadEnvVars().then(() => {
expect(process.env.providerVar).to.be.equal('providerValue');
})
);
it('it should load function env vars', () => awsInvokeLocal
.loadEnvVars().then(() => {
expect(process.env.functionVar).to.be.equal('functionValue');
})
);
it('it should load default lambda env vars', () => awsInvokeLocal
.loadEnvVars().then(() => {
expect(process.env.PATH)
.to.equal('/usr/local/lib64/node-v4.3.x/bin:/usr/local/bin:/usr/bin/:/bin');
expect(process.env.LANG).to.equal('en_US.UTF-8');
expect(process.env.LD_LIBRARY_PATH)
.to.equal('/usr/local/lib64/node-v4.3.x/lib:/lib64:/usr/lib64:/var/runtime:/var/runtime/lib:/var/task:/var/task/lib'); // eslint-disable-line max-len
expect(process.env.LAMBDA_TASK_ROOT).to.equal('/var/task');
expect(process.env.LAMBDA_RUNTIME_DIR).to.equal('/var/runtime');
expect(process.env.AWS_REGION).to.equal('us-east-1');
expect(process.env.AWS_DEFAULT_REGION).to.equal('us-east-1');
expect(process.env.AWS_LAMBDA_LOG_GROUP_NAME).to.equal('/aws/lambda/serviceName-dev-hello');
expect(process.env.AWS_LAMBDA_LOG_STREAM_NAME)
.to.equal('2016/12/02/[$LATEST]f77ff5e4026c45bda9a9ebcec6bc9cad');
expect(process.env.AWS_LAMBDA_FUNCTION_NAME).to.equal('serviceName-dev-hello');
expect(process.env.AWS_LAMBDA_FUNCTION_MEMORY_SIZE).to.equal('1024');
expect(process.env.AWS_LAMBDA_FUNCTION_VERSION).to.equal('$LATEST');
expect(process.env.NODE_PATH).to.equal('/var/runtime:/var/task:/var/runtime/node_modules');
})
);
it('should fallback to service provider configuration when options are not available', () => {
awsInvokeLocal.options.region = null;
awsInvokeLocal.serverless.service.provider.region = 'us-west-1';
return awsInvokeLocal.loadEnvVars().then(() => {
expect(process.env.AWS_REGION).to.equal('us-west-1');
expect(process.env.AWS_DEFAULT_REGION).to.equal('us-west-1');
});
});
it('it should overwrite provider env vars', () => {
awsInvokeLocal.options.functionObj.environment.providerVar = 'providerValueOverwritten';
return awsInvokeLocal.loadEnvVars().then(() => {
expect(process.env.providerVar).to.be.equal('providerValueOverwritten');
});
});
});
describe('#invokeLocal()', () => {
let invokeLocalNodeJsStub;
let invokeLocalPythonStub;
beforeEach(() => {
invokeLocalNodeJsStub =
sinon.stub(awsInvokeLocal, 'invokeLocalNodeJs').resolves();
invokeLocalPythonStub =
sinon.stub(awsInvokeLocal, 'invokeLocalPython').resolves();
awsInvokeLocal.serverless.service.service = 'new-service';
awsInvokeLocal.options = {
stage: 'dev',
function: 'first',
functionObj: {
handler: 'handler.hello',
name: 'hello',
},
data: {},
};
});
afterEach(() => {
invokeLocalNodeJsStub.restore();
invokeLocalPythonStub.restore();
});
it('should call invokeLocalNodeJs when no runtime is set', () => awsInvokeLocal.invokeLocal()
.then(() => {
expect(invokeLocalNodeJsStub.calledOnce).to.be.equal(true);
expect(invokeLocalNodeJsStub.calledWithExactly(
'handler',
'hello',
{}
)).to.be.equal(true);
awsInvokeLocal.invokeLocalNodeJs.restore();
})
);
it('should call invokeLocalNodeJs for any node.js runtime version', () => {
awsInvokeLocal.options.functionObj.runtime = 'nodejs6.10';
awsInvokeLocal.invokeLocal()
.then(() => {
expect(invokeLocalNodeJsStub.calledOnce).to.be.equal(true);
expect(invokeLocalNodeJsStub.calledWithExactly(
'handler',
'hello',
{}
)).to.be.equal(true);
awsInvokeLocal.invokeLocalNodeJs.restore();
});
});
it('should call invokeLocalPython when python2.7 runtime is set', () => {
awsInvokeLocal.options.functionObj.runtime = 'python2.7';
awsInvokeLocal.invokeLocal()
.then(() => {
expect(invokeLocalPythonStub.calledOnce).to.be.equal(true);
expect(invokeLocalPythonStub.calledWithExactly(
'python2.7',
'handler',
'hello',
{}
)).to.be.equal(true);
awsInvokeLocal.invokeLocalPython.restore();
});
delete awsInvokeLocal.options.functionObj.runtime;
});
it('throw error when using runtime other than Node.js or Python', () => {
awsInvokeLocal.options.functionObj.runtime = 'java8';
expect(() => awsInvokeLocal.invokeLocal()).to.throw(Error);
delete awsInvokeLocal.options.functionObj.runtime;
});
});
describe('#invokeLocalNodeJs', () => {
beforeEach(() => {
awsInvokeLocal.options = {
functionObj: {
name: '',
},
};
serverless.cli = new CLI(serverless);
sinon.stub(serverless.cli, 'consoleLog');
});
afterEach(() => {
serverless.cli.consoleLog.restore();
});
describe('with done method', () => {
it('should exit with error exit code', () => {
awsInvokeLocal.serverless.config.servicePath = __dirname;
awsInvokeLocal.invokeLocalNodeJs('fixture/handlerWithSuccess', 'withErrorByDone');
expect(process.exitCode).to.be.equal(1);
});
it('should succeed if succeed', () => {
awsInvokeLocal.serverless.config.servicePath = __dirname;
awsInvokeLocal.invokeLocalNodeJs('fixture/handlerWithSuccess', 'withMessageByDone');
expect(serverless.cli.consoleLog.lastCall.args[0]).to.contain('"Succeed"');
});
});
describe('with Lambda Proxy with application/json response', () => {
it('should succeed if succeed', () => {
awsInvokeLocal.serverless.config.servicePath = __dirname;
awsInvokeLocal.invokeLocalNodeJs('fixture/handlerWithSuccess', 'withMessageByLambdaProxy');
expect(serverless.cli.consoleLog.lastCall.args[0]).to.contain('{\n "statusCode": 200,\n "headers": {\n "Content-Type": "application/json"\n },\n "body": {\n "result": true,\n "message": "Whatever"\n }\n}'); // eslint-disable-line
});
});
describe('with extraServicePath', () => {
it('should succeed if succeed', () => {
awsInvokeLocal.serverless.config.servicePath = __dirname;
awsInvokeLocal.options.extraServicePath = 'fixture';
awsInvokeLocal.invokeLocalNodeJs('handlerWithSuccess', 'withMessageByLambdaProxy');
expect(serverless.cli.consoleLog.lastCall.args[0]).to.contain('{\n "statusCode": 200,\n "headers": {\n "Content-Type": "application/json"\n },\n "body": {\n "result": true,\n "message": "Whatever"\n }\n}'); // eslint-disable-line
});
});
it('should exit with error exit code', () => {
awsInvokeLocal.serverless.config.servicePath = __dirname;
awsInvokeLocal.invokeLocalNodeJs('fixture/handlerWithError', 'withError');
expect(process.exitCode).to.be.equal(1);
});
it('should log Error instance when called back', () => {
awsInvokeLocal.serverless.config.servicePath = __dirname;
awsInvokeLocal.invokeLocalNodeJs('fixture/handlerWithError', 'withError');
expect(serverless.cli.consoleLog.lastCall.args[0]).to.contain('"errorMessage": "failed"');
expect(serverless.cli.consoleLog.lastCall.args[0]).to.contain('"errorType": "Error"');
});
it('should log error when called back', () => {
awsInvokeLocal.serverless.config.servicePath = __dirname;
awsInvokeLocal.invokeLocalNodeJs('fixture/handlerWithError', 'withMessage');
expect(serverless.cli.consoleLog.lastCall.args[0]).to.contain('"errorMessage": "failed"');
});
});
});