remove printout leftovers in tests

Signed-off-by: Xun Zhang <[email protected]>
opensearch-project · Jan 18, 2024 · a3f1d62 · a3f1d62
1 parent 826b130
commit a3f1d62
Show file tree

Hide file tree

Showing 5 changed files with 1 addition and 6 deletions.
diff --git a/...test/java/org/opensearch/ml/memory/action/conversation/CreateInteractionRequestTests.java b/...test/java/org/opensearch/ml/memory/action/conversation/CreateInteractionRequestTests.java
@@ -108,8 +108,6 @@ public void testFromRestRequest() throws IOException {
             .withContent(new BytesArray(gson.toJson(params)), MediaTypeRegistry.JSON)
             .build();
         CreateInteractionRequest request = CreateInteractionRequest.fromRestRequest(rrequest);
-        System.out.println(request.getConversationId());
-        System.out.println(request.getInput());
 
         assert (request.validate() == null);
         assert (request.getConversationId().equals("cid"));

diff --git a/...c/test/java/org/opensearch/ml/memory/action/conversation/GetInteractionResponseTests.java b/...c/test/java/org/opensearch/ml/memory/action/conversation/GetInteractionResponseTests.java
@@ -73,7 +73,6 @@ public void testToXContent() throws IOException {
         XContentBuilder builder = XContentBuilder.builder(XContentType.JSON.xContent());
         response.toXContent(builder, ToXContent.EMPTY_PARAMS);
         String result = BytesReference.bytes(builder).utf8ToString();
-        System.out.println(result);
         String expected = "{\"memory_id\":\"cid\",\"message_id\":\"iid\",\"create_time\":\""
             + interaction.getCreateTime()
             + "\",\"input\":\"inp\",\"prompt_template\":\"pt\",\"response\":\"rsp\",\"origin\":\"ogn\",\"additional_info\":{\"metadata\":\"some meta\"}}";

diff --git a/...ry/src/test/java/org/opensearch/ml/memory/action/conversation/GetTracesResponseTests.java b/...ry/src/test/java/org/opensearch/ml/memory/action/conversation/GetTracesResponseTests.java
@@ -93,7 +93,6 @@ public void testToXContent_MoreTokens() throws IOException {
         XContentBuilder builder = XContentBuilder.builder(XContentType.JSON.xContent());
         response.toXContent(builder, ToXContent.EMPTY_PARAMS);
         String result = BytesReference.bytes(builder).utf8ToString();
-        System.out.println(result);
         String expected = "{\"traces\":[{\"memory_id\":\"cid\",\"message_id\":\"id0\",\"create_time\":"
             + trace.getCreateTime()
             + ",\"input\":\"input\",\"prompt_template\":\"pt\",\"response\":\"response\",\"origin\":\"origin\",\"additional_info\":{\"metadata\":\"some meta\"},\"parent_message_id\":\"parent_id\",\"trace_number\":1}],\"next_token\":2}";

diff --git a/ml-algorithms/src/test/java/org/opensearch/ml/engine/memory/MLMemoryManagerTests.java b/ml-algorithms/src/test/java/org/opensearch/ml/engine/memory/MLMemoryManagerTests.java
@@ -264,7 +264,6 @@ public void testGetInteractions_NoAccessNoUser_ThenFail() {
         mlMemoryManager.getFinalInteractions("cid", 10, interactionListActionListener);
         ArgumentCaptor<Exception> argCaptor = ArgumentCaptor.forClass(Exception.class);
         verify(interactionListActionListener, times(1)).onFailure(argCaptor.capture());
-        System.out.println(argCaptor.getValue().getMessage());
         assert (argCaptor.getValue().getMessage().equals("User [] does not have access to conversation cid"));
     }
 

diff --git a/plugin/src/main/java/org/opensearch/ml/action/register/TransportRegisterModelAction.java b/plugin/src/main/java/org/opensearch/ml/action/register/TransportRegisterModelAction.java
@@ -299,7 +299,7 @@ private void registerModel(MLRegisterModelInput registerModelInput, ActionListen
                 throw new IllegalArgumentException("URL can't match trusted url regex");
             }
         }
-        System.out.println("registering the model");
+
         boolean isAsync = registerModelInput.getFunctionName() != FunctionName.REMOTE;
         MLTask mlTask = MLTask
             .builder()