@Test public void shouldProperlySortByInheritance() { List<EntityDto> initialList = new ArrayList<>(); initialList.add(childEntity); initialList.add(parentEntity); List<EntityDto> entities = EntitySorter.sortByInheritance(initialList); Assert.assertTrue(entities.indexOf(parentEntity) < entities.indexOf(childEntity)); }