feat(task): Add wait for taskID (#439)

algolia · Jan 3, 2018 · 6953bab · 6953bab
1 parent 126b7ac
commit 6953bab
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/src/main/scala/algolia/definitions/WaitForTaskDefinition.scala b/src/main/scala/algolia/definitions/WaitForTaskDefinition.scala
@@ -62,6 +62,9 @@ trait WaitForTaskDsl {
   case object waitFor {
     def task(task: AlgoliaTask): WaitForTaskDefinition =
       WaitForTaskDefinition(task.idToWaitFor())
+
+    def task(taskID: Long): WaitForTaskDefinition =
+      WaitForTaskDefinition(taskID)
   }
 
   implicit object WaitForTaskDefinitionExecutable

diff --git a/src/test/scala/algolia/dsl/WaitForTaskTest.scala b/src/test/scala/algolia/dsl/WaitForTaskTest.scala
@@ -38,6 +38,10 @@ class WaitForTaskTest extends AlgoliaTest {
       waitFor task Task(1L, None) from "toto" baseDelay 5 maxDelay 100
     }
 
+    it("wait for a taskId") {
+      waitFor task 1L from "toto" baseDelay 5 maxDelay 100
+    }
+
     it("should call API") {
       (waitFor task Task(1L, None) from "toto").build() should be(
         HttpPayload(
@@ -48,6 +52,17 @@ class WaitForTaskTest extends AlgoliaTest {
         )
       )
     }
+
+    it("should call API for a taskID") {
+      (waitFor task 1L from "toto").build() should be(
+        HttpPayload(
+          GET,
+          Seq("1", "indexes", "toto", "task", "1"),
+          isSearch = true,
+          requestOptions = None
+        )
+      )
+    }
   }
 
 }