java - JUnit 测试 Hadoop 可写

Question

我正在编写一些自定义 Hadoop 可写类。我想使用 JUnit 来测试 readFields() 和 write() 函数。

有没有办法将 write(DataOutput out) 重定向到 readFields(DataInput in)？我想做类似的事情:

CustomWritable writeable1 = new CustomWritable();
CustomWritable writeable2 = new CustomWritable();

//build writable1 with some data.
DataInputAndOutput io = ...

writeable1.write(io);
writable2.read(io);

assertEquals(writable1,writable2);

想法？

Answer 1

最终使用 ByteArrayOutputStream byteout = ... 和 ByteArrayInputStream(byteout.toByteArray()) 来完成此任务。

我尝试使用:

http://docs.oracle.com/javase/7/docs/api/java/io/PipedOutputStream.html http://docs.oracle.com/javase/7/docs/api/java/io/PipedInputStream.html

但发现如果从同一个线程运行它们有时会死锁。