Included both before and before test

Author: mridula-s109

x-pack/plugin/inference/qa/inference-service-tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/SemanticTextEISDefaultIT.java

@@ -7,6 +7,8 @@
 
 import org.elasticsearch.common.settings.Settings;
 import org.elasticsearch.common.xcontent.support.XContentMapValues;
+import org.junit.Before;
+import org.junit.BeforeClass;
 
 import java.io.IOException;
 import java.util.Map;
@@ -20,6 +22,28 @@
  */
 public class SemanticTextEISDefaultIT extends BaseMockEISAuthServerTest {
 
+    @Before
+    public void setUp() throws Exception {
+        super.setUp();
+        // Ensure the mock EIS server has an authorized response ready before each test
+        mockEISServer.enqueueAuthorizeAllModelsResponse();
+    }
+
+    /**
+     * This is done before the class because I've run into issues where another class that extends {@link BaseMockEISAuthServerTest}
+     * results in an authorization response not being queued up for the new Elasticsearch Node in time. When the node starts up, it
+     * retrieves authorization. If the request isn't queued up when that happens the tests will fail. From my testing locally it seems
+     * like the base class's static functionality to queue a response is only done once and not for each subclass.
+     *
+     * My understanding is that the @Before will be run after the node starts up and wouldn't be sufficient to handle
+     * this scenario. That is why this needs to be @BeforeClass.
+     */
+    @BeforeClass
+    public static void init() {
+        // Ensure the mock EIS server has an authorized response ready
+        mockEISServer.enqueueAuthorizeAllModelsResponse();
+    }
+
     public void testDefaultInferenceIdForSemanticText() throws IOException {
         String indexName = "semantic-index";
         String mapping = """