1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
package org.brotli.wrapper.enc;
import static org.junit.Assert.assertEquals;
import org.brotli.integration.BundleHelper;
import org.brotli.wrapper.dec.BrotliInputStream;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.List;
import junit.framework.TestCase;
import junit.framework.TestSuite;
import org.junit.runner.RunWith;
import org.junit.runners.AllTests;
/** Tests for {@link org.brotli.wrapper.enc.BrotliOutputStream}. */
@RunWith(AllTests.class)
public class BrotliOutputStreamTest {
private enum TestMode {
WRITE_ALL,
WRITE_CHUNKS,
WRITE_BYTE
}
// TODO: remove when Bazel get JNI support.
static {
System.load(new java.io.File(new java.io.File(System.getProperty("java.library.path")),
"liblibjni.so").getAbsolutePath());
}
private static final int CHUNK_SIZE = 256;
static InputStream getBundle() throws IOException {
return new FileInputStream(System.getProperty("TEST_BUNDLE"));
}
/** Creates a test suite. */
public static TestSuite suite() throws IOException {
TestSuite suite = new TestSuite();
InputStream bundle = getBundle();
try {
List<String> entries = BundleHelper.listEntries(bundle);
for (String entry : entries) {
suite.addTest(new StreamTestCase(entry, TestMode.WRITE_ALL));
suite.addTest(new StreamTestCase(entry, TestMode.WRITE_CHUNKS));
suite.addTest(new StreamTestCase(entry, TestMode.WRITE_BYTE));
}
} finally {
bundle.close();
}
return suite;
}
/** Test case with a unique name. */
static class StreamTestCase extends TestCase {
final String entryName;
final TestMode mode;
StreamTestCase(String entryName, TestMode mode) {
super("BrotliOutputStreamTest." + entryName + "." + mode.name());
this.entryName = entryName;
this.mode = mode;
}
@Override
protected void runTest() throws Throwable {
BrotliOutputStreamTest.run(entryName, mode);
}
}
private static void run(String entryName, TestMode mode) throws Throwable {
InputStream bundle = getBundle();
byte[] original;
try {
original = BundleHelper.readEntry(bundle, entryName);
} finally {
bundle.close();
}
if (original == null) {
throw new RuntimeException("Can't read bundle entry: " + entryName);
}
if ((mode == TestMode.WRITE_CHUNKS) && (original.length <= CHUNK_SIZE)) {
return;
}
ByteArrayOutputStream dst = new ByteArrayOutputStream();
OutputStream encoder = new BrotliOutputStream(dst);
try {
switch (mode) {
case WRITE_ALL:
encoder.write(original);
break;
case WRITE_CHUNKS:
for (int offset = 0; offset < original.length; offset += CHUNK_SIZE) {
encoder.write(original, offset, Math.min(CHUNK_SIZE, original.length - offset));
}
break;
case WRITE_BYTE:
for (byte singleByte : original) {
encoder.write(singleByte);
}
break;
}
} finally {
encoder.close();
}
InputStream decoder = new BrotliInputStream(new ByteArrayInputStream(dst.toByteArray()));
try {
long originalCrc = BundleHelper.fingerprintStream(new ByteArrayInputStream(original));
long crc = BundleHelper.fingerprintStream(decoder);
assertEquals(originalCrc, crc);
} finally {
decoder.close();
}
}
}
|
