Relatorio extends Entidade { public List<Relato> relatosPorTipo(String tipo) { List<Relato> procurados = new ArrayList<>(); for(Relato relato : relatos) { if (relato.getClasse().equals(tipo)) { procurados.add(relato); } } return procurados; } Relatorio(String id, int anoBase, List<Relato> relatos); List<Relato> relatosPorTipo(String tipo); int getAnoBase(); List<Relato> getRelatos();  }
@Test public void semRelatosBuscaPorTipoRetornaZero() { Relatorio r = new Relatorio("r", 0, new ArrayList<>()); assertEquals(0, r.relatosPorTipo("qualquer coisa").size()); }
@Test public void relatosPorTipo() { Relatorio r = new Relatorio("id", 0, relatos); assertEquals(0, r.relatosPorTipo("x").size()); assertEquals(0, r.relatosPorTipo(null).size()); assertEquals(2, r.relatosPorTipo("a").size()); assertEquals(1, r.relatosPorTipo("b").size()); }