Update tests that expect certain amount of headers in a response

Signed-off-by: Craig Perkins <cwperx@amazon.com>
opensearch-project · Jul 17, 2023 · a104bf4 · a104bf4
1 parent c1eb5b8
commit a104bf4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/test/java/org/opensearch/security/dlic/dlsfls/DlsFlsCrossClusterSearchTest.java b/src/test/java/org/opensearch/security/dlic/dlsfls/DlsFlsCrossClusterSearchTest.java
@@ -164,7 +164,7 @@ public void testCcs() throws Exception {
         Assert.assertFalse(ccs.getBody().contains("secret1"));
         Assert.assertFalse(ccs.getBody().contains("AnotherSecredField"));
         Assert.assertFalse(ccs.getBody().contains("xxx1"));
-        Assert.assertEquals(ccs.getHeaders().toString(), 2, ccs.getHeaders().size());
+        Assert.assertEquals(ccs.getHeaders().toString(), 3, ccs.getHeaders().size());
     }
 
     @Test
@@ -251,7 +251,7 @@ public void testCcsDifferentConfig() throws Exception {
         Assert.assertTrue(ccs.getBody().contains("__fn__crl2"));
         Assert.assertFalse(ccs.getBody().contains("secret1"));
         Assert.assertFalse(ccs.getBody().contains("AnotherSecredField"));
-        Assert.assertEquals(ccs.getHeaders().toString(), 2, ccs.getHeaders().size());
+        Assert.assertEquals(ccs.getHeaders().toString(), 3, ccs.getHeaders().size());
     }
 
     @Test
@@ -382,6 +382,6 @@ public void testCcsDifferentConfigBoth() throws Exception {
         Assert.assertFalse(ccs.getBody().contains("secret1"));
         Assert.assertFalse(ccs.getBody().contains("AnotherSecredField"));
         Assert.assertTrue(ccs.getBody().contains("someoneelse"));
-        Assert.assertEquals(ccs.getHeaders().toString(), 2, ccs.getHeaders().size());
+        Assert.assertEquals(ccs.getHeaders().toString(), 3, ccs.getHeaders().size());
     }
 }
diff --git a/src/test/java/org/opensearch/security/dlic/dlsfls/DlsTest.java b/src/test/java/org/opensearch/security/dlic/dlsfls/DlsTest.java
@@ -111,7 +111,7 @@ public void testDls() throws Exception {
         );
         Assert.assertTrue(res.getBody().contains("\"value\" : 1,\n      \"relation"));
         Assert.assertTrue(res.getBody().contains("\"failed\" : 0"));
-        Assert.assertEquals(res.getHeaders().toString(), 2, res.getHeaders().size());
+        Assert.assertEquals(res.getHeaders().toString(), 3, res.getHeaders().size());
 
         Assert.assertEquals(
             HttpStatus.SC_OK,