### Question:
DMNDiagramsSessionState { DMNDiagramElement getDMNDiagramElement(final String dmnDiagramElementId) { return dmnDiagramsByDiagramId.get(dmnDiagramElementId); } @Inject  DMNDiagramsSessionState(final DMNDiagramUtils dmnDiagramUtils); Diagram getDRGDiagram();  }### Answer:
@Test public void testGetDMNDiagramElement() { assertEquals(dmnDiagram1, sessionState.getDMNDiagramElement(id1)); assertEquals(dmnDiagram2, sessionState.getDMNDiagramElement(id2)); assertEquals(dmnDiagram3, sessionState.getDMNDiagramElement(id3)); }