Merge pull request #1 from rassokhin-s/add-env-vars

Add support for environment variables
rassokhin-s · May 18, 2017 · e8bd96f · e8bd96f
2 parents 6fb35e1 + 73fa8c7
commit e8bd96f
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -68,6 +68,9 @@ module.exports = {
   memorySize: 128,
   publish: true, // default: false,
   runtime: 'nodejs4.3', // default: 'nodejs4.3',
+  env: {, // optionl
+    [<Environment key/values>]
+  },
   vpc: { // optional
     SecurityGroupIds: [<security group id>, ...],
     SubnetIds: [<subnet id>, ...]

diff --git a/index.js b/index.js
@@ -39,6 +39,7 @@ exports.deploy = function(codePackage, config, callback, logger, lambda) {
     MemorySize: config.memorySize
   };
   if (config.vpc) params.VpcConfig = config.vpc;
+  if (config.env) params.Environment = { Variables: config.env };
   var isPublish = (config.publish === true);
 
   var updateEventSource = function(eventSource, callback) {

diff --git a/test/all.js b/test/all.js
@@ -28,6 +28,9 @@ describe('node aws lambda module', function() {
     timeout: 10,
     memorySize: 128,
     publish: true,
+    env: {
+      MyEvnVar: 'helloworld var',
+    },
     vpc: {
       SecurityGroupIds: ['sg-xxxxxxx1', 'sg-xxxxxxx2'],
       SubnetIds: ['subnet-xxxxxxxx']
@@ -86,6 +89,11 @@ describe('node aws lambda module', function() {
           MemorySize: 128,
           Publish: true,
           Runtime: "nodejs4.3",
+          Environment: {
+            Variables : {
+              MyEvnVar: 'helloworld var'
+            }
+          },
           VpcConfig: {
             SecurityGroupIds: ['sg-xxxxxxx1', 'sg-xxxxxxx2'],
             SubnetIds: ['subnet-xxxxxxxx']
@@ -142,6 +150,11 @@ describe('node aws lambda module', function() {
           MemorySize: 128,
           Publish: false,
           Runtime: "nodejs4.3",
+          Environment: {
+            Variables: {
+              MyEvnVar: 'helloworld var',
+            },
+          },
           VpcConfig: {
             SecurityGroupIds: ['sg-xxxxxxx3'],
             SubnetIds: ['subnet-xxxxxxx1', 'subnet-xxxxxxx2']

diff --git a/test/fake-lambda-service.js b/test/fake-lambda-service.js
@@ -50,7 +50,7 @@ module.exports = function() {
     createFunction: function(params, callback) {
       validateParams(params,
                      ['FunctionName', 'Code', 'Handler', 'Role', 'Runtime'],
-                     ['Description', 'MemorySize', 'Timeout', 'Publish', 'VpcConfig'], 'createFunction')
+                     ['Description', 'MemorySize', 'Timeout', 'Publish', 'VpcConfig', 'Environment'], 'createFunction')
 
       var name = params.FunctionName;
       var code = params.Code;
@@ -154,7 +154,7 @@ module.exports = function() {
     updateFunctionConfiguration: function(params, callback) {
       validateParams(params,
                      ['FunctionName'],
-                     ['Description', 'Handler', 'MemorySize', 'Role', 'Timeout', 'VpcConfig'],
+                     ['Description', 'Handler', 'MemorySize', 'Role', 'Timeout', 'VpcConfig', 'Environment'],
                      'updateFunctionConfiguration')
 
       var fun = getFun(params.FunctionName);