Revert unnecessary fixes in tests

Author: shnapz

sdks/java/io/jms/src/test/java/org/apache/beam/sdk/io/jms/JmsIOIT.java

@@ -215,7 +215,7 @@ public void testPublishingThenReadingAll() throws IOException, JMSException {
     int unackRecords = countRemain(QUEUE);
     assertTrue(
         String.format("Too many unacknowledged messages: %d", unackRecords),
-        unackRecords < OPTIONS.getNumberOfRecords() * 0.005);
+        unackRecords < OPTIONS.getNumberOfRecords() * 0.003);
 
     // acknowledged records
     int ackRecords = OPTIONS.getNumberOfRecords() - unackRecords;

sdks/java/io/jms/src/test/java/org/apache/beam/sdk/io/jms/JmsIOTest.java

@@ -660,9 +660,6 @@ private JmsIO.UnboundedJmsReader setupReaderForTest() throws JMSException {
     final int delay = 10;
     return connectorClass == JmsConnectionFactory.class
         ? (JmsTextMessage message) -> {
-          if (message == null) {
-            return null;
-          }
           final JmsAcknowledgeCallback originalCallback = message.getAcknowledgeCallback();
           JmsAcknowledgeCallback jmsAcknowledgeCallbackMock =
               Mockito.mock(JmsAcknowledgeCallback.class);
@@ -682,9 +679,6 @@ private JmsIO.UnboundedJmsReader setupReaderForTest() throws JMSException {
           return message;
         }
         : (ActiveMQMessage message) -> {
-          if (message == null) {
-            return null;
-          }
           final Callback originalCallback = message.getAcknowledgeCallback();
           message.setAcknowledgeCallback(
               () -> {