From 20eaf5178facedd1486c2880121bc555272ad4a3 Mon Sep 17 00:00:00 2001
From: Dao Lam <daolam112@gmail.com>
Date: Fri, 8 Mar 2019 13:25:27 -0800
Subject: [PATCH] fix: promise request (#50)

---
 index.js           | 14 ++++++++------
 package.json       |  2 +-
 test/index.test.js | 29 +++++++++++++++++++++--------
 3 files changed, 30 insertions(+), 15 deletions(-)

diff --git a/index.js b/index.js
index 5e2fdca..5fbe63a 100644
--- a/index.js
+++ b/index.js
@@ -206,12 +206,14 @@ class ExecutorQueue extends Executor {
             options.body.parentEventId = eventId;
         }
 
-        return req(options, (err, response) => {
-            if (!err && response.statusCode === 201) {
-                return Promise.resolve(response);
-            }
+        return new Promise((resolve, reject) => {
+            req(options, (err, response) => {
+                if (!err && response.statusCode === 201) {
+                    return resolve(response);
+                }
 
-            return Promise.reject(err);
+                return reject(err);
+            });
         });
     }
 
@@ -268,7 +270,7 @@ class ExecutorQueue extends Executor {
         }
 
         if (triggerBuild) {
-            await this.postBuildEvent(config)
+            return this.postBuildEvent(config)
                 .catch((err) => {
                     winston.error(`failed to post build event for job ${job.id}: ${err}`);
 
diff --git a/package.json b/package.json
index b0cedc7..68f517e 100644
--- a/package.json
+++ b/package.json
@@ -47,7 +47,7 @@
     "ioredis": "^3.2.2",
     "node-resque": "^4.0.9",
     "request": "^2.88.0",
-    "screwdriver-data-schema": "^18.43.3",
+    "screwdriver-data-schema": "^18.44.1",
     "screwdriver-executor-base": "^7.0.0",
     "string-hash": "^1.1.3",
     "winston": "^2.4.4"
diff --git a/test/index.test.js b/test/index.test.js
index 9bf27ef..6f87d67 100644
--- a/test/index.test.js
+++ b/test/index.test.js
@@ -52,6 +52,7 @@ describe('index test', () => {
     let pipelineMock;
     let buildMock;
     let pipelineFactoryMock;
+    let fakeResponse;
 
     before(() => {
         mockery.enable({
@@ -107,7 +108,14 @@ describe('index test', () => {
         freezeWindowsMock = {
             timeOutOfWindows: (windows, date) => date
         };
-        reqMock = sinon.stub().resolves();
+
+        fakeResponse = {
+            statusCode: 201,
+            body: {
+                id: '12345'
+            }
+        };
+        reqMock = sinon.stub();
         pipelineMock = {
             getFirstAdmin: sinon.stub().resolves(testAdmin)
         };
@@ -212,6 +220,9 @@ describe('index test', () => {
     });
 
     describe('_startPeriodic', () => {
+        beforeEach(() => {
+            reqMock.yieldsAsync(null, fakeResponse, fakeResponse.body);
+        });
         it('rejects if it can\'t establish a connection', function () {
             queueMock.connect.yieldsAsync(new Error('couldn\'t connect'));
 
@@ -230,7 +241,7 @@ describe('index test', () => {
             });
         });
 
-        it('enqueues a new delayed job in the queue', () => {
+        it('enqueues a new delayed job in the queue', () =>
             executor.startPeriodic(testDelayedConfig).then(() => {
                 assert.calledOnce(queueMock.connect);
                 assert.calledWith(redisMock.hset, 'periodicBuildConfigs', testJob.id,
@@ -240,12 +251,12 @@ describe('index test', () => {
                 assert.calledWith(queueMock.enqueueAt, 1500000, 'periodicBuilds', 'startDelayed', [{
                     jobId: testJob.id
                 }]);
-            });
-        });
+            }));
 
         it('stops and reEnqueues an existing job if isUpdate flag is passed', () => {
             testDelayedConfig.isUpdate = true;
-            executor.startPeriodic(testDelayedConfig).then(() => {
+
+            return executor.startPeriodic(testDelayedConfig).then(() => {
                 assert.calledTwice(queueMock.connect);
                 assert.calledWith(redisMock.hset, 'periodicBuildConfigs', testJob.id,
                     JSON.stringify(testDelayedConfig));
@@ -263,7 +274,8 @@ describe('index test', () => {
             testDelayedConfig.isUpdate = true;
             testDelayedConfig.job.state = 'DISABLED';
             testDelayedConfig.job.archived = false;
-            executor.startPeriodic(testDelayedConfig).then(() => {
+
+            return executor.startPeriodic(testDelayedConfig).then(() => {
                 assert.calledOnce(queueMock.connect);
                 assert.notCalled(redisMock.hset);
                 assert.notCalled(queueMock.enqueueAt);
@@ -278,7 +290,8 @@ describe('index test', () => {
             testDelayedConfig.isUpdate = true;
             testDelayedConfig.job.state = 'ENABLED';
             testDelayedConfig.job.archived = true;
-            executor.startPeriodic(testDelayedConfig).then(() => {
+
+            return executor.startPeriodic(testDelayedConfig).then(() => {
                 assert.calledOnce(queueMock.connect);
                 assert.notCalled(redisMock.hset);
                 assert.notCalled(queueMock.enqueueAt);
@@ -311,7 +324,7 @@ describe('index test', () => {
 
             executor.tokenGen = tokenGen;
 
-            executor.startPeriodic(testDelayedConfig).then(() => {
+            return executor.startPeriodic(testDelayedConfig).then(() => {
                 assert.calledOnce(queueMock.connect);
                 assert.notCalled(redisMock.hset);
                 assert.notCalled(queueMock.enqueueAt);