### Question:
DMNDiagramsSessionState { List<DRGElement> getModelDRGElements() { return getDMNDiagrams() .stream() .flatMap(diagram -> dmnDiagramUtils.getDRGElements(diagram.getStunnerDiagram()).stream()) .distinct() .collect(Collectors.toList()); } @Inject  DMNDiagramsSessionState(final DMNDiagramUtils dmnDiagramUtils); Diagram getDRGDiagram();  }### Answer:
@Test public void testGetModelDRGElements() { final DRGElement drgElement1 = mock(DRGElement.class); final DRGElement drgElement2 = mock(DRGElement.class); final DRGElement drgElement3 = mock(DRGElement.class); final DRGElement drgElement4 = mock(DRGElement.class); final DRGElement drgElement5 = mock(DRGElement.class); final DRGElement drgElement6 = mock(DRGElement.class); when(dmnDiagramUtils.getDRGElements(stunnerDiagram1)).thenReturn(asList(drgElement1, drgElement2)); when(dmnDiagramUtils.getDRGElements(stunnerDiagram2)).thenReturn(asList(drgElement3, drgElement4, drgElement5)); when(dmnDiagramUtils.getDRGElements(stunnerDiagram3)).thenReturn(singletonList(drgElement6)); final List<DRGElement> drgElements = sessionState.getModelDRGElements(); assertTrue(drgElements.contains(drgElement1)); assertTrue(drgElements.contains(drgElement2)); assertTrue(drgElements.contains(drgElement3)); assertTrue(drgElements.contains(drgElement4)); assertTrue(drgElements.contains(drgElement5)); assertTrue(drgElements.contains(drgElement6)); }