test: exception when batchSize is zero or negative integer

opensearch-project · Nov 17, 2024 · 90b81ab · 90b81ab
1 parent 7843e43
commit 90b81ab
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/...ent/src/test/java/org/opensearch/client/opensearch/ingest/TextEmbeddingProcessorTest.java b/...ent/src/test/java/org/opensearch/client/opensearch/ingest/TextEmbeddingProcessorTest.java
@@ -73,4 +73,21 @@ public void testJsonRoundtripWithoutBatchSize() {
         assertEquals("some-tag", deserialized.tag());
         assertNull(deserialized.batchSize());
     }
+
+    @Test
+    public void testInvalidBatchSizeThrowsException() {
+        IllegalArgumentException exceptionWhenBatchSizeIsZero = assertThrows(IllegalArgumentException.class, () -> {
+            new Processor.Builder().textEmbedding(
+                    baseTextEmbeddingProcessor().batchSize(0).build()
+            ).build();
+        });
+        assertEquals("batchSize must be a positive integer", exceptionWhenBatchSizeIsZero.getMessage());
+
+        IllegalArgumentException exceptionWhenBatchSizeIsNegative = assertThrows(IllegalArgumentException.class, () -> {
+            new Processor.Builder().textEmbedding(
+                    baseTextEmbeddingProcessor().batchSize(-1).build()
+            ).build();
+        });
+        assertEquals("batchSize must be a positive integer", exceptionWhenBatchSizeIsNegative.getMessage());
+    }
 }