Тестовый сервис с Mockito, когда Id установлен базой данных

Question

У меня есть функция createObject () в моем сервисе rest Service:

@Service
public class MyService {

    //Repos and contructor

   @Transactional
   public ObjectDto createObject(Object) {

       Mother mother = new Mother(name, age);
       Kid kid = new Kid(name, age);

       mother.addKid(kid);

       this.motherRepo.saveAndFlush(mother); 

       Long kidId = kid.getId();

       doStuffWithKidId();

       return new ObjectDto()
            .withMother(mother)
            .withKid(kid)
            .build();
  }
}

Моя сущность для мамы / ребенка в основном такая:

@Entity
@Table("mother")
public class mother() {

   @Id
   @GeneratedValue(strategy = GenerationType.IDENTITY)
   @Column(name="id)
   private Long id;

   //other attributes, including @OneToMany for Kid
   //Getter/Setter

}

Существует похожая сущностьдля детей.

Как видите, идентификатор устанавливается базой данных.В сущности нет идентификатора для идентификатора.Конструктор также не имеет идентификатора.

Теперь я хочу проверить свой сервис.Я копирую свои репозитории и хочу убедиться, что мой ObjectDto содержит значения, такие как id.

@RunWith(MockitoJUnitRunner.class)
@SpringBootTest
public MyServiceTest {

    @Mock
    private MotherRepo motherRepo;

    @InjectMocks
    private MyService myService;

    @Test
    void createObjectTest() {

        ObjectDto expectedObjectDto = setup...;
        Object inputObject = setup...;

        assertThat.(this.myService.createObject(inputObject))
             .isEqualToComparingFieldByField(expectedObjectDto);

    }
}

Ожидаемый ObjectDto выглядит примерно так:

{
   "motherId":1,
   "motherAge":40,
   "kidId":1
   ...
}

Проблема заключается в том, что идентификатор устанавливается базой данных.Поскольку базы данных нет, а в хранилище используется Mockito, это значение всегда равно нулю.Даже если я установил свой ожидаемый объект с нулевым значением в качестве идентификатора, мне нужен идентификатор в "doStuffWithKidId ()" в службе.Я получаю исключение NullPointerException.

Есть ли возможность установить идентификатор, как в ReflectionTestUtils.setField ()?В литературе я читал, что сервис всегда должен быть проверен с использованием макетов.Это правильно, или мне нужна БД в памяти, например, H2?

Спасибо за помощь.

Answer 1

Используйте doAnswer ...

import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.junit.MockitoJUnitRunner;
import org.springframework.test.util.ReflectionTestUtils;

import static org.assertj.core.api.Java6Assertions.assertThat;
import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.mock;

@RunWith(MockitoJUnitRunner.class)
public class MockitoSettingDatabaseIds {

    private static class TestEntity {
        private long id;
        private String text;

        public TestEntity(String text) {
            this.text = text;
        }

        public long getId() {
            return id;
        }

        public String getText() {
            return text;
        }
    }

    private interface TestEntityDAO {
        void save(TestEntity entity);
    }

    private static long someLogicToTest(TestEntityDAO dao, TestEntity entity) {
        dao.save(entity);
        return entity.getId();
    }

    @Test
    public void shouldReturnDatabaseGeneratedId() {
        long expectedId = 12345L;

        TestEntityDAO dao = mock(TestEntityDAO.class);
        TestEntity entity = new TestEntity("[MESSAGE]");

        doAnswer(invocation -> {
            ReflectionTestUtils.setField((TestEntity) invocation.getArgument(0), "id", expectedId);
            return null;
        }).when(dao).save(entity);

        assertThat(someLogicToTest(dao, entity)).isEqualTo(expectedId);
    }
}

Чтобы ответить на ваш комментарий, просто сделайте то же самое с коллекцией Kid, например ...

        doAnswer(invocation -> {
            Mother toSave = (Mother) invocation.getArgument(0);
            ReflectionTestUtils.setField(toSave, "id", expectedId);

            for (int k = 0; k < toSave.getKids().size(); k++) {
                ReflectionTestUtils.setField(toSave.getKids().get(k), "id", expectedId + k + 1);
            }

            return null;
        }).when(dao).save(entity);

Это установит id из Mother в expectedId и идентификаторы Kid s в expectedId + 1, expectedId + 2 и т. Д.