From 7191a2bed014619775468923666aa33ae92073bd Mon Sep 17 00:00:00 2001
From: Joe Fleming <joe.fleming@gmail.com>
Date: Mon, 2 May 2016 12:45:46 -0700
Subject: [PATCH] add tests for handling job failures

---
 test/src/worker.js | 44 ++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 42 insertions(+), 2 deletions(-)

diff --git a/test/src/worker.js b/test/src/worker.js
index 1ed1bd1..a67cb77 100644
--- a/test/src/worker.js
+++ b/test/src/worker.js
@@ -129,7 +129,7 @@ describe('Worker class', function () {
       expect(query).to.have.property('version', job._version);
     });
 
-    it('should increment the attempts', function () {
+    it('should increment the job attempts', function () {
       worker._claimJob(job);
       const doc = updateSpy.firstCall.args[0].body.doc;
       expect(doc).to.have.property('attempts', job._source.attempts + 1);
@@ -141,7 +141,7 @@ describe('Worker class', function () {
       expect(doc).to.have.property('status', JOB_STATUS_PROCESSING);
     });
 
-    it('should set expiration time', function () {
+    it('should set job expiration time', function () {
       worker._claimJob(job);
       const doc = updateSpy.firstCall.args[0].body.doc;
       const expiration = anchorMoment.add(defaults.timeout).toISOString();
@@ -272,6 +272,46 @@ describe('Worker class', function () {
         }
       });
     });
+
+    it('should reject on job errors', function (done) {
+      const workerFn = function (jobPayload, cb) {
+        cb(new Error('test error'));
+      };
+      const worker = new Worker(mockQueue, 'test', workerFn);
+
+      worker._performJob(job)
+      .then(() => done(new Error('should not resolve')))
+      .catch((err) => {
+        expect(err.message).to.equal('test error');
+        done();
+      });
+    });
+
+    it('should append erorr output to job', function (done) {
+      const workerFn = function (jobPayload, cb) {
+        cb(new Error('test error'));
+      };
+      const worker = new Worker(mockQueue, 'test', workerFn);
+
+      worker._performJob(job)
+      .then(() => done(new Error('should not resolve')))
+      .catch((err) => {
+        try {
+          sinon.assert.calledOnce(updateSpy);
+          const query = updateSpy.firstCall.args[0];
+          expect(query).to.have.property('index', job._index);
+          expect(query).to.have.property('type', job._type);
+          expect(query).to.have.property('id', job._id);
+          expect(query).to.have.property('version', job._version);
+          expect(query.body.doc).to.have.property('output');
+          expect(query.body.doc.output).to.have.property('content_type', false);
+          expect(query.body.doc.output).to.have.property('content', err.toString());
+          done();
+        } catch (e) {
+          done(e);
+        }
+      });
+    });
   });
 
 });