Use a source in the reply.

[xudocci.git] / src / test / java / net / pterodactylus / irc / ReplyTest.java

diff --git a/src/test/java/net/pterodactylus/irc/ReplyTest.java b/src/test/java/net/pterodactylus/irc/ReplyTest.java
index 9d84c54..4ea7157 100644 (file)
--- a/src/test/java/net/pterodactylus/irc/ReplyTest.java
+++ b/src/test/java/net/pterodactylus/irc/ReplyTest.java
@@ -38,7 +38,7 @@ public class ReplyTest {
 
                reply = Reply.parseLine("NOTICE AUTH :*** Processing your connection");
                assertThat(reply, notNullValue());
-               assertThat(reply.source(), is(Optional.<String>absent()));
+               assertThat(reply.source(), is(Optional.<Source>absent()));
                assertThat(reply.command(), is("NOTICE"));
                assertThat(reply.parameters(), hasSize(2));
                assertThat(reply.parameters().get(0), is("AUTH"));
@@ -46,7 +46,7 @@ public class ReplyTest {
 
                reply = Reply.parseLine(":ParaDMON!services@paraphysics.services PRIVMSG QshelTier :\u0001VERSION\u0001");
                assertThat(reply, notNullValue());
-               assertThat(reply.source(), is(Optional.of("ParaDMON!services@paraphysics.services")));
+               assertThat(reply.source(), is(Optional.of(Source.parseSource("ParaDMON!services@paraphysics.services"))));
                assertThat(reply.command(), is("PRIVMSG"));
                assertThat(reply.parameters(), hasSize(2));
                assertThat(reply.parameters().get(0), is("QshelTier"));