Add UT

Signed-off-by: b4sjoo <[email protected]>
b4sjoo · Dec 19, 2024 · b639ffc · b639ffc
1 parent 3d54eb5
commit b639ffc
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/plugin/src/test/java/org/opensearch/ml/utils/MLNodeUtilsTests.java b/plugin/src/test/java/org/opensearch/ml/utils/MLNodeUtilsTests.java
@@ -66,6 +66,31 @@ public void testValidateSchema() throws IOException {
         MLNodeUtils.validateSchema(schema, json);
     }
 
+    @Test
+    public void testValidateEmbeddingInputWithRemoteSchema() throws IOException {
+        String schema = "{\n"
+                + "    \"type\": \"object\",\n"
+                + "    \"properties\": {\n"
+                + "        \"parameters\": {\n"
+                + "            \"type\": \"object\",\n"
+                + "            \"properties\": {\n"
+                + "                \"texts\": {\n"
+                + "                    \"type\": \"array\",\n"
+                + "                    \"items\": {\n"
+                + "                        \"type\": \"string\"\n"
+                + "                    }\n"
+                + "                }\n"
+                + "            },\n"
+                + "            \"required\": [\n"
+                + "                \"texts\"\n"
+                + "            ]\n"
+                + "        }\n"
+                + "    }\n"
+                + "}";
+        String json = "{\"text_docs\":[ \"today is sunny\", \"today is sunny\"]}";
+        MLNodeUtils.validateSchema(schema, json);
+    }
+
     @Test
     public void testProcessRemoteInferenceInputDataSetParametersValueNoParameters() throws IOException {
         String json = "{\"key1\":\"foo\",\"key2\":123,\"key3\":true}";