X-Git-Url: https://git.pterodactylus.net/?p=xudocci.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Firc%2Fconnection%2FReplies.java;h=841d3aff1f28e400133c1d82d83c0f3ff9a16a22;hp=214c3d4dcc9b6744974b07669b93397c0ebbc365;hb=329cfa999c0814d337900b5be19b5ff4f9ec8579;hpb=767b7be6c7e6f5a0d96895ab5019b7eca0982a59

diff --git a/src/test/java/net/pterodactylus/irc/connection/Replies.java b/src/test/java/net/pterodactylus/irc/connection/Replies.java
index 214c3d4..841d3af 100644
--- a/src/test/java/net/pterodactylus/irc/connection/Replies.java
+++ b/src/test/java/net/pterodactylus/irc/connection/Replies.java
@@ -1,10 +1,15 @@
 package net.pterodactylus.irc.connection;
 
+import static com.google.common.base.Optional.fromNullable;
 import static java.util.Arrays.asList;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
+import java.util.ArrayList;
+import java.util.List;
+
 import net.pterodactylus.irc.Reply;
+import net.pterodactylus.irc.Source;
 
 /**
  * Helper class to mock {@link Reply}s for testing {@link Handler}s.
@@ -13,10 +18,20 @@ import net.pterodactylus.irc.Reply;
  */
 public class Replies {
 
-	public static Reply createReply(String command, String channel) {
+	public static Reply createReply(Source source, String command,
+			String... parameters) {
+		Reply reply = createReply(command, parameters);
+		when(reply.source()).thenReturn(fromNullable(source));
+		return reply;
+	}
+
+	public static Reply createReply(String command, String... parameters) {
 		final Reply reply = mock(Reply.class);
 		when(reply.command()).thenReturn(command);
-		when(reply.parameters()).thenReturn(asList(":some.server", channel));
+		List<String> allParameters = new ArrayList<>();
+		allParameters.add(":some.server");
+		allParameters.addAll(asList(parameters));
+		when(reply.parameters()).thenReturn(allParameters);
 		return reply;
 	}