[ML] Fixing leaked reference in CCM tests (#137839)

* Fixing leaked reference

* Unmuting tests

Author: jonathan-buttner

muted-tests.yml

@@ -459,9 +459,6 @@ tests:
   method: testSimilarityWithOneDimVector {functionName=v_cosine
     similarityFunction=org.elasticsearch.xpack.esql.expression.function.vector.CosineSimilarity$1@5b068087 elementType=byte}
   issue: https://github.com/elastic/elasticsearch/issues/137778
-- class: org.elasticsearch.xpack.inference.integration.CCMPersistentStorageServiceIT
-  method: testDelete_RemovesCCMConfiguration
-  issue: https://github.com/elastic/elasticsearch/issues/137786
 - class: org.elasticsearch.upgrades.WildcardRollingUpgradeIT
   method: testIndexing {upgradedNodes=0}
   issue: https://github.com/elastic/elasticsearch/issues/137793
@@ -474,12 +471,6 @@ tests:
 - class: org.elasticsearch.upgrades.TextRollingUpgradeIT
   method: testIndexing {upgradedNodes=1}
   issue: https://github.com/elastic/elasticsearch/issues/137796
-- class: org.elasticsearch.xpack.inference.integration.CCMPersistentStorageServiceIT
-  method: testDelete_DoesNotThrow_WhenTheConfigurationDoesNotExist
-  issue: https://github.com/elastic/elasticsearch/issues/137797
-- class: org.elasticsearch.xpack.inference.integration.CCMServiceIT
-  method: testIsEnabled_ReturnsTrue_WhenCCMConfigurationIsPresent
-  issue: https://github.com/elastic/elasticsearch/issues/137798
 - class: org.elasticsearch.xpack.esql.vector.VectorSimilarityFunctionsIT
   method: testSimilarityWithOneDimVector {functionName=v_cosine
     similarityFunction=org.elasticsearch.xpack.esql.expression.function.vector.CosineSimilarity$1@3300f4fd elementType=byte}

x-pack/plugin/inference/src/internalClusterTest/java/org/elasticsearch/xpack/inference/integration/CCMSingleNodeIT.java

@@ -27,6 +27,7 @@
 import java.util.Collection;
 import java.util.Objects;
 
+import static org.elasticsearch.test.hamcrest.ElasticsearchAssertions.assertHitCount;
 import static org.elasticsearch.xpack.inference.services.elastic.ccm.CCMPersistentStorageService.CCM_DOC_ID;
 import static org.hamcrest.Matchers.containsString;
 import static org.hamcrest.Matchers.is;
@@ -77,12 +78,7 @@ public void testStore_OverwritesConfiguration_WhenItAlreadyExists() {
         assertStoreCCMConfiguration();
         assertStoreCCMConfiguration("new_secret");
 
-        var results = client().prepareSearch(CCMIndex.INDEX_PATTERN)
-            .setQuery(QueryBuilders.idsQuery().addIds(CCM_DOC_ID))
-            .execute()
-            .actionGet(TimeValue.THIRTY_SECONDS);
-
-        assertThat(results.getHits().getHits().length, is(1));
+        assertHitCount(client().prepareSearch(CCMIndex.INDEX_PATTERN).setQuery(QueryBuilders.idsQuery().addIds(CCM_DOC_ID)), 1);
     }
 
     public void testGet_ThrowsResourceNotFoundException_WhenCCMIndexDoesNotExist() {