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
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.io;
import java.io.*;
import java.nio.charset.StandardCharsets;
import org.junit.Test;
import static org.hamcrest.Matchers.greaterThan;
import static org.hamcrest.core.Is.is;
import static org.junit.Assert.*;
/**
* @author arnej
*/
public class TeeInputStreamTest {
@Test
public void testSimpleInput() throws IOException {
byte[] input = "very simple input".getBytes(StandardCharsets.UTF_8);
ByteArrayInputStream in = new ByteArrayInputStream(input);
ByteArrayOutputStream gotten = new ByteArrayOutputStream();
ByteArrayOutputStream output = new ByteArrayOutputStream();
TeeInputStream tee = new TeeInputStream(in, gotten);
int b = tee.read();
assertThat(b, is((int)'v'));
output.write(b);
assertThat(gotten.toString(), is("very simple input"));
for (int i = 0; i < 16; i++) {
b = tee.read();
// System.out.println("got["+i+"]: "+(char)b);
assertThat(b, is(greaterThan(0)));
output.write(b);
}
assertThat(tee.read(), is(-1));
assertThat(gotten.toString(), is("very simple input"));
assertThat(output.toString(), is("very simple input"));
}
private class Generator implements Runnable {
private OutputStream dst;
public Generator(OutputStream dst) { this.dst = dst; }
@Override
public void run() {
for (int i = 0; i < 123456789; i++) {
int b = i & 0x7f;
if (b < 32) continue;
if (b > 126) b = '\n';
try {
dst.write(b);
} catch (IOException e) {
return;
}
}
}
}
@Test
public void testPipedInput() throws IOException {
PipedOutputStream input = new PipedOutputStream();
PipedInputStream in = new PipedInputStream(input);
ByteArrayOutputStream gotten = new ByteArrayOutputStream();
ByteArrayOutputStream output = new ByteArrayOutputStream();
TeeInputStream tee = new TeeInputStream(in, gotten);
input.write("first input".getBytes(StandardCharsets.UTF_8));
int b = tee.read();
assertThat(b, is((int)'f'));
output.write(b);
assertThat(gotten.toString(), is("first input"));
input.write(" second input".getBytes(StandardCharsets.UTF_8));
b = tee.read();
assertThat(b, is((int)'i'));
output.write(b);
assertThat(gotten.toString(), is("first input second input"));
new Thread(new Generator(input)).start();
b = tee.read();
assertThat(b, is((int)'r'));
output.write(b);
byte[] ba = new byte[9];
for (int i = 0; i < 12345; i++) {
b = tee.read();
// System.out.println("got["+i+"]: "+(char)b);
assertThat(b, is(greaterThan(0)));
output.write(b);
int l = tee.read(ba);
assertThat(l, is(greaterThan(0)));
output.write(ba, 0, l);
l = tee.read(ba, 3, 3);
assertThat(l, is(greaterThan(0)));
output.write(ba, 3, l);
}
tee.close();
String got = gotten.toString();
// System.out.println("got length: "+got.length());
// System.out.println("output length: "+output.toString().length());
// System.out.println("got: "+got);
assertThat(got.length(), is(greaterThan(34567)));
assertTrue(got.startsWith(output.toString()));
}
}
|