[ML] Integrate calls to EIS for preconfigured inference endpoints for ModelRegistry functionality

x-pack/plugin/esql/qa/server/src/main/java/org/elasticsearch/xpack/esql/qa/rest/SemanticMatchTestCase.java

@@ -9,6 +9,7 @@
 
 import org.elasticsearch.client.Request;
 import org.elasticsearch.client.ResponseException;
+import org.elasticsearch.common.Strings;
 import org.elasticsearch.common.settings.Settings;
 import org.elasticsearch.test.rest.ESRestTestCase;
 import org.elasticsearch.xpack.esql.AssertWarnings;
@@ -61,13 +62,15 @@ public void testWithMultipleInferenceIds() throws IOException {
     public void testWithInferenceNotConfigured() {
         assumeTrue("semantic text capability not available", EsqlCapabilities.Cap.SEMANTIC_TEXT_FIELD_CAPS.isEnabled());
 
-        String query = """
-            from test-semantic3
+        var inferenceId = "test-semantic3";
+
+        String query = Strings.format("""
+            from %s
             | where match(semantic_text_field, "something")
-            """;
+            """, inferenceId);
         ResponseException re = expectThrows(ResponseException.class, () -> runEsqlQuery(query));
 
-        assertThat(re.getMessage(), containsString("Inference endpoint not found"));
+        assertThat(re.getMessage(), containsString(Strings.format("Inference endpoint [%s] not found", inferenceId)));
         assertEquals(404, re.getResponse().getStatusLine().getStatusCode());
     }
 

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

@@ -0,0 +1,89 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0; you may not use this file except in compliance with the Elastic License
+ * 2.0.
+ */
+
+package org.elasticsearch.xpack.inference.integration;
+
+import org.elasticsearch.common.settings.Settings;
+import org.elasticsearch.core.TimeValue;
+import org.elasticsearch.inference.Model;
+import org.elasticsearch.inference.TaskType;
+import org.elasticsearch.plugins.Plugin;
+import org.elasticsearch.reindex.ReindexPlugin;
+import org.elasticsearch.test.ESSingleNodeTestCase;
+import org.elasticsearch.test.http.MockWebServer;
+import org.elasticsearch.xpack.inference.LocalStateInferencePlugin;
+import org.elasticsearch.xpack.inference.registry.ModelRegistry;
+import org.elasticsearch.xpack.inference.registry.ModelRegistryTests;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.concurrent.TimeUnit;
+
+import static org.elasticsearch.xpack.inference.external.http.Utils.getUrl;
+import static org.elasticsearch.xpack.inference.integration.ModelRegistryIT.createModel;
+import static org.elasticsearch.xpack.inference.services.elastic.ElasticInferenceServiceSettings.ELASTIC_INFERENCE_SERVICE_URL;
+
+public class ModelRegistryEisBase extends ESSingleNodeTestCase {
+    protected static final TimeValue TIMEOUT = new TimeValue(30, TimeUnit.SECONDS);
+    protected static final MockWebServer webServer = new MockWebServer();
+
+    protected ModelRegistry modelRegistry;
+    private String eisUrl;
+
+    public ModelRegistryEisBase() {}
+
+    public ModelRegistryEisBase(String eisUrl) {
+        this.eisUrl = eisUrl;
+    }
+
+    @BeforeClass
+    public static void init() throws Exception {
+        webServer.start();
+    }
+
+    @AfterClass
+    public static void shutdown() {
+        webServer.close();
+    }
+
+    @Before
+    public void createComponents() {
+        modelRegistry = node().injector().getInstance(ModelRegistry.class);
+        modelRegistry.clearDefaultIds();
+    }
+
+    @Override
+    protected Collection<Class<? extends Plugin>> getPlugins() {
+        return pluginList(ReindexPlugin.class, LocalStateInferencePlugin.class);
+    }
+
+    @Override
+    protected Settings nodeSettings() {
+        return Settings.builder().put(super.nodeSettings()).put(ELASTIC_INFERENCE_SERVICE_URL.getKey(), getEisUrl()).build();
+    }
+
+    private String getEisUrl() {
+        return eisUrl != null ? eisUrl : getUrl(webServer);
+    }
+
+    protected void initializeModels() {
+        var service = "foo";
+        var sparseAndTextEmbeddingModels = new ArrayList<Model>();
+        sparseAndTextEmbeddingModels.add(createModel("sparse-1", TaskType.SPARSE_EMBEDDING, service));
+        sparseAndTextEmbeddingModels.add(createModel("sparse-2", TaskType.SPARSE_EMBEDDING, service));
+        sparseAndTextEmbeddingModels.add(createModel("sparse-3", TaskType.SPARSE_EMBEDDING, service));
+        sparseAndTextEmbeddingModels.add(createModel("embedding-1", TaskType.TEXT_EMBEDDING, service));
+        sparseAndTextEmbeddingModels.add(createModel("embedding-2", TaskType.TEXT_EMBEDDING, service));
+
+        for (var model : sparseAndTextEmbeddingModels) {
+            ModelRegistryTests.assertStoreModel(modelRegistry, model);
+        }
+    }
+}