From 9727195ac731d1a47461b3f757c8d3c1069478d1 Mon Sep 17 00:00:00 2001
From: Eetu Tuomala <eetu.tuomala@sc5.io>
Date: Sun, 6 Nov 2016 22:23:14 +0200
Subject: [PATCH] changes tests to return promise

---
 lib/plugins/install/install.test.js | 16 ++++++----------
 1 file changed, 6 insertions(+), 10 deletions(-)

diff --git a/lib/plugins/install/install.test.js b/lib/plugins/install/install.test.js
index 94193188c..a642bb063 100644
--- a/lib/plugins/install/install.test.js
+++ b/lib/plugins/install/install.test.js
@@ -42,7 +42,7 @@ describe('Install', () => {
       });
     });
 
-    it('should set new service in serverless.yml and name in package.json', (done) => {
+    it('should set new service in serverless.yml and name in package.json', () => {
       const defaultServiceYml =
         'service: service-name\n\nprovider:\n  name: aws\n';
       const newServiceYml =
@@ -58,18 +58,16 @@ describe('Install', () => {
       serverless.utils.writeFileSync(packageFile, { name: defaultServiceName });
       fs.writeFileSync(serviceFile, defaultServiceYml);
 
-      install.renameService(newServiceName, servicePath)
+      return install.renameService(newServiceName, servicePath)
         .then(() => {
           const serviceYml = fs.readFileSync(serviceFile, 'utf-8');
           const packageJson = serverless.utils.readFileSync(packageFile);
           expect(serviceYml).to.equal(newServiceYml);
           expect(packageJson.name).to.equal(newServiceName);
-          done();
-        })
-        .catch(done);
+        });
     });
 
-    it('should set new service in commented serverless.yml and name in package.json', (done) => {
+    it('should set new service in commented serverless.yml and name in package.json', () => {
       const defaultServiceYml =
         '# comment\nservice: service-name #comment\n\nprovider:\n  name: aws\n# comment';
       const newServiceYml =
@@ -85,15 +83,13 @@ describe('Install', () => {
       serverless.utils.writeFileSync(packageFile, { name: defaultServiceName });
       fs.writeFileSync(serviceFile, defaultServiceYml);
 
-      install.renameService(newServiceName, servicePath)
+      return install.renameService(newServiceName, servicePath)
         .then(() => {
           const serviceYml = fs.readFileSync(serviceFile, 'utf-8');
           const packageJson = serverless.utils.readFileSync(packageFile);
           expect(serviceYml).to.equal(newServiceYml);
           expect(packageJson.name).to.equal(newServiceName);
-          done();
-        })
-        .catch(done);
+        });
     });
   });