@Override public void resetProposalStatistics() { leader.getProposalStats().reset(); } }
public void reset() { resetLatency(); resetRequestCounters(); clientResponseStats.reset(); ServerMetrics.resetAll(); }
@Override public void commandRun() { if (!isZKServerRunning()) { pw.println(ZK_NOT_SERVING); } else { ServerStats serverStats = zkServer.serverStats(); serverStats.reset(); if (serverStats.getServerState().equals("leader")) { ((LeaderZooKeeperServer)zkServer).getLeader().getProposalStats().reset(); } pw.println("Server stats reset."); } } }
@Test public void testStatResetWithLeader() { // Arrange LeaderZooKeeperServer leaderZks = mock(LeaderZooKeeperServer.class); when(leaderZks.isRunning()).thenReturn(true); when(leaderZks.serverStats()).thenReturn(serverStats); Leader leader = mock(Leader.class); when(leaderZks.getLeader()).thenReturn(leader); statResetCommand.setZkServer(leaderZks); when(serverStats.getServerState()).thenReturn("leader"); BufferStats bufferStats = mock(BufferStats.class); when(leader.getProposalStats()).thenReturn(bufferStats); // Act statResetCommand.commandRun(); // Assert String output = outputWriter.toString(); assertEquals("Server stats reset.\n", output); verify(serverStats, times(1)).reset(); verify(bufferStats, times(1)).reset(); } }
@Test public void testReset() { BufferStats stats = new BufferStats(); stats.setLastBufferSize(10); assertEquals(10, stats.getLastBufferSize()); assertEquals(10, stats.getMinBufferSize()); assertEquals(10, stats.getMaxBufferSize()); stats.reset(); assertEquals(-1, stats.getLastBufferSize()); assertEquals(-1, stats.getMinBufferSize()); assertEquals(-1, stats.getMaxBufferSize()); } }