Add back localnode var

Signed-off-by: Tyler Ohlsen <[email protected]>
opensearch-project · Dec 26, 2023 · b7a7a88 · b7a7a88
1 parent d5f3de6
commit b7a7a88
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/test/java/org/opensearch/ad/client/AnomalyDetectionNodeClientTests.java b/src/test/java/org/opensearch/ad/client/AnomalyDetectionNodeClientTests.java
@@ -39,6 +39,7 @@
 import org.opensearch.ad.transport.ADTaskProfileResponse;
 import org.opensearch.client.Client;
 import org.opensearch.cluster.ClusterName;
+import org.opensearch.cluster.node.DiscoveryNode;
 import org.opensearch.core.action.ActionListener;
 import org.opensearch.index.query.BoolQueryBuilder;
 import org.opensearch.index.query.TermQueryBuilder;
@@ -161,13 +162,14 @@ public void testGetDetectorProfile_NoIndices() throws ExecutionException, Interr
 
     @Test
     public void testGetDetectorProfile_Populated() {
+        DiscoveryNode localNode = clusterService().localNode();
         ADTaskProfile adTaskProfile = new ADTaskProfile("foo-task-id", 0, 0L, false, 0, 0L, localNode.getId());
 
         doAnswer(invocation -> {
             Object[] args = invocation.getArguments();
 
             ActionListener<ADTaskProfileResponse> listener = (ActionListener<ADTaskProfileResponse>) args[2];
-            ADTaskProfileNodeResponse nodeResponse = new ADTaskProfileNodeResponse(clusterService().localNode(), adTaskProfile, null);
+            ADTaskProfileNodeResponse nodeResponse = new ADTaskProfileNodeResponse(localNode, adTaskProfile, null);
 
             List<ADTaskProfileNodeResponse> nodeResponses = Arrays.asList(nodeResponse);
             listener.onResponse(new ADTaskProfileResponse(new ClusterName("test-cluster"), nodeResponses, Collections.emptyList()));