diff --git a/warehouse/query-core/src/test/java/datawave/query/tables/async/event/VisitorFunctionTest.java b/warehouse/query-core/src/test/java/datawave/query/tables/async/event/VisitorFunctionTest.java index 8e3a2661ce..6f550b0107 100644 --- a/warehouse/query-core/src/test/java/datawave/query/tables/async/event/VisitorFunctionTest.java +++ b/warehouse/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);