From 45bd1fbd1e8fd22d67c1d290b3ce6a925d5851da Mon Sep 17 00:00:00 2001
From: Heemin Kim <heemin@amazon.com>
Date: Fri, 28 Jul 2023 14:41:13 -0700
Subject: [PATCH] Refactor XContentFactory to MediaTypeRegistry (#374)

Signed-off-by: Heemin Kim <heemin@amazon.com>
---
 .../stats/upload/TotalUploadStatsTests.java        | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/test/java/org/opensearch/geospatial/stats/upload/TotalUploadStatsTests.java b/src/test/java/org/opensearch/geospatial/stats/upload/TotalUploadStatsTests.java
index 021ff88b..2775e9b7 100644
--- a/src/test/java/org/opensearch/geospatial/stats/upload/TotalUploadStatsTests.java
+++ b/src/test/java/org/opensearch/geospatial/stats/upload/TotalUploadStatsTests.java
@@ -13,8 +13,8 @@
 import java.util.List;
 
 import org.opensearch.common.Strings;
-import org.opensearch.common.xcontent.XContentFactory;
 import org.opensearch.common.xcontent.XContentType;
+import org.opensearch.core.xcontent.MediaTypeRegistry;
 import org.opensearch.core.xcontent.ToXContent;
 import org.opensearch.core.xcontent.XContentBuilder;
 import org.opensearch.test.OpenSearchTestCase;
@@ -43,7 +43,7 @@ public void testEmptyUploadStats() {
 
     public void testToXContentWithEmptyUploadStats() throws IOException {
         TotalUploadStats totalUploadStats = new TotalUploadStats(Collections.emptyList());
-        XContentBuilder contentBuilder = XContentFactory.contentBuilder(XContentType.JSON);
+        XContentBuilder contentBuilder = MediaTypeRegistry.contentBuilder(XContentType.JSON);
         contentBuilder.startObject();
         totalUploadStats.toXContent(contentBuilder, ToXContent.EMPTY_PARAMS);
         contentBuilder.endObject();
@@ -58,7 +58,7 @@ public void testToXContentWithRequestAPICount() throws IOException {
         long expectedSum = INIT;
         expectedSum += randomUploadStats.stream().mapToLong(UploadStats::getTotalAPICount).sum();
         TotalUploadStats totalUploadStats = new TotalUploadStats(randomUploadStats);
-        XContentBuilder contentBuilder = XContentFactory.contentBuilder(XContentType.JSON);
+        XContentBuilder contentBuilder = MediaTypeRegistry.contentBuilder(XContentType.JSON);
         contentBuilder.startObject();
         totalUploadStats.toXContent(contentBuilder, ToXContent.EMPTY_PARAMS);
         contentBuilder.endObject();
@@ -74,7 +74,7 @@ public void testToXContentWithUploadCount() throws IOException {
             expectedSum += stats.getMetrics().stream().mapToLong(UploadMetric::getUploadCount).sum();
         }
         TotalUploadStats totalUploadStats = new TotalUploadStats(randomUploadStats);
-        XContentBuilder contentBuilder = XContentFactory.contentBuilder(XContentType.JSON);
+        XContentBuilder contentBuilder = MediaTypeRegistry.contentBuilder(XContentType.JSON);
         contentBuilder.startObject();
         totalUploadStats.toXContent(contentBuilder, ToXContent.EMPTY_PARAMS);
         contentBuilder.endObject();
@@ -90,7 +90,7 @@ public void testToXContentWithSuccessCount() throws IOException {
             expectedSum += stats.getMetrics().stream().mapToLong(UploadMetric::getSuccessCount).sum();
         }
         TotalUploadStats totalUploadStats = new TotalUploadStats(randomUploadStats);
-        XContentBuilder contentBuilder = XContentFactory.contentBuilder(XContentType.JSON);
+        XContentBuilder contentBuilder = MediaTypeRegistry.contentBuilder(XContentType.JSON);
         contentBuilder.startObject();
         totalUploadStats.toXContent(contentBuilder, ToXContent.EMPTY_PARAMS);
         contentBuilder.endObject();
@@ -106,7 +106,7 @@ public void testToXContentWithFailedCount() throws IOException {
             expectedSum += stats.getMetrics().stream().mapToLong(UploadMetric::getFailedCount).sum();
         }
         TotalUploadStats totalUploadStats = new TotalUploadStats(randomUploadStats);
-        XContentBuilder contentBuilder = XContentFactory.contentBuilder(XContentType.JSON);
+        XContentBuilder contentBuilder = MediaTypeRegistry.contentBuilder(XContentType.JSON);
         contentBuilder.startObject();
         totalUploadStats.toXContent(contentBuilder, ToXContent.EMPTY_PARAMS);
         contentBuilder.endObject();
@@ -122,7 +122,7 @@ public void testToXContentWithDuration() throws IOException {
             expectedSum += stats.getMetrics().stream().mapToLong(UploadMetric::getDuration).sum();
         }
         TotalUploadStats totalUploadStats = new TotalUploadStats(randomUploadStats);
-        XContentBuilder contentBuilder = XContentFactory.contentBuilder(XContentType.JSON);
+        XContentBuilder contentBuilder = MediaTypeRegistry.contentBuilder(XContentType.JSON);
         contentBuilder.startObject();
         totalUploadStats.toXContent(contentBuilder, ToXContent.EMPTY_PARAMS);
         contentBuilder.endObject();