NetworkLoggingDecorator implements Network, PerformanceLogger, Closeable { @Override public void reset() { partyStatsMap.clear(); } NetworkLoggingDecorator(Network network); protected  NetworkLoggingDecorator(Network network, Map<Integer, PartyStats> partyStatsMap); @Override byte[] receive(int partyId); @Override int getNoOfParties(); @Override void send(int partyId, byte[] data); @Override void reset(); @Override void close(); @Override Map<String, Long> getLoggedValues(); static final String NETWORK_PARTY_BYTES; static final String NETWORK_TOTAL_BYTES; static final String NETWORK_TOTAL_BATCHES; }
@Test public void reset() { HashMap<Integer, PartyStats> partyStatsMap = mock(HashMap.class); NetworkLoggingDecorator decorator = new NetworkLoggingDecorator(mock(Network.class), partyStatsMap); decorator.reset(); verify(partyStatsMap, times(1)).clear(); }