### Question:
DMNDiagramsSessionState { Diagram getDiagram(final String dmnDiagramElementId) { return diagramsByDiagramId.get(dmnDiagramElementId); } @Inject  DMNDiagramsSessionState(final DMNDiagramUtils dmnDiagramUtils); Diagram getDRGDiagram();  }### Answer:
@Test public void testGetDiagram() { assertEquals(stunnerDiagram1, sessionState.getDiagram(id1)); assertEquals(stunnerDiagram2, sessionState.getDiagram(id2)); assertEquals(stunnerDiagram3, sessionState.getDiagram(id3)); }