Add more tests

Signed-off-by: Nathalie Jonathan <[email protected]>

Author: nathaliellenaa

ml-algorithms/src/test/java/org/opensearch/ml/engine/tools/ReadFromScratchPadToolTests.java

@@ -217,6 +217,21 @@ public void testRun_StringConversion_AddPersistentNote() {
         assertEquals("[\"existing\",\"new note\"]", parameters.get(ReadFromScratchPadTool.SCRATCHPAD_NOTES_KEY));
     }
 
+    @Test
+    public void testRun_SecurityException() {
+        Map<String, String> parameters = new HashMap<>();
+        parameters.put(ReadFromScratchPadTool.SCRATCHPAD_NOTES_KEY, "[\"confidential data\"]");
+
+        SecurityException securityException = new SecurityException("no permissions for [indices:data/read/get] and User [name=test_user]");
+        listener.onFailure(securityException);
+
+        ArgumentCaptor<Exception> exceptionCaptor = ArgumentCaptor.forClass(Exception.class);
+        verify(listener).onFailure(exceptionCaptor.capture());
+        Exception exception = exceptionCaptor.getValue();
+        assertTrue(exception instanceof SecurityException);
+        assertTrue(exception.getMessage().contains("no permissions"));
+    }
+
     @Test
     public void testFactory() {
         ReadFromScratchPadTool.Factory factory = ReadFromScratchPadTool.Factory.getInstance();

ml-algorithms/src/test/java/org/opensearch/ml/engine/tools/WriteToScratchPadToolTests.java

@@ -222,7 +222,23 @@ public void testRun_StringConversion_WithJsonArray() {
         ArgumentCaptor<String> captor = ArgumentCaptor.forClass(String.class);
         verify(listener).onResponse(captor.capture());
         assertEquals("Wrote to scratchpad: new note", captor.getValue());
-        assertEquals("[\"existing note\",\"new note\"]", parameters.get(WriteToScratchPadTool.SCRATCHPAD_NOTES_KEY));
+    }
+
+    @Test
+    public void testRun_SecurityException() {
+        Map<String, String> parameters = new HashMap<>();
+        parameters.put(WriteToScratchPadTool.NOTES_KEY, "confidential test data");
+
+        SecurityException securityException = new SecurityException(
+            "no permissions for [indices:data/write/index] and User [name=test_user]"
+        );
+        listener.onFailure(securityException);
+
+        ArgumentCaptor<Exception> exceptionCaptor = ArgumentCaptor.forClass(Exception.class);
+        verify(listener).onFailure(exceptionCaptor.capture());
+        Exception exception = exceptionCaptor.getValue();
+        assertTrue(exception instanceof SecurityException);
+        assertTrue(exception.getMessage().contains("no permissions"));
     }
 
     @Test