@Test public void should_init_read_consistency_level_map() throws Exception { configMap.put( CONSISTENCY_LEVEL_READ_MAP_PARAM, ImmutableMap.of("cf1", "ONE", "cf2", "LOCAL_QUORUM")); doCallRealMethod().when(extractor).initReadConsistencyMap(configMap); Map<String, ConsistencyLevel> consistencyMap = extractor.initReadConsistencyMap(configMap); assertThat(consistencyMap.get("cf1")).isEqualTo(ConsistencyLevel.ONE); assertThat(consistencyMap.get("cf2")).isEqualTo(ConsistencyLevel.LOCAL_QUORUM); }