fixing SamplingServiceTests

Author: masseyke

server/src/test/java/org/elasticsearch/ingest/SamplingServiceTests.java

@@ -59,6 +59,7 @@ public void testMaybeSample() {
         ProjectMetadata.Builder projectBuilder = ProjectMetadata.builder(ProjectId.DEFAULT);
         final ProjectId projectId = projectBuilder.getId();
         ProjectMetadata projectMetadata = projectBuilder.build();
+        ClusterState originalClusterState = ClusterState.builder(ClusterState.EMPTY_STATE).putProjectMetadata(projectMetadata).build();
         Map<String, Object> inputRawDocSource = randomMap(1, 100, () -> Tuple.tuple(randomAlphaOfLength(10), randomAlphaOfLength(10)));
         final IndexRequest indexRequest = new IndexRequest(indexName).id("_id").source(inputRawDocSource);
         samplingService.maybeSample(projectMetadata, indexRequest);
@@ -74,6 +75,26 @@ public void testMaybeSample() {
                 )
             );
         projectMetadata = projectBuilder.build();
+        {
+            /*
+             * First we ingest some docs without notifying samplingService of the cluster state change. It will have cached the fact that
+             * there is no config for this index, and so it will not store any samples.
+             */
+            int docsToSample = randomIntBetween(1, maxSize);
+            for (int i = 0; i < docsToSample; i++) {
+                samplingService.maybeSample(projectMetadata, indexRequest);
+            }
+            List<SamplingService.RawDocument> sample = samplingService.getLocalSample(projectId, indexName);
+            assertThat(sample, empty());
+        }
+        // Now we notify samplingService that the cluster state has changed, and it will pick up the new sampling config
+        samplingService.clusterChanged(
+            new ClusterChangedEvent(
+                "test",
+                ClusterState.builder(ClusterState.EMPTY_STATE).putProjectMetadata(projectMetadata).build(),
+                originalClusterState
+            )
+        );
         int docsToSample = randomIntBetween(1, maxSize);
         for (int i = 0; i < docsToSample; i++) {
             samplingService.maybeSample(projectMetadata, indexRequest);