@Test
public void shouldProducePartitionMap() {

    partition = source.partition(REPLICA_SET_NAME);
    assertThat(partition.get(SourceInfo.REPLICA_SET_NAME)).isEqualTo(REPLICA_SET_NAME);
    assertThat(partition.get(SourceInfo.SERVER_ID_KEY)).isEqualTo("serverX");
    assertThat(partition.size()).isEqualTo(2);
}