@Test public void testIsServerReady() throws Exception { when(serviceState.getState()).thenReturn(ServiceState.ServiceStateValue.ACTIVE); LocalAtlasClient atlasClient = new LocalAtlasClient(serviceState, entityResource); assertTrue(atlasClient.isServerReady()); when(serviceState.getState()).thenReturn(ServiceState.ServiceStateValue.BECOMING_ACTIVE); assertFalse(atlasClient.isServerReady()); }