JUnit 测试。 使用ModelMapper库将entity转换为DTO时的问题

Question

我正在开发 Spring Boot Web 应用程序，并且我有一个 ModelMapper 库的自定义实现，它允许我转换单个对象和对象列表。

@Component
public class ObjectMapperUtils {

@Autowired
private static ModelMapper modelMapper;

static {
    modelMapper = new ModelMapper();
    modelMapper.getConfiguration().setMatchingStrategy(MatchingStrategies.STRICT);
}

private ObjectMapperUtils() {
}

public <D, T> D map(final T entity, Class<D> outClass) {
    return modelMapper.map(entity, outClass);
}

public <D, T> List<D> mapAll(final Collection<T> entityList, Class<D> outCLass) {
    return entityList.stream().map(entity -> map(entity, outCLass)).collect(Collectors.toList());
}
}

在服务层，我有一个方法从 DB UserEntity 对象返回并将其转换为 UserDTO。

@Autowired
private UserRepository userRepository;

@Autowired
private ObjectMapperUtils modelMapper;

@Override
public UserDTO getByUserId(String userId) {
    UserEntity userEntity = userRepository.findByUserId(userId)
            .orElseThrow(() -> new NotFoundException("User with userId[" + userId + "] not found"));
    //UserDTO userDTO = new UserDTO();
    //BeanUtils.copyProperties(userEntity, userDTO);
    return modelMapper.map(userEntity, UserDTO.class); // userDTO;
}

当我尝试为此方法创建测试时会出现问题。 UserDTO 始终作为 NULL 值返回。

class UserServiceImplTest {

@InjectMocks
private UserServiceImpl userService;

@Mock
private UserRepository userRepository;

@Mock
private ObjectMapperUtils modelMapper;

@BeforeEach
void setUp() throws Exception {
    MockitoAnnotations.initMocks(this);
}

@Test
void testGetByUserId() {
    UserEntity userEntity = new UserEntity();
    userEntity.setId(1L);
    userEntity.setUsername("zavada");
    userEntity.setUserId("33b4c069-e907-45a9-8d49-2042044c56e0");

    when(userRepository.findByUserId(anyString()))
                 .thenReturn(Optional.of(userEntity));

    UserDTO userDTO = userService.getByUserId("33b4c069-e907-45a9-8d49-2042044c56e0");
    System.out.println(userDTO); <--- NULL

    assertEquals("zavada", userDTO.getUsername());
    assertNotNull(userDTO);

}
}

当我在服务层使用 BeanUtils.copyProperties(obj1, obj2) 转换时； - 测试成功通过。 使用 ModelMapper 我得到 NULL。 任何想法如何解决此错误或重构代码？ 提前致谢

Answer 1

如果您有@Mock private ObjectMapperUtils modelMapper; 那么默认情况下，您没有真正的ObjectMapperUtils 。 因此，您不是在调用实现，而是在调用Mockito为您提供的默认存根。 这就是为什么您从modelMapper.map()获得null返回值的modelMapper.map() 。

要么不模拟ObjectMapperUtils bean，要么使用Mockito API中的when()等安排它进行适当的处​​理。

Answer 2

要建立在user268396答案的基础上，您需要执行以下操作才能使其工作：

@RunWith(MockitoJUnitRunner.class)
    public class StackOverflowTest {

        @InjectMocks
        private StackOverflow userService = new StackOverflow();

        @Mock
        private UserRepository userRepository;

        @Mock
        private ObjectMapperUtils modelMapper;
        private UserDTO userDTO = new UserDTO();
        private UserEntity userEntity = new UserEntity();

        @Before
        public void setUp() {
            when(modelMapper.map(any(), any())).thenReturn(userDTO);

            userDTO.setId(1L);
            userDTO.setUsername("zavada");
            userDTO.setUserId("33b4c069-e907-45a9-8d49-2042044c56e0");
        }

        @Test
        public void testGetByUserId() throws Throwable {
            when(userRepository.findByUserId(anyString())).thenReturn(Optional.of(userEntity));

            UserDTO result = userService.getByUserId("33b4c069-e907-45a9-8d49-2042044c56e0");
            System.out.println(result);

            assertEquals("zavada", result.getUsername());
            assertNotNull(result);

        }
    }

这是一个很容易犯的错误，重要的是要记住，所有@mock ed对象不再是真正的实现，并且如果您期望任何行为都需要预先定义。

Answer 3

对于同样的情况，你也可以这样做

1. 不要在测试类中模拟 ObjectMapperUtils 类。
2. 在@Before init() ..{} 方法中，将您的 UserServiceImpl 实例化为 ObjectMapperUtils 的新实例。 然后在 UserServiceImpl 中进行设置。 即，userServiceImpl.setMapper(UserServiceImpl)。

所以你的设置方法看起来像这样，

@InjectMocks
private UserServiceImple userServiceImpl;

@Mock
private UserRepository userRepository;

private ObjectMapperUtil objectMapperUitl; // not mocked

@BeforeEach
public void init() { 
    objectMapperUitls = new ObjectMapperUtils();
    userServiceImple.setMapper(objectMapperUitls);
}