assert read ammount

apache · Nov 29, 2023 · 0ce8627 · 0ce8627
1 parent 7859d5e
commit 0ce8627
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/server/src/test/java/org/apache/druid/server/AsyncManagementForwardingServletTest.java b/server/src/test/java/org/apache/druid/server/AsyncManagementForwardingServletTest.java
@@ -326,7 +326,7 @@ public void testProxyEnebledCheck() throws Exception
 
     Assert.assertEquals(200, connection.getResponseCode());
     byte[] bytes = new byte[connection.getContentLength()];
-    connection.getInputStream().read(bytes);
+    Assert.assertEquals(connection.getInputStream().read(bytes), connection.getContentLength());
     Assert.assertEquals(ImmutableMap.of("enabled", true), new ObjectMapper().readValue(bytes, Map.class));
     Assert.assertFalse("coordinator called", COORDINATOR_EXPECTED_REQUEST.called);
     Assert.assertFalse("overlord called", OVERLORD_EXPECTED_REQUEST.called);