Testovanie cassandra Session.State - ako sa vysmieť - java, unit-testing, cassandra

Pozerám sa na túto funkciu:

Session.State state = session.getState();
if(state.getConnectedHosts().size() == 0)
return  Result.unhealthy("No cassandra connections");
else return Result.healthy();

Rád by som testoval rôzne podmienky tu, ale nie som jasný, ako sa vysmievať / simulovať / riešiť Session.State hodnota. Môžem si vysmievať zasadania ale druhý je rozhranie - pri behu je to a SessionManager ktorý nemá prístup verejnosti.

Existuje spôsob, ako (Power) vysmievať? Ďalšou cestou, ako to vyskúšať?

odpovede:

Použitím Mockito, pre mňa nasledujúce práce:

import static org.mockito.Mockito.*;

Session session = mock(Session.class);
Session.State state = mock(Session.State.class);
when(session.getState()).thenReturn(state);
when(state.getConnectedHosts()).thenReturn(Collections.<Host>emptyList());