Răsfoiți Sursa

Add ParentTaskAssigningClient.getParentTask accessor method (#101103)

Przemysław Witek 2 ani în urmă
părinte
comite
1ddd9879f1

+ 4 - 0
server/src/main/java/org/elasticsearch/client/internal/ParentTaskAssigningClient.java

@@ -38,6 +38,10 @@ public class ParentTaskAssigningClient extends FilterClient {
         this(in, new TaskId(localNode.getId(), parentTask.getId()));
     }
 
+    public TaskId getParentTask() {
+        return parentTask;
+    }
+
     /**
      * Fetch the wrapped client. Use this to make calls that don't set {@link ActionRequest#setParentTask(TaskId)}.
      */

+ 2 - 0
server/src/test/java/org/elasticsearch/client/internal/ParentTaskAssigningClientTests.java

@@ -36,6 +36,8 @@ public class ParentTaskAssigningClientTests extends ESTestCase {
             }
         };
         try (ParentTaskAssigningClient client = new ParentTaskAssigningClient(mock, parentTaskId[0])) {
+            assertEquals(parentTaskId[0], client.getParentTask());
+
             // All of these should have the parentTaskId set
             client.bulk(new BulkRequest());
             client.search(new SearchRequest());