[GitHub] drill pull request #1045: DRILL-5730 Test Mocking Improvements

[paul-rogers]

Github user paul-rogers commented on a diff in the pull request:

    https://github.com/apache/drill/pull/1045#discussion_r162246538
  
    --- Diff: 
exec/java-exec/src/main/java/org/apache/drill/exec/work/batch/BaseRawBatchBuffer.java
 ---
    @@ -184,7 +184,7 @@ public RawFragmentBatch getNext() throws IOException {
           return null;
         }
     
    -    if (context.isOverMemoryLimit()) {
    +    if (context.getAllocator().isOverLimit()) {
    --- End diff --
    
    This one is fine. We must expose an allocator even for testing.


---