Extract helpers from FileOpenerTest. (#514)

Author: blackwinter

metafacture-io/src/test/java/org/metafacture/io/FileOpenerCompressionTest.java

@@ -93,7 +93,7 @@ public void testOpenCompressedFiles() throws IOException {
             Files.copy(in, file.toPath(), StandardCopyOption.REPLACE_EXISTING);
         }
 
-        FileOpenerTest.assertData(receiver, DATA, file, o -> o.setCompression(compression));
+        TestHelpers.assertFile(receiver, DATA, file, o -> o.setCompression(compression));
     }
 
 }

metafacture-io/src/test/java/org/metafacture/io/FileOpenerTest.java

@@ -25,7 +25,6 @@
 import org.junit.Test;
 import org.junit.rules.TemporaryFolder;
 import org.mockito.Mock;
-import org.mockito.Mockito;
 import org.mockito.junit.MockitoJUnit;
 import org.mockito.junit.MockitoRule;
 
@@ -40,7 +39,6 @@
 import java.nio.charset.StandardCharsets;
 import java.nio.file.Files;
 import java.nio.file.StandardCopyOption;
-import java.util.function.Consumer;
 
 /**
  * Tests for class {@link FileOpener}.
@@ -66,7 +64,7 @@ public void testUtf8IsDefaultEncoding() throws IOException {
         Assume.assumeFalse("Default encoding is UTF-8: It is not possible to test whether FileOpener sets " +
                 "the encoding to UTF-8 correctly.", StandardCharsets.UTF_8.equals(Charset.defaultCharset()));
 
-        assertData(receiver, DATA, createTestFile(), null);
+        TestHelpers.assertFile(receiver, DATA, createTestFile(), null);
     }
 
     @Test
@@ -93,27 +91,10 @@ private void testDecompressConcatenated(final boolean decompressConcatenated) th
         final String data = sb.toString();
         Assert.assertTrue(data.length() + " > " + maxBytes, data.length() > maxBytes);
 
-        assertData(receiver, decompressConcatenated ? data : data.substring(0, maxBytes),
+        TestHelpers.assertFile(receiver, decompressConcatenated ? data : data.substring(0, maxBytes),
                 copyResourceToTempFile("compressed-large.txt.bgzf"), o -> o.setDecompressConcatenated(decompressConcatenated));
     }
 
-    /*package-private*/ static void assertData(final ObjectReceiver<Reader> receiver, final String expected, final File file, final Consumer<FileOpener> consumer) {
-        final StringBuilder sb = new StringBuilder();
-        Mockito.doAnswer(i -> sb.append(ResourceUtil.readAll(i.getArgument(0)))).when(receiver).process(Mockito.any(Reader.class));
-
-        final FileOpener opener = new FileOpener();
-        if (consumer != null) {
-            consumer.accept(opener);
-        }
-
-        opener.setReceiver(receiver);
-        opener.process(file.getAbsolutePath());
-        opener.closeStream();
-
-        Mockito.verify(receiver).process(Mockito.any(Reader.class));
-        Assert.assertEquals(expected, sb.toString());
-    }
-
     private File createTestFile() throws IOException {
         final File file = tempFolder.newFile();
         try (OutputStream stream = new FileOutputStream(file)) {

metafacture-io/src/test/java/org/metafacture/io/TestHelpers.java

@@ -0,0 +1,63 @@
+/*
+ * Copyright 2024 hbz NRW
+ *
+ * Licensed under the Apache License, Version 2.0 the "License";
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.metafacture.io;
+
+import org.metafacture.commons.ResourceUtil;
+import org.metafacture.framework.ObjectReceiver;
+
+import org.junit.Assert;
+import org.mockito.Mockito;
+
+import java.io.File;
+import java.io.Reader;
+import java.util.function.Consumer;
+import java.util.function.IntSupplier;
+
+public final class TestHelpers {
+
+    public static void assertFile(final ObjectReceiver<Reader> receiver, final String expected, final File file, final Consumer<FileOpener> consumer) {
+        assertReader(receiver, expected, () -> {
+            final FileOpener opener = new FileOpener();
+            if (consumer != null) {
+                consumer.accept(opener);
+            }
+
+            opener.setReceiver(receiver);
+            opener.process(file.getAbsolutePath());
+            opener.closeStream();
+
+            return 1;
+        });
+    }
+
+    public static void assertReader(final ObjectReceiver<Reader> receiver, final String expected, final IntSupplier supplier) {
+        final StringBuilder sb = new StringBuilder();
+
+        Mockito.doAnswer(i -> {
+            sb.delete(0, sb.length());
+            sb.append(ResourceUtil.readAll(i.getArgument(0)));
+
+            return null;
+        }).when(receiver).process(Mockito.any(Reader.class));
+
+        final int times = supplier.getAsInt();
+
+        Mockito.verify(receiver, Mockito.times(times)).process(Mockito.any(Reader.class));
+        Assert.assertEquals(expected, sb.toString());
+    }
+
+}