如何使用powermock部分模拟外部方法调用

Question

我有两个类在交互。说ServiceLayer.class和ApiAdaptor.class:

public class ApiAdaptor {
  public String getCountry(String latitude, String longitude) {
    // REST call to api goes here and a the country string is returned
    return country;
  }
}
public class ServiceLayer {
 public String findCountry(int positionId) {
   // some business logic here to get latitude and longitude from postitionId 
   ApiAdaptor api = new ApiAdaptor();
   String country = api.getCountry(latitude, longitude);
   return country;
 }
}

现在在单元测试中，我只想测试这个方法ServiceLayer.findcountry()，同时模拟对ApiAdaptor.getCountry(latitude, longitude)的内部调用。有什么方法可以用Powermock做到这一点吗。在Ruby On Rails中使用Rspec时，我看到过类似类型的stubing。我也想在我的javaSpringMVC项目中做类似的测试。

Answer 1

当然，您可以使用PowerMock只关注该方法。例如，专门使用PowerMockito，您可以编写以下测试：

@RunWith(PowerMockRunner.class)
@PrepareForTest( {ServiceLayer.class} )
public class PowerMockitoJan10Test {
    private static final java.lang.String DESIRED_COUNTRY_VALUE = "USA";
    @Test
    public void testServiceLayerFindCountry() throws Exception {
        ApiAdaptor mock = Mockito.mock(ApiAdaptor.class);
        PowerMockito.whenNew(ApiAdaptor.class).withAnyArguments().thenReturn(mock);
        Mockito.when(mock.getCountry(Mockito.anyString(), Mockito.anyString())).thenReturn(DESIRED_COUNTRY_VALUE);
        String country = new ServiceLayer().findCountry(1);
        Assert.assertEquals(DESIRED_COUNTRY_VALUE, country);
    }
}

如果您使用Spring，很可能还需要一个JUnit运行程序，因此您可以使用PowerMockito的JUnit规则——请参阅本例。

---

编辑：这很有趣。在使用该规则时，除非将ServiceLayer.class添加到@PrepareForTest列表中，否则它实际上不起作用。在撰写本文时，我使用了最新的PowerMockito版本1.6.4。可能值得报道。在任何情况下，这就是您的测试将如何使用Spring:

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration("mycontext.xml")
@PrepareForTest({ApiAdaptor.class, ServiceLayer.class})
public class PowerMockitoJan10_WithRuleTest {
    private static final String DESIRED_COUNTRY_VALUE = "USA";
    @Rule
    public PowerMockRule rule = new PowerMockRule();
    @Test
    public void testServiceLayerFindCountry() throws Exception {
        PowerMockito.whenNew(ApiAdaptor.class).withNoArguments().thenReturn(new ApiAdaptor() {
            @Override
            public String getCountry(String latitude, String longitude) {
                return DESIRED_COUNTRY_VALUE;
            }
        });
        String country = new ServiceLayer().findCountry(1);
        Assert.assertEquals(DESIRED_COUNTRY_VALUE, country);
    }
}

或者，如果覆盖是一个问题，您可以模拟ApiAdaptor:

    ...
    ApiAdaptor mock = PowerMockito.mock(ApiAdaptor.class);
    PowerMockito.when(mock.getCountry(Mockito.anyString(), Mockito.anyString())).thenReturn(DESIRED_COUNTRY_VALUE);
    PowerMockito.whenNew(ApiAdaptor.class).withNoArguments().thenReturn(mock);
    ...

Answer 2

如果您可以更改代码，我建议您通过对类进行依赖注入来提高代码的可测试性。

你会有这样的东西：

public class ServiceLayer {
    private ApiAdaptor _api;
    public ServiceLayer(ApiAdaptor api) {
        _api = api;
    }
    [snip]
}

然后在代码的其余部分使用_api。

当你需要测试这个类时，如果你必须模拟ApiAdaptor，你现在可以写：

ApiAdaptor mock = Mockito.mock(ApiAdaptor.class);
[Api adaptor behavior mocking]
String country = new ServiceLayer(mock).findCountry(1);
[asserts and all]

这消除了对PowerMockito、其运行程序和/或规则的需求…