### Question:
ExtensionBreakingQuestion { public EntitySelection get(EntityReference entityReference) { return concernedEntities.get(entityReference); } ExtensionBreakingQuestion(Map<EntityReference, EntitySelection> concernedEntities); EntitySelection get(EntityReference entityReference); Map<EntityReference, EntitySelection> getConcernedEntities(); Map<String, ExtensionSelection> getExtensions(); List<EntitySelection> getFreePages(); void selectAllExtensions(); void setSelectAllExtensions(boolean select); void selectAllFreePages(); void setSelectAllFreePages(boolean select); void markAsFreePage(EntitySelection entitySelection); void pageBelongsToExtension(EntitySelection entitySelection, XarInstalledExtension extension); ExtensionSelection getExtension(String extensionId); void unselectAll(); void setSelectedExtensions(Set<String> extensionIds); void setSelectedDocuments(Set<DocumentReference> documents);  }### Answer:
@Test public void get() { DocumentReference document1 = new DocumentReference("wiki1", "space1", "document1"); DocumentReference document2 = new DocumentReference("wiki2", "space2", "document2"); Map<EntityReference, EntitySelection> concernedEntities = new HashMap<>(); concernedEntities.put(document1, new EntitySelection(document1)); concernedEntities.put(document2, new EntitySelection(document2)); ExtensionBreakingQuestion question = new ExtensionBreakingQuestion(concernedEntities); assertEquals(document1, question.get(document1).getEntityReference()); assertTrue(question.get(document1).isSelected()); assertEquals(document2, question.get(document2).getEntityReference()); assertTrue(question.get(document2).isSelected()); }