Junit Unit Test For a Method That Includes TransactionTemplate

Question

I have a method that includes transactionTemplate in it. My test works when I run debug mode but not when executed directly.

My test works when I run debug mode because I think transactional operation is done while working debug mode. But when I just run it works wrong. How can I overcome this problem?

This is my method:

@Override
public void saveAndUpdate(DataDictionaryDTO input) {
    try {
        transactionTemplate.execute(new TransactionCallbackWithoutResult() {
            protected void doInTransactionWithoutResult(TransactionStatus status) {
                DataDictionaryEntity ent = null;
                if (NullUtils.isNotNullOrZero(input.getId())) {
                    ent = dataDictionaryRepository.findById(input.getId()).orElse(null);
                }
                dataDictionaryRepository.save(dataDictionaryDTOMapper.map(input, ent));
            }
        });
    }
    catch (DataIntegrityViolationException e) {
        throw new UserFriendlyException("there.is.already.such.a.value");
    }
}

Rubricate

pls show us your test – Bourgeon

And please explain what you mean by your test does not work. – Wilbertwilborn

Answer 1

Try this mock code

when(transactionTemplate.execute(any())).thenAnswer(invocation -> {
        TransactionCallbackWithoutResult transactionCallbackWithoutResult = (TransactionCallbackWithoutResult)invocation.getArguments()[0];
        transactionCallbackWithoutResult.doInTransaction(new SimpleTransactionStatus());
        return null;
    });

Guttapercha

Answer 2

You can try this, it works on me.

doAnswer(invocation -> {
    Consumer<TransactionStatus> consumer = invocation.getArgument(0);
    consumer.accept(null);
    return null;
  }).when(transactionTemplate).executeWithoutResult(any(Consumer.class));

If you think null argument to consumer is not a good idea, you can also try this one:

doAnswer(invocation -> {
    Consumer<TransactionStatus> consumer = invocation.getArgument(0);
    consumer.accept(null);
    return null;
  }).when(transactionTemplate).executeWithoutResult(any(Consumer.class));

....

private TransactionStatus newTransactionStatus() {
  return new DefaultTransactionStatus(null, true, false, false, false, null);
}

Sort