JUnit用于抽象类的最终方法

Question

我写JUnit测试有困难。问题是在下面的代码块中有最后一个方法getMessage（）：

if(these_conditions_are_true) {
    String message = messageResources.getMessage(CONSTANT_STRING, null, LocaleContextHolder.getLocale());
    modelView.getModelMap().addAttribute(INLINE_MSG, message);
    return modelView;
}

MessageResources是一个可重新加载的ResourceBundleMessageSource，它扩展了AbstractMessageSource，后者是一个抽象类。

getMessage（）是AbstractMessageSource中的最后一个方法。

这是我的Test类的浓缩版本：

@RunWith(PowerMockRunner.class)
@PrepareForTest({AbstractMessageSource.class, ClassBeingTested.class})
public class ClassBeingTestedTest {

@InjectMocks
ClassBeingTested classBeingTested;


@Mock
ReloadableResourceBundleMessageSource messageResources; //being used by a different test in this class, including in case it matters

@Before
public void setUp() {
    MockitoAnnotations.initMocks(this);
    // Build the mock
    this.mockMvc = MockMvcBuilders.standaloneSetup(classBeingTested).build();
}

@Test
public void Method_Being_Tested_Test() {
    AbstractMessageSource amsMock = PowerMockito.mock(AbstractMessageSource.class);     
    Mockito.when(amsMock.getMessage(any(), any(), any())).thenReturn("test01"); 
    ModelAndView mv = classBeingTested.methodBeingTested(itemVO);

    ModelAndView mv_Expected = createExpectedMV(); //not really how i'm doing it, but for the sake of simplicity i'll omit it

    assertEquals(mv_Expected, mv); 
}

}

我使用PowerMockito，因为我知道Mockito不能存根final方法。

我得到的错误是

在区域设置“EN_US”的代码“CONSTANT_STRING”下找不到消息。

看起来我的当...thenBack实际上并没有被用作getMessage（）正在被调用。我尝试将CALLS_REAL_METHOD添加到我创建Mock的行中，但这似乎不起作用。

有人知道怎么删掉这个吗？我模仿AbstractMessageSource，而不是ReloadableResourceBundleMessageSource，对吗？任何帮助都将不胜感激！

更新：

我仍在为此寻求帮助。我仍然遇到无法拦截getMessage（）实际被调用的问题...

Answer 1

试试这个

Mockito.when(
    amsMock.getMessage(
        eq(CONSTANT_STRING),
        eq(null),
        any(Locale.class))).thenReturn("test01");

编辑：这种策略很糟糕，因为在单元测试期间，需要对每个访问的属性模拟getMessage方法。

您似乎想要的是设置属性值以在单元测试期间使用的能力。如果是这种情况，请为您的单元测试包含一个属性配置。对于Spring，您可以创建一个引用测试属性的单元测试配置，并使用SpringRunner加载它。