Check self references in metric agg after last doc collection (elasti…

…c#33593) (elastic#34001)

* Check self references in metric agg after last doc collection (elastic#33593)

* Revert 0aff5a3 (elastic#33593)

* Check self refs in metric agg only once in post collection hook (elastic#33593)

* Remove unnecessary mocking (elastic#33593)

server/src/main/java/org/elasticsearch/search/aggregations/metrics/ScriptedMetricAggregator.java

@@ -80,7 +80,6 @@ public void collect(int doc, long bucket) throws IOException {
 
                 leafMapScript.setDocument(doc);
                 leafMapScript.execute();
-                CollectionUtils.ensureNoSelfReferences(aggState, "Scripted metric aggs map script");
             }
         };
     }
@@ -103,4 +102,10 @@ public InternalAggregation buildEmptyAggregation() {
         return new InternalScriptedMetric(name, null, reduceScript, pipelineAggregators(), metaData());
     }
 
+    @Override
+    protected void doPostCollection() throws IOException {
+        CollectionUtils.ensureNoSelfReferences(aggState, "Scripted metric aggs map script");
+
+        super.doPostCollection();
+    }
 }