fix test

Author: fang-xing-esql

x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/optimizer/rules/physical/local/ReplaceRoundToWithQueryAndTagsTests.java

@@ -132,6 +132,7 @@ public void testDateTruncBucketTransformToQueryAndTags() {
                 null
             );
             verifyQueryAndTags(expectedQueryBuilderAndTags, queryBuilderAndTags);
+            assertThrows(UnsupportedOperationException.class, esQueryExec::query);
         }
     }
 
@@ -163,7 +164,12 @@ public void testDateTruncNotTransformToQueryAndTags() {
             assertEquals(4, roundTo.points().size());
             fieldExtractExec = as(evalExec.child(), FieldExtractExec.class);
             EsQueryExec esQueryExec = as(fieldExtractExec.child(), EsQueryExec.class);
-            assertNull(esQueryExec.queryBuilderAndTags());
+            List<EsQueryExec.QueryBuilderAndTags> queryBuilderAndTags = esQueryExec.queryBuilderAndTags();
+            assertEquals(1, queryBuilderAndTags.size());
+            EsQueryExec.QueryBuilderAndTags queryBuilder = queryBuilderAndTags.get(0);
+            assertNull(queryBuilder.query());
+            assertTrue(queryBuilder.tags().isEmpty());
+            assertNull(esQueryExec.query());
         }
     }
 
@@ -208,6 +214,7 @@ public void testRoundToTransformToQueryAndTags() {
                 null
             );
             verifyQueryAndTags(expectedQueryBuilderAndTags, queryBuilderAndTags);
+            assertThrows(UnsupportedOperationException.class, esQueryExec::query);
         }
     }
 
@@ -254,6 +261,7 @@ public void testDateTruncBucketTransformToQueryAndTagsWithOtherPushdownFunctions
                     mainQueryBuilder
                 );
                 verifyQueryAndTags(expectedQueryBuilderAndTags, queryBuilderAndTags);
+                assertThrows(UnsupportedOperationException.class, esQueryExec::query);
             }
         }
     }