TranslationKitStep extends BasePipelineStep { TMTextUnitVariant getTMTextUnitVariant(ITextUnit textUnit) { TMTextUnitVariant tmTextUnitVariant = null; TextUnitDTO textUnitDTO = textUnitDTOAnnotations.getTextUnitDTO(textUnit); if (textUnitDTO != null && textUnitDTO.getTmTextUnitVariantId() != null) { tmTextUnitVariant = tmTextUnitVariantRepository.getOne(textUnitDTO.getTmTextUnitVariantId()); } return tmTextUnitVariant; } TranslationKitStep(Long translationKitId); @Override String getName(); @Override String getDescription();  }
@Test public void testGetTMTextUnitVariant() { logger.debug("Ids for entities worked on"); Long tmTextUnitVariantId = 123456789L; logger.debug("Create mocks"); TMTextUnitVariantRepository mockTmTextUnitVariantRepository = getJpaRepositoryMockForGetOne(TMTextUnitVariantRepository.class, TMTextUnitVariant.class, tmTextUnitVariantId); logger.debug("Create instance to be tested, inject mocks and state"); TranslationKitStep translationKitStep = new TranslationKitStep(987654321L); translationKitStep.textUnitDTOAnnotations = new TextUnitDTOAnnotations(); translationKitStep.tmTextUnitVariantRepository = mockTmTextUnitVariantRepository; logger.debug("Build test input"); ITextUnit textUnit = new TextUnit(); TextUnitDTO textUnitDTO = new TextUnitDTO(); textUnitDTO.setTmTextUnitVariantId(tmTextUnitVariantId); textUnit.setAnnotation(new TextUnitDTOAnnotation(textUnitDTO)); logger.debug("Test and assert"); TMTextUnitVariant result = translationKitStep.getTMTextUnitVariant(textUnit); assertEquals("The tmTextUnitVariant#Id extracted from the text unit must be the same as the one set in the TextUnitDTO", tmTextUnitVariantId, result.getId()); }