From 19686a75525868134d762a6bd0e149bba420e364 Mon Sep 17 00:00:00 2001
From: Fernando Briano <fernando@picandocodigo.net>
Date: Fri, 11 Oct 2024 10:19:30 +0100
Subject: [PATCH] Adds tasks test for serverless, updates tasks test for stack

---
 tests/tasks.yml            | 26 ++++++++++++++++++++++++++
 tests/tasks_serverless.yml | 31 +++++++++++++++++++++++++++++++
 2 files changed, 57 insertions(+)
 create mode 100644 tests/tasks_serverless.yml

diff --git a/tests/tasks.yml b/tests/tasks.yml
index 8423104..767d186 100644
--- a/tests/tasks.yml
+++ b/tests/tasks.yml
@@ -3,11 +3,37 @@ requires:
   serverless: false
   stack: true
 ---
+setup:
+  - do:
+      indices.create:
+        index: task_test
+---
+teardown:
+  - do:
+      indices.delete:
+        index: task_test
+        ignore: 404
+---
 "tasks":
   - do:
       tasks.get: {}
   - is_true: nodes
 
+  - do:
+      update_by_query:
+        index: task_test
+        wait_for_completion: false
+        body: {}
+  - set: { task: task }
+
+  - do:
+      tasks.get: { task_id: $task }
+  - is_true: task
+
+  - do:
+      tasks.get: { task_id: 'node_id:123456' }
+      catch: resource_not_found_exception
+
   - do:
       tasks.list: {}
   - is_true: nodes
diff --git a/tests/tasks_serverless.yml b/tests/tasks_serverless.yml
new file mode 100644
index 0000000..3929127
--- /dev/null
+++ b/tests/tasks_serverless.yml
@@ -0,0 +1,31 @@
+---
+requires:
+  serverless: true
+  stack: false
+---
+setup:
+  - do:
+      indices.create:
+        index: task_test
+---
+teardown:
+  - do:
+      indices.delete:
+        index: task_test
+        ignore: 404
+---
+"Task":
+  - do:
+      update_by_query:
+        index: task_test
+        wait_for_completion: false
+        body: {}
+  - set: { task: task }
+
+  - do:
+      tasks.get: { task_id: $task }
+  - is_true: task
+
+  - do:
+      tasks.get: { task_id: 'node_id:123456' }
+      catch: resource_not_found_exception