add more tests

Signed-off-by: Dhrubo Saha <[email protected]>
opensearch-project · Oct 3, 2023 · f99a063 · f99a063
1 parent a9c1ec8
commit f99a063
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/plugin/src/test/java/org/opensearch/ml/stats/MLModelStatsTests.java b/plugin/src/test/java/org/opensearch/ml/stats/MLModelStatsTests.java
@@ -51,6 +51,15 @@ public void testSerializationDeserialization() throws IOException {
         assertEquals(failureCount, parsedMLActionStats.getActionStat(ML_ACTION_FAILURE_COUNT));
     }
 
+    public void testEmptySerializationDeserialization() throws IOException {
+
+        Map<ActionName, MLActionStats> modelStats = new HashMap<>();
+        MLModelStats mlModelEmptyStats = new MLModelStats(modelStats);
+        BytesStreamOutput output = new BytesStreamOutput();
+        mlModelEmptyStats.writeTo(output);
+        assertEquals(0, output.bytes().length());
+    }
+
     public void testToXContent() throws IOException {
         XContentBuilder builder = XContentBuilder.builder(XContentType.JSON.xContent());
         builder.startObject();