X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Ffanfix%2Ftest%2FConversionTest.java;h=d730b3b5efecc85f8a1b581c0ae16dc719e66187;hb=d66deb8d8b30cff6b54db352eef34a3508939f84;hp=92aca369d899eda7858828ce9378271478e654f5;hpb=86d49dbc7c3eca665b7823b5de49bb73a31c7722;p=nikiroo-utils.git diff --git a/src/be/nikiroo/fanfix/test/ConversionTest.java b/src/be/nikiroo/fanfix/test/ConversionTest.java index 92aca36..d730b3b 100644 --- a/src/be/nikiroo/fanfix/test/ConversionTest.java +++ b/src/be/nikiroo/fanfix/test/ConversionTest.java @@ -1,33 +1,37 @@ package be.nikiroo.fanfix.test; import java.io.File; -import java.io.FileInputStream; import java.io.FilenameFilter; -import java.io.IOException; import java.util.ArrayList; import java.util.Arrays; +import java.util.HashMap; import java.util.List; -import java.util.zip.ZipEntry; -import java.util.zip.ZipInputStream; +import java.util.Map; +import java.util.Map.Entry; import be.nikiroo.fanfix.Instance; import be.nikiroo.fanfix.Main; import be.nikiroo.fanfix.output.BasicOutput; import be.nikiroo.utils.IOUtils; -import be.nikiroo.utils.TempFiles; import be.nikiroo.utils.TraceHandler; import be.nikiroo.utils.test.TestCase; import be.nikiroo.utils.test.TestLauncher; class ConversionTest extends TestLauncher { - private TempFiles tempFiles; - private File testFile; - private File expectedDir; - private File resultDir; + private String testUri; + private String expectedDir; + private String resultDir; private List realTypes; + private Map> skipCompare; + private Map> skipCompareCross; - public ConversionTest(String[] args) { - super("Conversion", args); + public ConversionTest(String testName, final String testUri, + final String expectedDir, final String resultDir, String[] args) { + super("Conversion - " + testName, args); + + this.testUri = testUri; + this.expectedDir = expectedDir; + this.resultDir = resultDir; // Special mode SYSOUT is not a file type (System.out) realTypes = new ArrayList(); @@ -37,23 +41,28 @@ class ConversionTest extends TestLauncher { } } - addTest(new TestCase("Read the test file") { - @Override - public void test() throws Exception { - assertEquals("The test file \"" + testFile - + "\" cannot be found", true, testFile.exists()); - } - }); + if (!testUri.startsWith("http://") && !testUri.startsWith("https://")) { + addTest(new TestCase("Read the test file") { + @Override + public void test() throws Exception { + assertEquals("The test file \"" + testUri + + "\" cannot be found", true, + new File(testUri).exists()); + } + }); + } addTest(new TestCase("Assure directories exist") { @Override public void test() throws Exception { - expectedDir.mkdirs(); - resultDir.mkdirs(); + new File(expectedDir).mkdirs(); + new File(resultDir).mkdirs(); assertEquals("The Expected directory \"" + expectedDir - + "\" cannot be created", true, expectedDir.exists()); + + "\" cannot be created", true, + new File(expectedDir).exists()); assertEquals("The Result directory \"" + resultDir - + "\" cannot be created", true, resultDir.exists()); + + "\" cannot be created", true, + new File(resultDir).exists()); } }); @@ -64,28 +73,46 @@ class ConversionTest extends TestLauncher { @Override protected void start() throws Exception { - testFile = new File("test/test.story"); - expectedDir = new File("test/expected/"); - resultDir = new File("test/result/"); + skipCompare = new HashMap>(); + skipCompareCross = new HashMap>(); + + skipCompare.put("epb.ncx", Arrays.asList( + " ", + " ")); + skipCompare.put(".info", Arrays.asList("CREATION_DATE=", + "URL=\"file:/", "UUID=EPUBCREATOR=\"", "")); + skipCompare.put("URL", Arrays.asList("file:/")); + + for (String key : skipCompare.keySet()) { + skipCompareCross.put(key, skipCompare.get(key)); + } - tempFiles = new TempFiles("Fanfix-ConversionTest"); + skipCompareCross.put(".info", Arrays.asList("")); + skipCompareCross.put("epb.opf", Arrays.asList(" ")); + skipCompareCross.put("index.html", + Arrays.asList("
")); + skipCompareCross.put("URL", Arrays.asList("")); } @Override protected void stop() throws Exception { - tempFiles.close(); } private TestCase getTestFor(final BasicOutput.OutputType type) { return new TestCase(type + " output mode") { @Override public void test() throws Exception { - File target = generate(this, testFile, resultDir, type); + File target = generate(this, testUri, new File(resultDir), type); target = new File(target.getAbsolutePath() + type.getDefaultExtension(false)); // Check conversion: - compareFiles(this, expectedDir, resultDir, type, null); + compareFiles(this, new File(expectedDir), new File(resultDir), + type, "Generate " + type); // LATEX not supported as input if (BasicOutput.OutputType.LATEX.equals(type)) { @@ -94,21 +121,25 @@ class ConversionTest extends TestLauncher { // Cross-checks: for (BasicOutput.OutputType crossType : realTypes) { - File crossDir = tempFiles.createTempDir("cross-result"); - generate(this, target, crossDir, crossType); - compareFiles(this, resultDir, crossDir, crossType, + File crossDir = Test.tempFiles + .createTempDir("cross-result"); + + generate(this, target.getAbsolutePath(), crossDir, crossType); + compareFiles(this, new File(resultDir), crossDir, + crossType, "Cross compare " + crossType + + " generated from " + type); } } }; } - private File generate(TestCase testCase, File testFile, File resultDir, + private File generate(TestCase testCase, String testUri, File resultDir, BasicOutput.OutputType type) throws Exception { final List errors = new ArrayList(); - TraceHandler previousTraceHandler = Instance.getTraceHandler(); - Instance.setTraceHandler(new TraceHandler(true, true, 0) { + TraceHandler previousTraceHandler = Instance.getInstance().getTraceHandler(); + Instance.getInstance().setTraceHandler(new TraceHandler(true, true, 0) { @Override public void error(String message) { errors.add(message); @@ -130,8 +161,8 @@ class ConversionTest extends TestLauncher { try { File target = new File(resultDir, type.toString()); - int code = Main.convert(testFile.getAbsolutePath(), - type.toString(), target.getAbsolutePath(), false, null); + int code = Main.convert(testUri, type.toString(), + target.getAbsolutePath(), false, null); String error = ""; for (String err : errors) { @@ -147,57 +178,65 @@ class ConversionTest extends TestLauncher { return target; } finally { - Instance.setTraceHandler(previousTraceHandler); + Instance.getInstance().setTraceHandler(previousTraceHandler); } } private void compareFiles(TestCase testCase, File expectedDir, - File resultDir, final BasicOutput.OutputType typeToCompare, - final BasicOutput.OutputType sourceType) throws Exception { + File resultDir, final BasicOutput.OutputType limitTiFiles, + final String errMess) throws Exception { + + Map> skipCompare = errMess.startsWith("Cross") ? this.skipCompareCross + : this.skipCompare; FilenameFilter filter = null; - if (typeToCompare != null) { + if (limitTiFiles != null) { filter = new FilenameFilter() { @Override public boolean accept(File dir, String name) { - return name.startsWith(typeToCompare.toString()); + return name.toLowerCase().startsWith( + limitTiFiles.toString().toLowerCase()); } }; } - List resultFiles = Arrays.asList(resultDir.list(filter)); - resultFiles.sort(null); - List expectedFiles = Arrays.asList(expectedDir.list(filter)); - expectedFiles.sort(null); + List resultFiles; + List expectedFiles; + { + String[] resultArr = resultDir.list(filter); + Arrays.sort(resultArr); + resultFiles = Arrays.asList(resultArr); + String[] expectedArr = expectedDir.list(filter); + Arrays.sort(expectedArr); + expectedFiles = Arrays.asList(expectedArr); + } - testCase.assertEquals("The resulting file names are not expected", - expectedFiles, resultFiles); + testCase.assertEquals(errMess, expectedFiles, resultFiles); for (int i = 0; i < resultFiles.size(); i++) { File expected = new File(expectedDir, expectedFiles.get(i)); File result = new File(resultDir, resultFiles.get(i)); - testCase.assertEquals( - "Type mismatch: expected a " - + (expected.isDirectory() ? "directory" : "file") - + ", received a " - + (result.isDirectory() ? "directory" : "file"), + testCase.assertEquals(errMess + ": type mismatch: expected a " + + (expected.isDirectory() ? "directory" : "file") + + ", received a " + + (result.isDirectory() ? "directory" : "file"), expected.isDirectory(), result.isDirectory()); if (expected.isDirectory()) { - compareFiles(testCase, expected, result, null, sourceType); + compareFiles(testCase, expected, result, null, errMess); continue; } if (expected.getName().endsWith(".cbz") || expected.getName().endsWith(".epub")) { - File tmpExpected = tempFiles.createTempDir(expected.getName() - + "[zip-content]"); - File tmpResult = tempFiles.createTempDir(result.getName() + File tmpExpected = Test.tempFiles.createTempDir(expected + .getName() + "[zip-content]"); + File tmpResult = Test.tempFiles.createTempDir(result.getName() + "[zip-content]"); - unzip(expected, tmpExpected); - unzip(result, tmpResult); - compareFiles(testCase, tmpExpected, tmpResult, null, sourceType); + IOUtils.unzip(expected, tmpExpected); + IOUtils.unzip(result, tmpResult); + compareFiles(testCase, tmpExpected, tmpResult, null, errMess); } else { List expectedLines = Arrays.asList(IOUtils .readSmallFile(expected).split("\n")); @@ -210,18 +249,24 @@ class ConversionTest extends TestLauncher { + name.substring(expectedDir.getAbsolutePath() .length()); } + + testCase.assertEquals(errMess + ": " + name + + ": the number of lines is not the same", + expectedLines.size(), resultLines.size()); + for (int j = 0; j < expectedLines.size(); j++) { String expectedLine = expectedLines.get(j); String resultLine = resultLines.get(j); boolean skip = false; - for (String skipStart : new String[] { "CREATION_DATE=", - "SUBJECT=", "URL=", "UUID=" }) { - if (name.endsWith(".info") - && expectedLine.startsWith(skipStart) - && resultLine.startsWith(skipStart)) { - // TODO: check the format? - skip = true; + for (Entry> skipThose : skipCompare + .entrySet()) { + for (String skipStart : skipThose.getValue()) { + if (name.endsWith(skipThose.getKey()) + && expectedLine.startsWith(skipStart) + && resultLine.startsWith(skipStart)) { + skip = true; + } } } @@ -229,46 +274,11 @@ class ConversionTest extends TestLauncher { continue; } - testCase.assertEquals("Line " + (j + 1) + " (" + sourceType - + ") is not the same in file " + name, - expectedLine, resultLine); + testCase.assertEquals(errMess + ": line " + (j + 1) + + " is not the same in file " + name, expectedLine, + resultLine); } } } } - - // TODO: remove and use IOUtils when updated - private static void unzip(File zipFile, File targetDirectory) - throws IOException { - if (targetDirectory.exists() && targetDirectory.isFile()) { - throw new IOException("Cannot unzip " + zipFile + " into " - + targetDirectory + ": it is not a directory"); - } - - targetDirectory.mkdir(); - if (!targetDirectory.exists()) { - throw new IOException("Cannot create target directory " - + targetDirectory); - } - - FileInputStream in = new FileInputStream(zipFile); - try { - ZipInputStream zipStream = new ZipInputStream(in); - try { - for (ZipEntry entry = zipStream.getNextEntry(); entry != null; entry = zipStream - .getNextEntry()) { - File file = new File(targetDirectory, entry.getName()); - if (entry.isDirectory()) { - file.mkdirs(); - } else { - IOUtils.write(zipStream, file); - } - } - } finally { - zipStream.close(); - } - } finally { - in.close(); - } - } }