Don’t use TestNG’s Test.

[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..ca69196 100644 (file)
--- a/src/test/java/net/pterodactylus/irc/ReplyTest.java
+++ b/src/test/java/net/pterodactylus/irc/ReplyTest.java
@@ -23,7 +23,7 @@ import static org.hamcrest.Matchers.is;
 import static org.hamcrest.Matchers.notNullValue;
 
 import com.google.common.base.Optional;
-import org.testng.annotations.Test;
+import org.junit.Test;
 
 /**
  * Tests for {@link Reply}.
@@ -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"));