TextUnitDTOAnnotations { public TextUnitDTO getTextUnitDTO(ITextUnit textUnit) { TextUnitDTO textUnitDTO = null; TextUnitDTOAnnotation annotation = textUnit.getAnnotation(TextUnitDTOAnnotation.class); if (annotation != null) { textUnitDTO = annotation.getTextUnitDTO(); } return textUnitDTO; }  TextUnitDTO getTextUnitDTO(ITextUnit textUnit);  }
@Test public void testGetTextUnitDTO() { ITextUnit textUnit = new TextUnit(); TextUnitDTO textUnitDTO = new TextUnitDTO(); textUnit.setAnnotation(new TextUnitDTOAnnotation(textUnitDTO)); TextUnitDTOAnnotations instance = new TextUnitDTOAnnotations(); TextUnitDTO expResult = textUnitDTO; TextUnitDTO result = instance.getTextUnitDTO(textUnit); assertEquals(expResult, result); }
@Test public void testGetTextUnitDTONull() { ITextUnit textUnit = new TextUnit(); TextUnitDTOAnnotations instance = new TextUnitDTOAnnotations(); TextUnitDTO expResult = null; TextUnitDTO result = instance.getTextUnitDTO(textUnit); assertEquals(expResult, result); }