使用Mockito时如何使用@Value模拟映射

Question

我已经讨论过这个问题：如何使用Mockito在Spring中模拟自动构建的@Value字段？。我们如何模拟以下内容？

@Value("#{${patientTypes}}")
private Map<String, Integer> patientTypes;

这样我们就可以在进行模拟时获取其价值？

Answer 1

如果你只是想模拟你的映射并将其注入到测试中的类中，你应该创建一个模拟映射并通过ReflexiTestUtils注入它：

测试类别：

@Component
public class MyService {

    @Value("#{${patientTypes}}")
    private Map<String, Integer> patientTypes;

    public Integer getPatientTypeByKey(String key) {
        return patientTypes.get(key);
    }
}

对于Mockito测试，您可以使用InjectMocks：

@ExtendWith(MockitoExtension.class)
public class SimpleTest {

    @InjectMocks
    private MyService underTest;

    @Mock
    private Map<String, Integer> mockMap;


    @Test
    public void test() {
        when(mockMap.get(anyString())).thenReturn(15);

        Integer result = underTest.getPatientTypeByKey("some key");

        assertEquals(15, result);
    }
}

对于SpringBootTest，可以使用反射测试：

@SpringBootTest
public class SBTest {

    @Autowired
    private MyService underTest;

    @Mock
    private Map<String, Integer> mockMap;


    @Test
    public void test() {
        ReflectionTestUtils.setField(underTest, "patientTypes", mockMap);

        when(mockMap.get(anyString())).thenReturn(15);

        Integer result = underTest.getPatientTypeByKey("some key");

        assertEquals(15, result);
    }
}