使用 when + thenReturn 时 Mockito 和泛型的问题

Question

我一直在网上寻找很多关于这个问题的信息，但我找到的解决方案都没有对我有用。

我有一个测试类，以及我试图测试的服务。

在测试类中：

public class ElasticSearchSearchServiceImplTest{

@InjectMocks
private ElasticSearchSearchServiceImpl elasticSearchSearchService = new ElasticSearchSearchServiceImpl();

private SearchMyListCriteria searchMyListCriteria;

private SearchHitListCriteria searchHitListCriteria;

private SearchFieldLimitsCriteria searchFieldLimitsCriteria;

private SearchDetailCriteria searchDetailCriteria;

private SearchBrowseCriteria searchBrowseCriteria;

@Mock
private SearchRequestBuilder searchRequestBuilder;

@Mock
private SecurityService securityService;

@Mock(answer = Answers.RETURNS_DEEP_STUBS)
private Client client;

@Mock
private ListenableActionFuture<SearchResponse> listenableActionFuture;

@BeforeMethod
public void setup() throws InterruptedException
{
    MockitoAnnotations.initMocks(this);
    SearchResponse sr = new SearchResponse();
    Mockito.when(client.prepareSearch(Mockito.anyString())
            .setQuery(Mockito.any(QueryBuilder.class))
            .addFields(Mockito.anyObject())
            .setFrom(Mockito.anyInt())
            .setSize(Mockito.anyInt())
            ).thenReturn(searchRequestBuilder);
    Mockito.when(securityService.isAccountabilityEnabled()).thenReturn(false);

    Mockito.when(searchRequestBuilder.execute()).thenReturn(listenableActionFuture);
    Mockito.when(listenableActionFuture.actionGet()).thenReturn(sr);

    searchMyListCriteria = buildSearchCriteriaBase();
    searchHitListCriteria = buildSearchHitListCriteria();
    searchFieldLimitsCriteria = buildSearchFieldLimitsCriteria();
    searchDetailCriteria = buildSearchDetailCriteria();
    searchBrowseCriteria = buildSearchBrowseCriteria();
}

@Test
public void testDoMyListSearch()
{
    SearchResponse searchResponse = new SearchResponse();
    Boolean execOk = false;
    try {
        searchResponse = elasticSearchSearchService.doMyListSearch(searchMyListCriteria);
        Assert.assertNotNull(searchResponse);
        execOk = true;
    }catch (Exception e){
        e.printStackTrace();
    }
    Assert.assertTrue(execOk);
}

然后，在 ElasticSearchSearchServiceImpl.java 中，我在测试中调用了这个方法：

@Override
public SearchResponse doMyListSearch(SearchMyListCriteria criteria) throws Exception
{
    SearchTimer t = new SearchTimer();
    SearchResponse resp = null;
    int size = criteria.getDocIds().size();
    if(!criteria.getDocIds().isEmpty())
    {
        String indexName = getSearchAlias(criteria);
        String[] ids = criteria.getDocIds().toArray(new String[criteria.getDocIds().size()]);
        IdsQueryBuilder qb = QueryBuilders.idsQuery(getESTypes(criteria));
        qb.addIds(ids);

        SearchDataSourceAccountabilityEnum requiredAcc = securityService.isAccountabilityEnabled() && !criteria.getPermissions().isSuperUser()
                ? getAccountabilityForDataSources(criteria)
                : SearchDataSourceAccountabilityEnum.NONE;
        List<QueryBuilder> filters = new ArrayList<>();
        addSecurityToFilter(filters, criteria, requiredAcc);
        QueryBuilder filter = ElasticSearchUtils.and(filters);
        QueryBuilder fqb = filter == null ? qb : QueryBuilders.boolQuery().must(qb).filter(filter);

        SearchRequestBuilder req = client.prepareSearch(indexName)
                                         .setQuery(fqb)
                                         .addFields(determineMyListFieldsToReturn(criteria))
                                         .setFrom(0)
                                         .setSize(size);

        resp = req.execute().actionGet();
    }

    long searchMillis = t.getElapsedMillis();
    logPerform("Query Execution time for my list search " + criteria.toString() + COLON_SPACE + searchMillis + MILLISECONDS_MARKER);
return resp;
}

问题是，当它在“resp = req.execute().actionGet();”行通过 ElasticSearchSearchServiceImpl#doMyListSearch 时它抛出以下异常：

java.lang.ClassCastException: org.mockito.internal.creation.cglib.ClassImposterizer$ClassWithSuperclassToWorkAroundCglibBug$$EnhancerByMockitoWithCGLIB$$db3b9e0a cannot be cast to org.elasticsearch.action.search.SearchResponse
at com.sirsidynix.bcss.search.biz.svc.impl.ElasticSearchSearchServiceImpl.doMyListSearch(ElasticSearchSearchServiceImpl.java:228)
at com.sirsidynix.bcss.search.biz.svc.impl.ElasticSearchSearchServiceImplTest.testDoMyListSearch(ElasticSearchSearchServiceImplTest.java:116)
at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62)
at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)
at java.lang.reflect.Method.invoke(Method.java:498)
at org.testng.internal.MethodInvocationHelper.invokeMethod(MethodInvocationHelper.java:80)
at org.testng.internal.Invoker.invokeMethod(Invoker.java:714)
at org.testng.internal.Invoker.invokeTestMethod(Invoker.java:901)
at org.testng.internal.Invoker.invokeTestMethods(Invoker.java:1231)
at org.testng.internal.TestMethodWorker.invokeTestMethods(TestMethodWorker.java:127)
at org.testng.internal.TestMethodWorker.run(TestMethodWorker.java:111)
at org.testng.TestRunner.privateRun(TestRunner.java:767)
at org.testng.TestRunner.run(TestRunner.java:617)
at org.testng.SuiteRunner.runTest(SuiteRunner.java:334)
at org.testng.SuiteRunner.runSequentially(SuiteRunner.java:329)
at org.testng.SuiteRunner.privateRun(SuiteRunner.java:291)
at org.testng.SuiteRunner.run(SuiteRunner.java:240)
at org.testng.SuiteRunnerWorker.runSuite(SuiteRunnerWorker.java:52)
at org.testng.SuiteRunnerWorker.run(SuiteRunnerWorker.java:86)
at org.testng.TestNG.runSuitesSequentially(TestNG.java:1198)
at org.testng.TestNG.runSuitesLocally(TestNG.java:1123)
at org.testng.TestNG.run(TestNG.java:1031)
at org.testng.remote.AbstractRemoteTestNG.run(AbstractRemoteTestNG.java:132)
at org.testng.remote.RemoteTestNG.initAndRun(RemoteTestNG.java:236)
at org.testng.remote.RemoteTestNG.main(RemoteTestNG.java:81)

