X-Git-Url: https://git.pterodactylus.net/?p=jFCPlib.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Ffcp%2FFcpMessageTest.java;fp=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Ffcp%2FFcpMessageTest.java;h=bc0dbd1889076aecbd0cd119f882d3120757d58d;hp=0000000000000000000000000000000000000000;hb=14edf617ab865e3634f243c87ac1cb636c055b6c;hpb=a30d521e1add957b2ddfee455bc6e061efbf44b7
diff --git a/src/test/java/net/pterodactylus/fcp/FcpMessageTest.java b/src/test/java/net/pterodactylus/fcp/FcpMessageTest.java
new file mode 100644
index 0000000..bc0dbd1
--- /dev/null
+++ b/src/test/java/net/pterodactylus/fcp/FcpMessageTest.java
@@ -0,0 +1,35 @@
+package net.pterodactylus.fcp;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.contains;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.util.Arrays;
+import java.util.List;
+
+import org.junit.Test;
+
+/**
+ * Unit test for {@link FcpMessage}.
+ *
+ * @author David Roden
+ */
+public class FcpMessageTest {
+
+ private final FcpMessage fcpMessage = new FcpMessage("TestMessage");
+
+ @Test
+ public void fcpMessageWithPayloadIsTerminatedByData() throws Exception {
+ fcpMessage.setPayloadInputStream(new ByteArrayInputStream("Test".getBytes()));
+ ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
+ fcpMessage.write(outputStream);
+ List lines = Arrays.asList(outputStream.toString().split("\r?\n"));
+ assertThat(lines, contains(
+ "TestMessage",
+ "Data",
+ "Test"
+ ));
+ }
+
+}