Added tests for poison-pull pattern

2026-05-30 00:21:09 +00:00 · 2015-12-27 22:42:39 +01:00
parent b3d1c2b2ed
commit 47e1cd710c
6 changed files with 258 additions and 0 deletions
@@ -0,0 +1,56 @@
+package com.iluwatar.poison.pill;
+
+import org.junit.Test;
+import org.mockito.InOrder;
+
+import java.time.LocalDateTime;
+
+import static org.mockito.Mockito.inOrder;
+
+/**
+ * Date: 12/27/15 - 9:45 PM
+ *
+ * @author Jeroen Meulemeester
+ */
+public class ConsumerTest extends StdOutTest {
+
+  @Test
+  public void testConsume() throws Exception {
+    final Message[] messages = new Message[]{
+            createMessage("you", "Hello!"),
+            createMessage("me", "Hi!"),
+            Message.POISON_PILL,
+            createMessage("late_for_the_party", "Hello? Anyone here?"),
+    };
+
+    final MessageQueue queue = new SimpleMessageQueue(messages.length);
+    for (final Message message : messages) {
+      queue.put(message);
+    }
+
+    new Consumer("NSA", queue).consume();
+
+    final InOrder inOrder = inOrder(getStdOutMock());
+    inOrder.verify(getStdOutMock()).println("Message [Hello!] from [you] received by [NSA]");
+    inOrder.verify(getStdOutMock()).println("Message [Hi!] from [me] received by [NSA]");
+    inOrder.verify(getStdOutMock()).println("Consumer NSA receive request to terminate.");
+    inOrder.verifyNoMoreInteractions();
+
+  }
+
+  /**
+   * Create a new message from the given sender with the given message body
+   *
+   * @param sender  The sender's name
+   * @param message The message body
+   * @return The message instance
+   */
+  private static Message createMessage(final String sender, final String message) {
+    final SimpleMessage msg = new SimpleMessage();
+    msg.addHeader(Message.Headers.SENDER, sender);
+    msg.addHeader(Message.Headers.DATE, LocalDateTime.now().toString());
+    msg.setBody(message);
+    return msg;
+  }
+
+}