我已经尝试了网络中的大多数解决方案，但似乎没有任何效果对我有用。 请如果你有建议，我很感激。

问候！

Answer 1

您是否尝试将测试减少到导致测试失败的最小代码段？

你认为这是泛型的问题。 如果是这样，那么像下面这样简单的事情也会失败：

@Mock
private ListenableActionFuture<SearchResponse> listenableActionFuture;

@BeforeMethod
public void setup() throws InterruptedException {
    MockitoAnnotations.initMocks(this);
    SearchResponse sr = new SearchResponse();
    Mockito.when(listenableActionFuture.actionGet()).thenReturn(sr);
}

@Test
public void testDoMyListSearch() {
    SearchResponse sr = listenableActionFuture.actionGet();
}

可以？ 我的猜测不是..我认为这更有可能是由于深存根以及深存根返回与您在thenReturn返回的模拟相同类型的模拟thenReturn 。 我想知道是否有必要使用thenReturn ，或者深度存根是否会让它“神奇地”工作。

无论如何，我的建议是不断减少空间以找到问题所在。 此外，删除诸如searchDetailCriteria类的不必要的内容将有助于人们理解代码并更快地帮助您。

（对不起，我更愿意发表评论，但没有足够的声誉）。

Answer 2

我认为客户端会为某些方法调用返回一个 SearchRequestBuilder。 所以你可能需要模拟一个 SearchRequestBuilder。 您能否在存根客户端之前尝试添加以下内容。

@BeforeMethod
public void setup() throws InterruptedException
{
MockitoAnnotations.initMocks(this);
SearchResponse sr = new SearchResponse();

SearchRequestBuilder requestBuilder = mock(SearchRequestBuilder.class, RETURNS_DEEP_STUBS);
when(client.prepareSearch(Matchers.any())).thenReturn(requestBuilder);
when(requestBuilder.setQuery(Matchers.any())).thenReturn(requestBuilder);
when(requestBuilder.addFields(Matchers.any())).thenReturn(requestBuilder);
when(requestBuilder.setFrom(Matchers.anyInt())).thenReturn(requestBuilder);
when(requestBuilder.setSize(Matchers.anyInt())).thenReturn(requestBuilder);

MockitoAnnotations.initMocks(this);
SearchResponse sr = new SearchResponse();
Mockito.when(client.prepareSearch(Mockito.anyString())
        .setQuery(Mockito.any(QueryBuilder.class))
        .addFields(Mockito.anyObject())
        .setFrom(Mockito.anyInt())
        .setSize(Mockito.anyInt())
        ).thenReturn(searchRequestBuilder);