Added some network updates that work with the new Forwarder

test/resources/network/TestNetBufferStream.java

@@ -0,0 +1,119 @@
+package resources.network;
+
+import java.nio.charset.StandardCharsets;
+
+import org.junit.Assert;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+@RunWith(JUnit4.class)
+public class TestNetBufferStream {
+	
+	@Test
+	public void testExpansionSingle() {
+		try (NetBufferStream stream = new NetBufferStream(4)) {
+			for (int i = 0; i < 10; i++)
+				stream.write(5);
+		}
+	}
+	
+	@Test
+	public void testExpansionBulk() {
+		byte [] data = new byte[1024];
+		try (NetBufferStream stream = new NetBufferStream(4)) {
+			for (int i = 0; i < 10; i++)
+				stream.write(data);
+		}
+	}
+	
+	@Test
+	public void testWriteReadString() {
+		try (NetBufferStream stream = new NetBufferStream(4)) {
+			for (int i = 0; i < 10; i++)
+				stream.write(generateTestString(i));
+			for (int i = 0; i < 10; i++)
+				Assert.assertEquals(getTestString(i), stream.getAscii());
+		}
+	}
+	
+	@Test
+	public void testWriteReadCompactString() {
+		try (NetBufferStream stream = new NetBufferStream(4)) {
+			for (int i = 0; i < 10; i++)
+				stream.write(generateTestString(i));
+			for (int i = 0; i < 10; i++) {
+				Assert.assertEquals(getTestString(i), stream.getAscii());
+				if (i == 4) {
+					int rem = stream.remaining();
+					stream.compact();
+					Assert.assertEquals(0, stream.position());
+					Assert.assertEquals(rem, stream.remaining());
+				}
+			}
+		}
+	}
+	
+	@Test
+	public void testWriteReadInterleave() {
+		try (NetBufferStream stream = new NetBufferStream(4)) {
+			for (int i = 0; i < 10; i++) {
+				stream.write(generateTestString(i));
+				if (i % 2 == 1) {
+					Assert.assertEquals(getTestString(i-1), stream.getAscii());
+					Assert.assertEquals(getTestString(i), stream.getAscii());
+				}
+			}
+			Assert.assertEquals(0, stream.remaining());
+		}
+	}
+	
+	@Test
+	public void testWriteReadCompactInterleave() {
+		try (NetBufferStream stream = new NetBufferStream(4)) {
+			for (int i = 0; i < 10; i++) {
+				stream.write(generateTestString(i));
+				if (i % 2 == 1) {
+					Assert.assertEquals(getTestString(i-1), stream.getAscii());
+					Assert.assertEquals(getTestString(i), stream.getAscii());
+					int rem = stream.remaining();
+					stream.compact();
+					Assert.assertEquals(0, stream.position());
+					Assert.assertEquals(rem, stream.remaining());
+				}
+			}
+			Assert.assertEquals(0, stream.remaining());
+		}
+	}
+	
+	@Test
+	public void testRewind() {
+		try (NetBufferStream stream = new NetBufferStream(4)) {
+			for (int i = 0; i < 10; i++)
+				stream.write(generateTestString(i));
+			for (int i = 0; i < 10; i++) {
+				Assert.assertEquals(getTestString(i), stream.getAscii());
+				if (i == 4)
+					stream.mark();
+			}
+			Assert.assertEquals(0, stream.remaining());
+			stream.rewind();
+			Assert.assertEquals(5*(2+getTestString(0).length()), stream.remaining());
+			for (int i = 5; i < 10; i++)
+				Assert.assertEquals(getTestString(i), stream.getAscii());
+		}
+	}
+	
+	private byte [] generateTestString(int num) {
+		String test = getTestString(num);
+		byte [] data = new byte[2 + test.length()];
+		data[0] = (byte) test.length();
+		System.arraycopy(test.getBytes(StandardCharsets.US_ASCII), 0, data, 2, data[0]);
+		return data;
+	}
+	
+	private String getTestString(int num) {
+		return "Hello World " + num + "!";
+	}
+	
+}