Fix 2.x compatibility

Signed-off-by: Daniel Widdis <[email protected]>
opensearch-project · Dec 24, 2023 · 5771ef7 · 5771ef7
1 parent ed7050f
commit 5771ef7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/test/java/org/opensearch/flowframework/FlowFrameworkRestTestCase.java b/src/test/java/org/opensearch/flowframework/FlowFrameworkRestTestCase.java
@@ -445,7 +445,7 @@ protected SearchResponse searchWorkflowState(String query) throws Exception {
         assertEquals(RestStatus.OK, TestHelpers.restStatus(restSearchResponse));
 
         // Parse entity content into SearchResponse
-        MediaType mediaType = MediaType.fromMediaType(restSearchResponse.getEntity().getContentType());
+        MediaType mediaType = MediaType.fromMediaType(restSearchResponse.getEntity().getContentType().getValue());
         try (
             XContentParser parser = mediaType.xContent()
                 .createParser(
@@ -471,8 +471,8 @@ protected void getAndAssertWorkflowStatus(String workflowId, State stateStatus,
         assertEquals(RestStatus.OK, TestHelpers.restStatus(response));
 
         Map<String, Object> responseMap = entityAsMap(response);
-        assertEquals(stateStatus.name(), (String) responseMap.get(CommonValue.STATE_FIELD));
-        assertEquals(provisioningStatus.name(), (String) responseMap.get(CommonValue.PROVISIONING_PROGRESS_FIELD));
+        assertEquals(stateStatus.name(), responseMap.get(CommonValue.STATE_FIELD));
+        assertEquals(provisioningStatus.name(), responseMap.get(CommonValue.PROVISIONING_PROGRESS_FIELD));
 
     }