Model Profile Test (#748)

(cherry picked from commit ce3d747)

src/test/java/org/opensearch/ad/model/ModelProfileTests.java

@@ -0,0 +1,56 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * The OpenSearch Contributors require contributions made to
+ * this file be licensed under the Apache-2.0 license or a
+ * compatible open source license.
+ *
+ * Modifications Copyright OpenSearch Contributors. See
+ * GitHub history for details.
+ */
+
+package org.opensearch.ad.model;
+
+import static org.opensearch.common.xcontent.XContentFactory.jsonBuilder;
+
+import java.io.IOException;
+
+import org.opensearch.ad.AbstractADTest;
+import org.opensearch.ad.constant.CommonName;
+import org.opensearch.common.Strings;
+import org.opensearch.common.xcontent.ToXContent;
+import org.opensearch.common.xcontent.XContentBuilder;
+
+import test.org.opensearch.ad.util.JsonDeserializer;
+
+public class ModelProfileTests extends AbstractADTest {
+
+    public void testToXContent() throws IOException {
+        ModelProfile profile1 = new ModelProfile(
+            randomAlphaOfLength(5),
+            Entity.createSingleAttributeEntity(randomAlphaOfLength(5), randomAlphaOfLength(5)),
+            0
+        );
+        XContentBuilder builder = getBuilder(profile1);
+        String json = Strings.toString(builder);
+        assertTrue(JsonDeserializer.hasChildNode(json, CommonName.ENTITY_KEY));
+        assertFalse(JsonDeserializer.hasChildNode(json, CommonName.MODEL_SIZE_IN_BYTES));
+
+        ModelProfile profile2 = new ModelProfile(randomAlphaOfLength(5), null, 1);
+
+        builder = getBuilder(profile2);
+        json = Strings.toString(builder);
+
+        assertFalse(JsonDeserializer.hasChildNode(json, CommonName.ENTITY_KEY));
+        assertTrue(JsonDeserializer.hasChildNode(json, CommonName.MODEL_SIZE_IN_BYTES));
+
+    }
+
+    private XContentBuilder getBuilder(ModelProfile profile) throws IOException {
+        XContentBuilder builder = jsonBuilder();
+        builder.startObject();
+        profile.toXContent(builder, ToXContent.EMPTY_PARAMS);
+        builder.endObject();
+        return builder;
+    }
+}