Kaladaran/briar
0
0
Fork 0
mirror of https://code.briarproject.org/briar/briar.git synced 2026-02-22 23:59:54 +01:00
Code Issues Projects Releases Wiki Activity

Unit tests for ConnectionWriterImpl.

Browse Source
This commit is contained in:
akwizgran
2011-08-19 11:50:47 +02:00
parent 156bafb8a6
commit 7a53ea7814
1 changed files with 30 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
test/net/sf/briar/transport/ConnectionWriterImplTest.java
View File

@@ -1,5 +1,7 @@
package net.sf.briar.transport; package net.sf.briar.transport;
import static net.sf.briar.api.transport.TransportConstants.MAX_FRAME_LENGTH;
import java.io.ByteArrayOutputStream; import java.io.ByteArrayOutputStream;
import java.io.IOException; import java.io.IOException;
import java.io.OutputStream; import java.io.OutputStream;
@@ -30,7 +32,16 @@ public class ConnectionWriterImplTest extends TestCase {
mac.init(crypto.generateSecretKey()); mac.init(crypto.generateSecretKey());
} }
// FIXME: Test corner cases @Test
public void testFlushWithoutWriteProducesNothing() throws Exception {
ByteArrayOutputStream out = new ByteArrayOutputStream();
ConnectionEncrypter e = new NullConnectionEncrypter(out);
ConnectionWriter w = new ConnectionWriterImpl(e, mac);
w.getOutputStream().flush();
w.getOutputStream().flush();
w.getOutputStream().flush();
assertEquals(0, out.size());
}
@Test @Test
public void testSingleByteFrame() throws Exception { public void testSingleByteFrame() throws Exception {
@@ -49,6 +60,24 @@ public class ConnectionWriterImplTest extends TestCase {
assertTrue(Arrays.equals(frame, out.toByteArray())); assertTrue(Arrays.equals(frame, out.toByteArray()));
} }
@Test
public void testFrameIsWrittenAtMaxLength() throws Exception {
int maxPayloadLength = MAX_FRAME_LENGTH - 6 - mac.getMacLength();
ByteArrayOutputStream out = new ByteArrayOutputStream();
ConnectionEncrypter e = new NullConnectionEncrypter(out);
ConnectionWriter w = new ConnectionWriterImpl(e, mac);
OutputStream out1 = w.getOutputStream();
// The first maxPayloadLength bytes should be buffered
for(int i = 0; i < maxPayloadLength; i++) out1.write(0);
assertEquals(0, out.size());
// The next byte should trigger the writing of a frame
out1.write(0);
assertEquals(MAX_FRAME_LENGTH, out.size());
// Flushing the stream should write a single-byte frame
out1.flush();
assertEquals(MAX_FRAME_LENGTH + 6 + 1 + mac.getMacLength(), out.size());
}
@Test @Test
public void testMultipleFrames() throws Exception { public void testMultipleFrames() throws Exception {
// First frame: 123-byte payload // First frame: 123-byte payload