Use test ML model server in xpack rest tests

Author: jan-elastic

x-pack/plugin/build.gradle

@@ -18,6 +18,7 @@ base {
 }
 
 dependencies {
+  testImplementation(testArtifact(project(':x-pack:plugin:inference:qa:inference-service-tests'), "javaRestTest"))
   testImplementation project(xpackModule('core'))
   testImplementation(testArtifact(project(xpackModule('core'))))
   testImplementation(testArtifact(project(":x-pack:plugin:security:qa:service-account"), "javaRestTest"))

x-pack/plugin/inference/qa/inference-service-tests/build.gradle

@@ -1,4 +1,5 @@
 apply plugin: 'elasticsearch.internal-java-rest-test'
+apply plugin: 'elasticsearch.internal-test-artifact'
 
 dependencies {
   javaRestTestImplementation project(path: xpackModule('core'))

x-pack/plugin/src/test/java/org/elasticsearch/xpack/test/rest/AbstractXPackRestTest.java

@@ -37,6 +37,8 @@
 import java.util.concurrent.atomic.AtomicReference;
 import java.util.function.Predicate;
 import java.util.function.Supplier;
+import org.elasticsearch.xpack.inference.MlModelServer;
+import org.junit.ClassRule;
 
 import static java.util.Collections.emptyList;
 import static java.util.Collections.singletonMap;
@@ -50,6 +52,14 @@ public abstract class AbstractXPackRestTest extends ESClientYamlSuiteTestCase {
         SecuritySettingsSourceField.TEST_PASSWORD_SECURE_STRING
     );
 
+    @ClassRule
+    public static MlModelServer mlModelServer = new MlModelServer();
+
+    @Before
+    public void setMlModelRepository() throws IOException {
+        assertOK(mlModelServer.setMlModelRepository(client()));
+    }
+
     public AbstractXPackRestTest(ClientYamlTestCandidate testCandidate) {
         super(testCandidate);
     }