apply spotless

Signed-off-by: Jiaping Zeng <[email protected]>

Author: jiapingzeng

common/src/test/java/org/opensearch/ml/common/agui/AGUIInputConverterTest.java

@@ -14,8 +14,6 @@
 import org.opensearch.ml.common.dataset.remote.RemoteInferenceInputDataSet;
 import org.opensearch.ml.common.input.execute.agent.AgentMLInput;
 
-import com.google.gson.JsonObject;
-
 public class AGUIInputConverterTest {
 
     @Test
@@ -246,8 +244,6 @@ public void testConvertFromAGUIInput_WithTemplates() {
         );
     }
 
-
-
     @Test
     public void testConvertFromAGUIInput_WithContext() {
         String aguiInput = "{\n"

common/src/test/java/org/opensearch/ml/common/agui/RunFinishedEventTests.java

@@ -14,15 +14,16 @@
 import org.junit.Test;
 import org.opensearch.common.io.stream.BytesStreamOutput;
 import org.opensearch.core.common.io.stream.StreamInput;
+
 public class RunFinishedEventTests {
 
     @Test
     public void testConstructor() {
         String threadId = "thread_123";
         String runId = "run_456";
-        
+
         RunFinishedEvent event = new RunFinishedEvent(threadId, runId, null);
-        
+
         assertNotNull("Event should not be null", event);
         assertEquals("Type should be RUN_FINISHED", "RUN_FINISHED", event.getType());
         assertEquals("Thread ID should match", threadId, event.getThreadId());
@@ -33,13 +34,13 @@ public void testConstructor() {
     @Test
     public void testSerialization() throws IOException {
         RunFinishedEvent original = new RunFinishedEvent("thread_test", "run_test", null);
-        
+
         BytesStreamOutput output = new BytesStreamOutput();
         original.writeTo(output);
-        
+
         StreamInput input = output.bytes().streamInput();
         RunFinishedEvent deserialized = new RunFinishedEvent(input);
-        
+
         assertEquals("Type should match", original.getType(), deserialized.getType());
         assertEquals("Thread ID should match", original.getThreadId(), deserialized.getThreadId());
         assertEquals("Run ID should match", original.getRunId(), deserialized.getRunId());
@@ -49,9 +50,9 @@ public void testSerialization() throws IOException {
     @Test
     public void testToXContent() throws IOException {
         RunFinishedEvent event = new RunFinishedEvent("thread_xcontent", "run_xcontent", null);
-        
+
         String json = event.toJsonString();
-        
+
         assertNotNull("JSON should not be null", json);
         assertTrue("JSON should contain type", json.contains("\"type\":\"RUN_FINISHED\""));
         assertTrue("JSON should contain threadId", json.contains("\"threadId\":\"thread_xcontent\""));
@@ -62,9 +63,9 @@ public void testToXContent() throws IOException {
     @Test
     public void testToJsonString() {
         RunFinishedEvent event = new RunFinishedEvent("thread_json", "run_json", null);
-        
+
         String json = event.toJsonString();
-        
+
         assertNotNull("JSON string should not be null", json);
         assertTrue("JSON should be valid", json.startsWith("{") && json.endsWith("}"));
     }
@@ -74,9 +75,8 @@ public void testTimestampGeneration() {
         long before = System.currentTimeMillis();
         RunFinishedEvent event = new RunFinishedEvent("thread_time", "run_time", null);
         long after = System.currentTimeMillis();
-        
+
         assertNotNull("Timestamp should be set", event.getTimestamp());
-        assertTrue("Timestamp should be in valid range", 
-            event.getTimestamp() >= before && event.getTimestamp() <= after);
+        assertTrue("Timestamp should be in valid range", event.getTimestamp() >= before && event.getTimestamp() <= after);
     }
 }

common/src/test/java/org/opensearch/ml/common/agui/RunStartedEventTests.java

@@ -14,15 +14,16 @@
 import org.junit.Test;
 import org.opensearch.common.io.stream.BytesStreamOutput;
 import org.opensearch.core.common.io.stream.StreamInput;
+
 public class RunStartedEventTests {
 
     @Test
     public void testConstructor() {
         String threadId = "thread_123";
         String runId = "run_456";
-        
+
         RunStartedEvent event = new RunStartedEvent(threadId, runId);
-        
+
         assertNotNull("Event should not be null", event);
         assertEquals("Type should be RUN_STARTED", "RUN_STARTED", event.getType());
         assertEquals("Thread ID should match", threadId, event.getThreadId());
@@ -34,13 +35,13 @@ public void testConstructor() {
     @Test
     public void testSerialization() throws IOException {
         RunStartedEvent original = new RunStartedEvent("thread_test", "run_test");
-        
+
         BytesStreamOutput output = new BytesStreamOutput();
         original.writeTo(output);
-        
+
         StreamInput input = output.bytes().streamInput();
         RunStartedEvent deserialized = new RunStartedEvent(input);
-        
+
         assertEquals("Type should match", original.getType(), deserialized.getType());
         assertEquals("Thread ID should match", original.getThreadId(), deserialized.getThreadId());
         assertEquals("Run ID should match", original.getRunId(), deserialized.getRunId());
@@ -50,9 +51,9 @@ public void testSerialization() throws IOException {
     @Test
     public void testToXContent() throws IOException {
         RunStartedEvent event = new RunStartedEvent("thread_xcontent", "run_xcontent");
-        
+
         String json = event.toJsonString();
-        
+
         assertNotNull("JSON should not be null", json);
         assertTrue("JSON should contain type", json.contains("\"type\":\"RUN_STARTED\""));
         assertTrue("JSON should contain threadId", json.contains("\"threadId\":\"thread_xcontent\""));
@@ -63,25 +64,24 @@ public void testToXContent() throws IOException {
     @Test
     public void testToJsonString() {
         RunStartedEvent event = new RunStartedEvent("thread_json", "run_json");
-        
+
         String json = event.toJsonString();
-        
+
         assertNotNull("JSON string should not be null", json);
         assertTrue("JSON should be valid", json.startsWith("{") && json.endsWith("}"));
-        assertTrue("JSON should contain all fields", 
-            json.contains("type") && json.contains("threadId") && json.contains("runId"));
+        assertTrue("JSON should contain all fields", json.contains("type") && json.contains("threadId") && json.contains("runId"));
     }
 
     @Test
     public void testTimestampGeneration() throws InterruptedException {
         long before = System.currentTimeMillis();
         Thread.sleep(10); // Small delay to ensure different timestamps
-        
+
         RunStartedEvent event = new RunStartedEvent("thread_time", "run_time");
-        
+
         Thread.sleep(10);
         long after = System.currentTimeMillis();
-        
+
         assertNotNull("Timestamp should be set", event.getTimestamp());
         assertTrue("Timestamp should be after 'before'", event.getTimestamp() >= before);
         assertTrue("Timestamp should be before 'after'", event.getTimestamp() <= after);

common/src/test/java/org/opensearch/ml/common/agui/TextMessageContentEventTests.java

@@ -14,15 +14,16 @@
 import org.junit.Test;
 import org.opensearch.common.io.stream.BytesStreamOutput;
 import org.opensearch.core.common.io.stream.StreamInput;
+
 public class TextMessageContentEventTests {
 
     @Test
     public void testConstructor() {
         String messageId = "msg_123";
         String delta = "Hello, ";
-        
+
         TextMessageContentEvent event = new TextMessageContentEvent(messageId, delta);
-        
+
         assertNotNull("Event should not be null", event);
         assertEquals("Type should be TEXT_MESSAGE_CONTENT", "TEXT_MESSAGE_CONTENT", event.getType());
         assertEquals("Message ID should match", messageId, event.getMessageId());
@@ -33,13 +34,13 @@ public void testConstructor() {
     @Test
     public void testSerialization() throws IOException {
         TextMessageContentEvent original = new TextMessageContentEvent("msg_test", "world!");
-        
+
         BytesStreamOutput output = new BytesStreamOutput();
         original.writeTo(output);
-        
+
         StreamInput input = output.bytes().streamInput();
         TextMessageContentEvent deserialized = new TextMessageContentEvent(input);
-        
+
         assertEquals("Type should match", original.getType(), deserialized.getType());
         assertEquals("Message ID should match", original.getMessageId(), deserialized.getMessageId());
         assertEquals("Delta should match", original.getDelta(), deserialized.getDelta());
@@ -49,9 +50,9 @@ public void testSerialization() throws IOException {
     @Test
     public void testToXContent() throws IOException {
         TextMessageContentEvent event = new TextMessageContentEvent("msg_xcontent", "test content");
-        
+
         String json = event.toJsonString();
-        
+
         assertNotNull("JSON should not be null", json);
         assertTrue("JSON should contain type", json.contains("\"type\":\"TEXT_MESSAGE_CONTENT\""));
         assertTrue("JSON should contain messageId", json.contains("\"messageId\":\"msg_xcontent\""));
@@ -62,9 +63,9 @@ public void testToXContent() throws IOException {
     @Test
     public void testToJsonString() {
         TextMessageContentEvent event = new TextMessageContentEvent("msg_json", "content");
-        
+
         String json = event.toJsonString();
-        
+
         assertNotNull("JSON string should not be null", json);
         assertTrue("JSON should be valid", json.startsWith("{") && json.endsWith("}"));
     }
@@ -74,18 +75,17 @@ public void testTimestampGeneration() {
         long before = System.currentTimeMillis();
         TextMessageContentEvent event = new TextMessageContentEvent("msg_time", "delta");
         long after = System.currentTimeMillis();
-        
+
         assertNotNull("Timestamp should be set", event.getTimestamp());
-        assertTrue("Timestamp should be in valid range", 
-            event.getTimestamp() >= before && event.getTimestamp() <= after);
+        assertTrue("Timestamp should be in valid range", event.getTimestamp() >= before && event.getTimestamp() <= after);
     }
 
     @Test
     public void testEmptyDelta() {
         TextMessageContentEvent event = new TextMessageContentEvent("msg_empty", "");
-        
+
         assertEquals("Empty delta should be preserved", "", event.getDelta());
-        
+
         String json = event.toJsonString();
         assertTrue("JSON should contain empty delta", json.contains("\"delta\":\"\""));
     }
@@ -94,7 +94,7 @@ public void testEmptyDelta() {
     public void testSpecialCharactersInDelta() {
         String specialChars = "Hello \"world\" with\nnewlines\tand\ttabs";
         TextMessageContentEvent event = new TextMessageContentEvent("msg_special", specialChars);
-        
+
         assertEquals("Special characters should be preserved", specialChars, event.getDelta());
     }
 }

common/src/test/java/org/opensearch/ml/common/agui/TextMessageStartEventTests.java

@@ -14,15 +14,16 @@
 import org.junit.Test;
 import org.opensearch.common.io.stream.BytesStreamOutput;
 import org.opensearch.core.common.io.stream.StreamInput;
+
 public class TextMessageStartEventTests {
 
     @Test
     public void testConstructor() {
         String messageId = "msg_123";
         String role = "assistant";
-        
+
         TextMessageStartEvent event = new TextMessageStartEvent(messageId, role);
-        
+
         assertNotNull("Event should not be null", event);
         assertEquals("Type should be TEXT_MESSAGE_START", "TEXT_MESSAGE_START", event.getType());
         assertEquals("Message ID should match", messageId, event.getMessageId());
@@ -33,13 +34,13 @@ public void testConstructor() {
     @Test
     public void testSerialization() throws IOException {
         TextMessageStartEvent original = new TextMessageStartEvent("msg_test", "assistant");
-        
+
         BytesStreamOutput output = new BytesStreamOutput();
         original.writeTo(output);
-        
+
         StreamInput input = output.bytes().streamInput();
         TextMessageStartEvent deserialized = new TextMessageStartEvent(input);
-        
+
         assertEquals("Type should match", original.getType(), deserialized.getType());
         assertEquals("Message ID should match", original.getMessageId(), deserialized.getMessageId());
         assertEquals("Role should match", original.getRole(), deserialized.getRole());
@@ -49,9 +50,9 @@ public void testSerialization() throws IOException {
     @Test
     public void testToXContent() throws IOException {
         TextMessageStartEvent event = new TextMessageStartEvent("msg_xcontent", "assistant");
-        
+
         String json = event.toJsonString();
-        
+
         assertNotNull("JSON should not be null", json);
         assertTrue("JSON should contain type", json.contains("\"type\":\"TEXT_MESSAGE_START\""));
         assertTrue("JSON should contain messageId", json.contains("\"messageId\":\"msg_xcontent\""));
@@ -62,23 +63,21 @@ public void testToXContent() throws IOException {
     @Test
     public void testToJsonString() {
         TextMessageStartEvent event = new TextMessageStartEvent("msg_json", "user");
-        
+
         String json = event.toJsonString();
-        
+
         assertNotNull("JSON string should not be null", json);
         assertTrue("JSON should be valid", json.startsWith("{") && json.endsWith("}"));
-        assertTrue("JSON should contain messageId and role", 
-            json.contains("messageId") && json.contains("role"));
+        assertTrue("JSON should contain messageId and role", json.contains("messageId") && json.contains("role"));
     }
 
     @Test
     public void testTimestampGeneration() {
         long before = System.currentTimeMillis();
         TextMessageStartEvent event = new TextMessageStartEvent("msg_time", "assistant");
         long after = System.currentTimeMillis();
-        
+
         assertNotNull("Timestamp should be set", event.getTimestamp());
-        assertTrue("Timestamp should be in valid range", 
-            event.getTimestamp() >= before && event.getTimestamp() <= after);
+        assertTrue("Timestamp should be in valid range", event.getTimestamp() >= before && event.getTimestamp() <= after);
     }
 }