re #1229: Updated to expect subPlan additions.

NationalSecurityAgency · Jul 7, 2023 · 9de1f22 · 9de1f22
1 parent f3f6c2d
commit 9de1f22
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...house/query-core/src/test/java/datawave/query/tables/async/event/VisitorFunctionTest.java b/...house/query-core/src/test/java/datawave/query/tables/async/event/VisitorFunctionTest.java
@@ -91,6 +91,7 @@ public void underTermThresholdTest() throws IOException, TableNotFoundException,
         Query mockQuery = createMock(Query.class);
         config.setQuery(mockQuery);
         EasyMock.expect(mockQuery.getId()).andReturn(new UUID(0, 0)).anyTimes();
+        metric.addSubPlan(EasyMock.eq("20210101_0"), EasyMock.anyString());
 
         // set thresholds
         config.setFinalMaxTermThreshold(2);
@@ -123,6 +124,7 @@ public void overTermThresholdTest() throws IOException, TableNotFoundException,
         config.setQuery(mockQuery);
         EasyMock.expect(mockQuery.getId()).andReturn(new UUID(0, 0)).anyTimes();
         EasyMock.expect(mockQuery.duplicate("testQuery1")).andReturn(mockQuery).anyTimes();
+        metric.addSubPlan(EasyMock.eq("20210101_0"), EasyMock.anyString());
 
         // set thresholds
         config.setFinalMaxTermThreshold(1);
@@ -215,6 +217,7 @@ public void overTermThresholdAfterFirstReductionOverrideSecondTest() throws IOEx
         EasyMock.expect(mockQuery.getId()).andReturn(new UUID(0, 0)).anyTimes();
         EasyMock.expect(mockQuery.getQueryName()).andReturn("testQuery1").anyTimes();
         EasyMock.expect(mockQuery.duplicate("testQuery1")).andReturn(mockQuery).anyTimes();
+        metric.addSubPlan(EasyMock.eq("20210101_0"), EasyMock.anyString());
 
         // set thresholds
         config.setFinalMaxTermThreshold(5);
@@ -262,6 +265,7 @@ public void rangeOverTermThresholdTest() throws IOException, TableNotFoundExcept
         config.setQuery(mockQuery);
         EasyMock.expect(mockQuery.getId()).andReturn(new UUID(0, 0)).anyTimes();
         EasyMock.expect(mockQuery.duplicate("testQuery1")).andReturn(mockQuery).anyTimes();
+        metric.addSubPlan(EasyMock.eq("20210101_0"), EasyMock.anyString());
 
         // set thresholds
         config.setFinalMaxTermThreshold(1);