Add “WatchFeeds” command

[jFCPlib.git] / src / test / java / net / pterodactylus / fcp / FcpMessageTest.java

diff --git a/src/test/java/net/pterodactylus/fcp/FcpMessageTest.java b/src/test/java/net/pterodactylus/fcp/FcpMessageTest.java
index bc0dbd1..e2b9fcf 100644 (file)
--- a/src/test/java/net/pterodactylus/fcp/FcpMessageTest.java
+++ b/src/test/java/net/pterodactylus/fcp/FcpMessageTest.java
@@ -4,8 +4,6 @@ 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;
@@ -15,16 +13,14 @@ import org.junit.Test;
  *
  * @author <a href="mailto:david.roden@bietr.de">David Roden</a>
  */
-public class FcpMessageTest {
+public class FcpMessageTest extends AbstractFcpMessageTest {
 
        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<String> lines = Arrays.asList(outputStream.toString().split("\r?\n"));
+               List<String> lines = encodeMessage(fcpMessage);
                assertThat(lines, contains(
                        "TestMessage",
                        "Data",