Add more UTs

Signed-off-by: zane-neo <zaniu@amazon.com>
zane-neo · Apr 12, 2024 · b3aed63 · b3aed63
1 parent f0f3a5b
commit b3aed63
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ml-algorithms/src/test/java/org/opensearch/ml/engine/algorithms/remote/RemoteModelTest.java b/ml-algorithms/src/test/java/org/opensearch/ml/engine/algorithms/remote/RemoteModelTest.java
@@ -65,6 +65,13 @@ public void predict_ModelNotDeployed() {
         remoteModel.predict(mlInput, mlModel);
     }
 
+    @Test
+    public void test_predict_throw_IllegalStateException() {
+        exceptionRule.expect(IllegalStateException.class);
+        exceptionRule.expectMessage("Method is not implemented");
+        remoteModel.predict(mlInput);
+    }
+
     @Test
     public void predict_NullConnectorExecutor() {
         ActionListener<MLTaskResponse> actionListener = mock(ActionListener.class);