@Test
public void testGetTopEntities() {

    Interval timeInterval = new Interval(mentionDate, mentionDate.plusHours(3));
    Map<String, Long> result =
        underTest.getTopEntities(timeInterval, ImmutableList.of(), ImmutableList.of(), 10,
                                 true);
    assertEquals(2, result.size());
    assertEquals(3L, result.get("e1").longValue());
    assertEquals(1L, result.get("e2").longValue());
    result = underTest.getTopEntities(timeInterval, ImmutableList.of("room1"),
                                      ImmutableList.of(), 10, true);
    assertEquals(2, result.size());
    assertEquals(2L, result.get("e1").longValue());
    assertEquals(1L, result.get("e2").longValue());
    result = underTest.getTopEntities(timeInterval, ImmutableList.of("room1"),
                                      ImmutableList.of("giannis"), 10, true);
    assertEquals(2, result.size());
    assertEquals(1L, result.get("e1").longValue());
    assertEquals(1L, result.get("e2").longValue());
}