Add ClusterInstanceRegistry

apache · Dec 17, 2024 · 22144c1 · 22144c1
1 parent c728695
commit 22144c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...e/mode/manager/cluster/event/dispatch/subscriber/type/ComputeNodeStateSubscriberTest.java b/...e/mode/manager/cluster/event/dispatch/subscriber/type/ComputeNodeStateSubscriberTest.java
@@ -58,13 +58,13 @@ void assertRenewWithInstanceOnlineEvent() {
         ComputeNodeInstance computeNodeInstance = mock(ComputeNodeInstance.class);
         when(contextManager.getPersistServiceFacade().getComputeNodePersistService().loadComputeNodeInstance(instanceMetaData)).thenReturn(computeNodeInstance);
         subscriber.renew(new InstanceOnlineEvent(instanceMetaData));
-        verify(contextManager.getComputeNodeInstanceContext()).getClusterInstanceRegistry().add(computeNodeInstance);
+        verify(contextManager.getComputeNodeInstanceContext().getClusterInstanceRegistry()).add(computeNodeInstance);
     }
 
     @Test
     void assertRenewWithInstanceOfflineEvent() {
         subscriber.renew(new InstanceOfflineEvent(mock(InstanceMetaData.class)));
-        verify(contextManager.getComputeNodeInstanceContext()).getClusterInstanceRegistry().delete(any());
+        verify(contextManager.getComputeNodeInstanceContext().getClusterInstanceRegistry()).delete(any());
     }
 
     @Test