@Before public void setUp() { outputWriter = new StringWriter(); ServerCnxn serverCnxnMock = mock(ServerCnxn.class); LeaderZooKeeperServer zks = mock(LeaderZooKeeperServer.class); when(zks.isRunning()).thenReturn(true); providerMock = mock(ServerStats.Provider.class); when(zks.serverStats()).thenReturn(new ServerStats(providerMock)); ZKDatabase zkDatabaseMock = mock(ZKDatabase.class); when(zks.getZKDatabase()).thenReturn(zkDatabaseMock); Leader leaderMock = mock(Leader.class); when(leaderMock.getProposalStats()).thenReturn(new BufferStats()); when(zks.getLeader()).thenReturn(leaderMock); ServerCnxnFactory serverCnxnFactory = mock(ServerCnxnFactory.class); ServerCnxn serverCnxn = mock(ServerCnxn.class); List<ServerCnxn> connections = new ArrayList<>(); connections.add(serverCnxn); when(serverCnxnFactory.getConnections()).thenReturn(connections); statCommand = new StatCommand(new PrintWriter(outputWriter), serverCnxnMock, FourLetterCommands.statCmd); statCommand.setZkServer(zks); statCommand.setFactory(serverCnxnFactory); }
Leader(QuorumPeer self,LeaderZooKeeperServer zk) throws IOException { this.self = self; this.proposalStats = new BufferStats(); try { if (self.shouldUsePortUnification() || self.isSslQuorum()) {
@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()); } }
@Test public void testSetProposalSizeSetMinMax() { BufferStats stats = new BufferStats(); assertEquals(-1, stats.getLastBufferSize()); assertEquals(-1, stats.getMinBufferSize()); assertEquals(-1, stats.getMaxBufferSize()); stats.setLastBufferSize(10); assertEquals(10, stats.getLastBufferSize()); assertEquals(10, stats.getMinBufferSize()); assertEquals(10, stats.getMaxBufferSize()); stats.setLastBufferSize(20); assertEquals(20, stats.getLastBufferSize()); assertEquals(10, stats.getMinBufferSize()); assertEquals(20, stats.getMaxBufferSize()); stats.setLastBufferSize(5); assertEquals(5, stats.getLastBufferSize()); assertEquals(5, stats.getMinBufferSize()); assertEquals(20, stats.getMaxBufferSize()); }