From 81d2a78aabf0824ed963dfe87d161df9a09701c4 Mon Sep 17 00:00:00 2001
From: To-om <toom@thehive-project.org>
Date: Fri, 29 Jun 2018 10:49:19 +0200
Subject: [PATCH] Fix error when job is not found

---
 app/org/thp/cortex/services/JobSrv.scala | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/app/org/thp/cortex/services/JobSrv.scala b/app/org/thp/cortex/services/JobSrv.scala
index 3cf49a8f2..c22419e02 100644
--- a/app/org/thp/cortex/services/JobSrv.scala
+++ b/app/org/thp/cortex/services/JobSrv.scala
@@ -150,7 +150,11 @@ class JobSrv(
       findForOrganization(organizationId, withId(jobId), Some("0-1"), Nil)
     }
       ._1
-      .runWith(Sink.head)
+      .runWith(Sink.headOption)
+      .flatMap {
+        case Some(j) ⇒ Future.successful(j)
+        case None    ⇒ Future.failed(NotFoundError(s"job $jobId not found"))
+      }
   }
 
   def delete(job: Job)(implicit authContext: AuthContext): Future[Job] = deleteSrv(job)