crowetic/qortal
1
0
Fork 0
forked from Qortal/qortal
Code Pull Requests Activity

Added unit tests for zipping / unzipping

Browse Source
This commit is contained in:
CalDescent 2021-12-05 17:38:53 +00:00
parent c6e5c4e3b5
commit 4eef28f93d
1 changed files with 128 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

128
src/test/java/org/qortal/test/arbitrary/ArbitraryCompressionTests.java Normal file
View File

@ -0,0 +1,128 @@
package org.qortal.test.arbitrary;
import org.junit.Before;
import org.junit.Test;
import org.qortal.arbitrary.ArbitraryDataDigest;
import org.qortal.crypto.Crypto;
import org.qortal.repository.DataException;
import org.qortal.test.common.Common;
import org.qortal.utils.ZipUtils;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.StandardOpenOption;
import java.util.Arrays;
import java.util.Random;
import static org.junit.Assert.*;
public class ArbitraryCompressionTests extends Common {
@Before
public void beforeTest() throws DataException {
Common.useDefaultSettings();
}
@Test
public void testZipSingleFile() throws IOException, InterruptedException {
Path inputFile = Files.createTempFile("inputFile", null);
Path outputDirectory = Files.createTempDirectory("outputDirectory");
Path outputFile = Paths.get(outputDirectory.toString(), "data");
inputFile.toFile().deleteOnExit();
outputDirectory.toFile().deleteOnExit();
// Write random data to the input file
byte[] data = new byte[1024];
new Random().nextBytes(data);
Files.write(inputFile, data, StandardOpenOption.CREATE);
assertTrue(Files.exists(inputFile));
assertFalse(Files.exists(outputFile));
String fileName = "data"; //isSingleFile ? singleFileName : null;
ZipUtils.zip(inputFile.toString(), outputFile.toString(), fileName);
assertTrue(Files.exists(inputFile));
assertTrue(Files.exists(outputFile));
// Ensure zipped file's hash differs from the original
assertFalse(Arrays.equals(Crypto.digest(inputFile.toFile()), Crypto.digest(outputFile.toFile())));
// Create paths for unzipping
Path unzippedDirectory = Files.createTempDirectory("unzippedDirectory");
// Single file data is unzipped directly, without an enclosing folder
Path unzippedFile = Paths.get(unzippedDirectory.toString(), "data");
unzippedDirectory.toFile().deleteOnExit();
assertFalse(Files.exists(unzippedFile));
// Now unzip...
ZipUtils.unzip(outputFile.toString(), unzippedDirectory.toString());
// Ensure resulting file exists
assertTrue(Files.exists(unzippedFile));
// And make sure it matches the original input file
assertTrue(Arrays.equals(Crypto.digest(inputFile.toFile()), Crypto.digest(unzippedFile.toFile())));
}
@Test
public void testZipMultipleFiles() throws IOException, InterruptedException, DataException {
Path inputDirectory = Files.createTempDirectory("inputDirectory");
Path outputDirectory = Files.createTempDirectory("outputDirectory");
Path outputFile = Paths.get(outputDirectory.toString(), "data");
inputDirectory.toFile().deleteOnExit();
outputDirectory.toFile().deleteOnExit();
Path inputFile1 = Paths.get(inputDirectory.toString(), "file1");
Path inputFile2 = Paths.get(inputDirectory.toString(), "file2");
// Write random data to some files
byte[] data = new byte[1024];
new Random().nextBytes(data);
Files.write(inputFile1, data, StandardOpenOption.CREATE);
Files.write(inputFile2, data, StandardOpenOption.CREATE);
assertTrue(Files.exists(inputDirectory));
assertTrue(Files.exists(inputFile1));
assertTrue(Files.exists(inputFile2));
assertFalse(Files.exists(outputFile));
String fileName = "data"; //isSingleFile ? singleFileName : null;
ZipUtils.zip(inputDirectory.toString(), outputFile.toString(), fileName);
assertTrue(Files.exists(inputDirectory));
assertTrue(Files.exists(outputFile));
// Create paths for unzipping
Path unzippedDirectory = Files.createTempDirectory("unzippedDirectory");
unzippedDirectory.toFile().deleteOnExit();
Path unzippedFile1 = Paths.get(unzippedDirectory.toString(), fileName, "file1");
Path unzippedFile2 = Paths.get(unzippedDirectory.toString(), fileName, "file2");
assertFalse(Files.exists(unzippedFile1));
assertFalse(Files.exists(unzippedFile2));
// Now unzip...
ZipUtils.unzip(outputFile.toString(), unzippedDirectory.toString());
// Ensure resulting files exist
assertTrue(Files.exists(unzippedFile1));
assertTrue(Files.exists(unzippedFile2));
// And make sure they match the original input files
assertTrue(Arrays.equals(Crypto.digest(inputFile1.toFile()), Crypto.digest(unzippedFile1.toFile())));
assertTrue(Arrays.equals(Crypto.digest(inputFile2.toFile()), Crypto.digest(unzippedFile2.toFile())));
// Unzipped files are placed within a folder named by the supplied fileName
Path unzippedInnerDirectory = Paths.get(unzippedDirectory.toString(), fileName);
// Finally, make sure the directory digests match
ArbitraryDataDigest inputDirectoryDigest = new ArbitraryDataDigest(inputDirectory);
inputDirectoryDigest.compute();
ArbitraryDataDigest unzippedDirectoryDigest = new ArbitraryDataDigest(unzippedInnerDirectory);
unzippedDirectoryDigest.compute();
assertEquals(inputDirectoryDigest.getHash58(), unzippedDirectoryDigest.getHash58());
}
}