Assert status code

Signed-off-by: Craig Perkins <[email protected]>
cwperks · Oct 18, 2024 · e49214b · e49214b
1 parent c135492
commit e49214b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/integrationTest/java/org/opensearch/security/StoredFieldsTests.java b/src/integrationTest/java/org/opensearch/security/StoredFieldsTests.java
@@ -1,6 +1,7 @@
 package org.opensearch.security;
 
 import com.carrotsearch.randomizedtesting.annotations.ThreadLeakScope;
+import org.apache.http.HttpStatus;
 import org.junit.Assert;
 import org.junit.BeforeClass;
 import org.junit.ClassRule;
@@ -67,6 +68,7 @@ public void testStoredWithWithApplicableMaskedFieldRestrictions() {
  ]
  }
  """);
+ fieldSearchResponse.assertStatusCode(HttpStatus.SC_OK);
  Assert.assertTrue(fieldSearchResponse.getBody().contains("raw"));
  Assert.assertTrue(fieldSearchResponse.getBody().contains("hello"));
  Assert.assertTrue(fieldSearchResponse.getBody().contains("restricted"));
@@ -88,6 +90,7 @@ public void testStoredWithWithApplicableFlsRestrictions() {
  ]
  }
  """);
+ fieldSearchResponse.assertStatusCode(HttpStatus.SC_OK);
  Assert.assertTrue(fieldSearchResponse.getBody().contains("raw"));
  Assert.assertTrue(fieldSearchResponse.getBody().contains("hello"));
  Assert.assertFalse(fieldSearchResponse.getBody().contains("restricted"));