src/test/java/net/pterodactylus/fcp/FcpMessageTest.java

   1 package net.pterodactylus.fcp;
   2
   3 import static org.hamcrest.MatcherAssert.assertThat;
   4 import static org.hamcrest.Matchers.contains;
   5
   6 import java.io.ByteArrayInputStream;
   7 import java.util.List;
   8
   9 import org.junit.Test;
  10
  11 /**
  12  * Unit test for {@link FcpMessage}.
  13  *
  14  * @author <a href="mailto:david.roden@bietr.de">David Roden</a>
  15  */
  16 public class FcpMessageTest extends AbstractFcpMessageTest {
  17
  18         private final FcpMessage fcpMessage = new FcpMessage("TestMessage");
  19
  20         @Test
  21         public void fcpMessageWithPayloadIsTerminatedByData() throws Exception {
  22                 fcpMessage.setPayloadInputStream(new ByteArrayInputStream("Test".getBytes()));
  23                 List<String> lines = encodeMessage(fcpMessage);
  24                 assertThat(lines, contains(
  25                         "TestMessage",
  26                         "Data",
  27                         "Test"
  28                 ));
  29         }
  30
  31 }