updated tests to return name

Signed-off-by: Matthew Wells <[email protected]>
opensearch-project · Jun 28, 2023 · 3e7d814 · 3e7d814
1 parent 7de27e1
commit 3e7d814
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/integ-test/src/test/java/org/opensearch/sql/sql/CsvFormatIT.java b/integ-test/src/test/java/org/opensearch/sql/sql/CsvFormatIT.java
@@ -12,6 +12,9 @@
 import java.io.IOException;
 import java.util.Locale;
 
+import org.apache.http.Header;
+import org.apache.http.HeaderElement;
+import org.apache.http.ParseException;
 import org.junit.Test;
 import org.opensearch.client.Request;
 import org.opensearch.client.Response;
@@ -64,6 +67,6 @@ public void contentHeaderTest() throws IOException {
 
     Response response = client().performRequest(sqlRequest);
 
-    assertEquals(response.getEntity().getContentType(), "content-type: " + CONTENT_TYPE);
+    assertEquals(response.getEntity().getContentType().getName(), CONTENT_TYPE);
   }
 }
diff --git a/integ-test/src/test/java/org/opensearch/sql/sql/RawFormatIT.java b/integ-test/src/test/java/org/opensearch/sql/sql/RawFormatIT.java
@@ -49,6 +49,6 @@ public void contentHeaderTest() throws IOException {
 
     Response response = client().performRequest(sqlRequest);
 
-    assertEquals(response.getEntity().getContentType(), "content-type: " + CONTENT_TYPE);
+    assertEquals(response.getEntity().getContentType().getName(), CONTENT_TYPE);
   }
 }
diff --git a/integ-test/src/test/java/org/opensearch/sql/sql/SimpleQueryStringIT.java b/integ-test/src/test/java/org/opensearch/sql/sql/SimpleQueryStringIT.java
@@ -79,6 +79,6 @@ public void contentHeaderTest() throws IOException {
 
     Response response = client().performRequest(sqlRequest);
 
-    assertEquals(response.getEntity().getContentType(), "content-type: " + CONTENT_TYPE);
+    assertEquals(response.getEntity().getContentType().getName(), CONTENT_TYPE);
   }
 }