From f80b2f02bedfb5b9180c7e595e45a6fb07ea6650 Mon Sep 17 00:00:00 2001
From: Philipp Muens <raute1337@gmx.de>
Date: Tue, 11 Apr 2017 14:04:13 +0200
Subject: [PATCH] Fix tests for "saveCompiledTemplate" function

---
 lib/plugins/aws/package/lib/saveCompiledTemplate.test.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/lib/plugins/aws/package/lib/saveCompiledTemplate.test.js b/lib/plugins/aws/package/lib/saveCompiledTemplate.test.js
index e66bb8b4a..ce039f492 100644
--- a/lib/plugins/aws/package/lib/saveCompiledTemplate.test.js
+++ b/lib/plugins/aws/package/lib/saveCompiledTemplate.test.js
@@ -33,6 +33,7 @@ describe('#saveCompiledTemplate()', () => {
   });
 
   afterEach(() => {
+    saveCompiledTemplate.provider.naming.getCompiledTemplateFileName.restore();
     saveCompiledTemplate.serverless.utils.writeFileSync.restore();
   });
 
@@ -45,7 +46,7 @@ describe('#saveCompiledTemplate()', () => {
 
     return saveCompiledTemplate.saveCompiledTemplate().then(() => {
       expect(getCompiledTemplateFileNameStub.calledOnce).to.equal(true);
-      expect(writeFileSyncStub.calledWithExactly(filePath, 'compiled content'));
+      expect(writeFileSyncStub.calledWithExactly(filePath, 'compiled content')).to.equal(true);
     });
   });
 });