Add basic UT

Signed-off-by: Rishab Nahata <rnnahata@amazon.com>
opensearch-project · Sep 26, 2022 · bb0547d · bb0547d
1 parent c9e0f5b
commit bb0547d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/server/src/test/java/org/opensearch/discovery/PeerFinderTests.java b/server/src/test/java/org/opensearch/discovery/PeerFinderTests.java
@@ -807,6 +807,14 @@ public void testReconnectsToDisconnectedNodes() {
         assertFoundPeers(rebootedOtherNode);
     }
 
+    public void testNodeCommissioning() {
+        peerFinder.nodeCommissionedListener().onFailure(new Exception("unit-test"));
+        assertTrue(peerFinder.localNodeDecommissioned());
+
+        peerFinder.nodeCommissionedListener().onResponse(null);
+        assertFalse(peerFinder.localNodeDecommissioned());
+    }
+
     private void respondToRequests(Function<DiscoveryNode, PeersResponse> responseFactory) {
         final CapturedRequest[] capturedRequests = capturingTransport.getCapturedRequestsAndClear();
         for (final CapturedRequest capturedRequest : capturedRequests) {