From 7a1f4ece8e44ca7ef63c9fe40e1312c2c42dcd9f Mon Sep 17 00:00:00 2001
From: Daniel Schep <dschep@gmail.com>
Date: Wed, 5 Dec 2018 11:21:42 -0500
Subject: [PATCH] fix tests for invoke lcal env vars fix

---
 lib/plugins/aws/invokeLocal/index.test.js | 24 +++++++++++++++--------
 1 file changed, 16 insertions(+), 8 deletions(-)

diff --git a/lib/plugins/aws/invokeLocal/index.test.js b/lib/plugins/aws/invokeLocal/index.test.js
index 40cacc164..88fce29dc 100644
--- a/lib/plugins/aws/invokeLocal/index.test.js
+++ b/lib/plugins/aws/invokeLocal/index.test.js
@@ -44,23 +44,31 @@ describe('AwsInvokeLocal', () => {
     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();
+    it('should run invoke:local:invoke promise chain in order', () => {
       const invokeLocalStub = sinon
         .stub(awsInvokeLocal, 'invokeLocal').resolves();
 
 
       return awsInvokeLocal.hooks['invoke:local:invoke']().then(() => {
-        expect(validateStub.calledOnce).to.be.equal(true);
+        expect(invokeLocalStub.callCount).to.be.equal(1);
+
+        awsInvokeLocal.invokeLocal.restore();
+      });
+    });
+
+    it('should run before:invoke:local:loadEnvVars promise chain in order', () => {
+      const validateStub = sinon
+        .stub(awsInvokeLocal, 'extendedValidate').resolves();
+      const loadEnvVarsStub = sinon
+        .stub(awsInvokeLocal, 'loadEnvVars').resolves();
+
+
+      return awsInvokeLocal.hooks['before:invoke:local:loadEnvVars']().then(() => {
+        expect(validateStub.callCount).to.be.equal(1);
         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();
       });
     });