Как протестировать приложение против Zookeeper

Question

У меня есть приложение, которое использует структуру куратора для определения лидерства. Мы хотели иметь возможность протестировать изменения состояния с сервера Zookeeper, есть ли способ сделать это?

С нашей точки зрения было бы полезно, если бы мы могли отправить серверу состояния подключения к подключенным клиентам (ПОДКЛЮЧЕННЫЕ, РЕКОНДРОВАННЫЕ, ПОДОЗРЕВАЕМЫЕ, ПОТЕРЯННЫЕ). Я знаю, что куратор предоставляет класс TestingServer, но он, похоже, не обладает этой функциональностью.

Любые решения или мысли были бы высоко оценены!

Answer 1

https://apache.googlesource.com/curator/+/refs/heads/CURATOR-190/curator-recipes/src/test/java/org/apache/curator/framework/client/TestBackgroundStates.java

@Test 
public void testConnectionStateListener() throws Exception 
{ 
    server.close(); 
    Timing timing = new Timing(); 
    CuratorFramework client = CuratorFrameworkFactory.newClient(server.getConnectString(), timing.session(), timing.connection(), new RetryOneTime(timing.milliseconds())); 
    try 
    { 
     client.start(); 
     final BlockingQueue<ConnectionState> stateVector = Queues.newLinkedBlockingQueue(1); 
     ConnectionStateListener listener = new ConnectionStateListener() 
     { 
      @Override 
      public void stateChanged(CuratorFramework client, ConnectionState newState) 
      { 
       stateVector.offer(newState); 
      } 
     }; 
     Timing waitingTiming = timing.forWaiting(); 
     client.getConnectionStateListenable().addListener(listener); 
     server = new TestingServer(server.getPort()); 
     Assert.assertEquals(stateVector.poll(waitingTiming.milliseconds(), TimeUnit.MILLISECONDS), ConnectionState.CONNECTED); 
     server.stop(); 
     Assert.assertEquals(stateVector.poll(waitingTiming.milliseconds(), TimeUnit.MILLISECONDS), ConnectionState.SUSPENDED); 
     Assert.assertEquals(stateVector.poll(waitingTiming.milliseconds(), TimeUnit.MILLISECONDS), ConnectionState.LOST); 
     server.restart(); 
     Assert.assertEquals(stateVector.poll(waitingTiming.milliseconds(), TimeUnit.MILLISECONDS), ConnectionState.RECONNECTED); 
     server.close(); 
     Assert.assertEquals(stateVector.poll(waitingTiming.milliseconds(), TimeUnit.MILLISECONDS), ConnectionState.SUSPENDED); 
     Assert.assertEquals(stateVector.poll(waitingTiming.milliseconds(), TimeUnit.MILLISECONDS), ConnectionState.LOST); 
    } 
    finally 
    { 
     CloseableUtils.closeQuietly(client); 
    } 
}