diff --git a/web-services/query/src/main/java/datawave/webservice/query/runner/RunningQuery.java b/web-services/query/src/main/java/datawave/webservice/query/runner/RunningQuery.java index d5b4d2ef269..ed88333da2e 100644 --- a/web-services/query/src/main/java/datawave/webservice/query/runner/RunningQuery.java +++ b/web-services/query/src/main/java/datawave/webservice/query/runner/RunningQuery.java @@ -559,7 +559,7 @@ public ResultsPage next() throws Exception { log.info("Returning page of results"); // we have results! // we also indicate whether we returned less than the requested page size in the response - return new ResultsPage(resultList, ((iter.hasNext() && numResults < this.maxResults && currentPageCount < maxPageSize) || hitIntermediateResult + return new ResultsPage(resultList, ((hasNext.get() > 0 && numResults < this.maxResults && currentPageCount < maxPageSize) || hitIntermediateResult || hitShortCircuitForLongRunningQuery) ? ResultsPage.Status.PARTIAL : ResultsPage.Status.COMPLETE); } else { // we have no results. Let us determine whether we are done or not. diff --git a/web-services/query/src/test/java/datawave/webservice/query/runner/ExtendedRunningQueryTest.java b/web-services/query/src/test/java/datawave/webservice/query/runner/ExtendedRunningQueryTest.java index 5ed1fbc10ad..b8feb02d61b 100644 --- a/web-services/query/src/test/java/datawave/webservice/query/runner/ExtendedRunningQueryTest.java +++ b/web-services/query/src/test/java/datawave/webservice/query/runner/ExtendedRunningQueryTest.java @@ -173,7 +173,6 @@ public void testNext_HappyPathUsingDeprecatedConstructor() throws Exception { expect(this.transformIterator.next()).andReturn(iterator.next()); expect(this.transformIterator.getTransformer()).andReturn(transformer); } - expect(this.transformIterator.hasNext()).andReturn(iterator.hasNext()); expect(this.query.getPagesize()).andReturn(pageSize).anyTimes(); expect(this.queryLogic.getMaxPageSize()).andReturn(maxPageSize).anyTimes(); expect(this.queryLogic.getPageByteTrigger()).andReturn(pageByteTrigger).anyTimes(); @@ -266,7 +265,6 @@ public void testNextMaxResults_HappyPathUsingDeprecatedConstructor() throws Exce expect(this.transformIterator.next()).andReturn(iterator.next()); count++; } - expect(this.transformIterator.hasNext()).andReturn(iterator.hasNext()); expect(this.transformIterator.getTransformer()).andReturn(transformer).times(count); expect(this.query.getPagesize()).andReturn(pageSize).anyTimes(); @@ -481,7 +479,6 @@ public void testNextWithDnResultLimit_HappyPathUsingDeprecatedConstructor() thro expect(this.transformIterator.next()).andReturn(iterator.next()); count++; } - expect(this.transformIterator.hasNext()).andReturn(iterator.hasNext()); // now that the results thread is separate from the running query thread, we could continue getting stuff expect(this.transformIterator.getTransformer()).andReturn(transformer).anyTimes(); expect(this.transformIterator.hasNext()).andReturn(iterator.hasNext()).anyTimes();