fix issues/11

Author: vkkis93

build.js

@@ -84,9 +84,15 @@ module.exports = {
     _states(currentState, currentStateName) {
         switch (currentState.Type) {
         case 'Task': // just push task to general array
+            //before each task restore global default env variables
+            process.env = Object.assign({}, this.environmentVariables);
             let f = this.variables[currentStateName];
             f = this.functions[f];
             const {handler, filePath} = this._findFunctionPathAndHandler(f.handler);
+            // if function has additional variables - attach it to function
+            if (f.environment) {
+                process.env = _.extend(process.env, f.environment);
+            }
             return {
                 name: currentStateName,
                 f: () => require(path.join(this.location, filePath))[handler]

index.js

@@ -116,6 +116,7 @@ class StepFunctionsOfflinePlugin {
         this.environment = this.serverless.service.provider.environment;
         process.env.STEP_IS_OFFLINE = true;
         process.env = _.extend(process.env, this.environment);
+        this.environmentVariables = Object.assign({}, process.env); //store global env variables;
         return;
     }
 

package.json

@@ -13,7 +13,7 @@
     "url": "git@github.com:vkkis93/serverless-step-functions-offline.git"
   },
   "scripts": {
-    "test": "nyc ./node_modules/mocha/bin/mocha  ./tests/index.test.js ./tests/enum.test.js ./tests/parse.test.js ./tests/build.test.js",
+    "test": "nyc ./node_modules/mocha/bin/mocha  ./tests/index.test.js ./tests/enum.test.js ./tests/build.test.js",
     "lint": "eslint ./**  --ext .js"
   },
   "dependencies": {

tests/examples/firstLambda/index.js

@@ -6,7 +6,7 @@ exports.handler = (event, context, callback) => {
     console.log('First lambda', event);
     return Promise.resolve()
         .then(() => {
-            return context.succeed({foo: 1, expirydate: '2015-09-04T01:59:00Z', expiryseconds: 22});
+            return context.succeed({foo: 1, expirydate: '2015-09-04T01:59:00Z', expiryseconds: 2});
         });
 };
 

tests/serverless.yml

@@ -33,10 +33,10 @@ stepFunctions:
         States:
             FirstLambda:
               Type: Task
-              Next: FinalState
+              Next: wait_using_seconds
             wait_using_seconds:
               Type: Wait
-              Seconds: 10
+              Seconds: 2
               Next: wait_using_timestamp
             wait_using_timestamp:
               Type: Wait