Fix unit tests.

Change-Id: Ic5fe088387c4d67035dc578de31ffd02ddda5941
Reviewed-on: http://gerrit.dmdirc.com/623
Automatic-Compile: Shane Mc Cormack <shane@dmdirc.com>
Automatic-Compile: Gregory Holmes <greg@dmdirc.com>
Reviewed-by: Gregory Holmes <greg@dmdirc.com>

test/com/dmdirc/parser/irc/ClientInfoTest.java

@@ -22,6 +22,7 @@
 
 package com.dmdirc.parser.irc;
 
+import com.dmdirc.parser.common.AwayState;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -100,19 +101,19 @@ public class ClientInfoTest {
     @Test
     public void testAwayState() {
         final IRCClientInfo ci = new IRCClientInfo(new IRCParser(), "nick!ident@host");
-        assertFalse(ci.getAwayState());
-        ci.setAwayState(true);
-        assertTrue(ci.getAwayState());
+        assertFalse(ci.getAwayState() == AwayState.AWAY);
+        ci.setAwayState(AwayState.HERE);
+        assertTrue(ci.getAwayState() == AwayState.HERE);
     }
     
     @Test
     public void testAwayReason() {
         final IRCClientInfo ci = new IRCClientInfo(new IRCParser(), "nick!ident@host");
-        ci.setAwayState(true);
+        ci.setAwayState(AwayState.AWAY);
         ci.setAwayReason("away reason");
         
         assertEquals("away reason", ci.getAwayReason());
-        ci.setAwayState(false);
+        ci.setAwayState(AwayState.HERE);
         assertEquals("", ci.getAwayReason());
     }
     

test/com/dmdirc/parser/irc/ProcessWhoTest.java

@@ -22,6 +22,7 @@
 
 package com.dmdirc.parser.irc;
 
+import com.dmdirc.parser.common.AwayState;
 import com.dmdirc.harness.parser.TestParser;
 import com.dmdirc.parser.common.CallbackNotFoundException;
 import com.dmdirc.parser.interfaces.callbacks.AwayStateListener;
@@ -41,11 +42,11 @@ public class ProcessWhoTest {
 
         parser.injectLine(":server 352 nick #DMDirc_testing nick2 host2 server nick2 G@ :0 rn");
 
-        verify(test, never()).onAwayState((IRCParser) anyObject(),anyBoolean(), anyString());
+        verify(test, never()).onAwayState((IRCParser) anyObject(), (AwayState) anyObject(), (AwayState) anyObject() , anyString());
 
         parser.injectLine(":server 352 nick #DMDirc_testing nick host server nick G@ :0 rn");
 
-        verify(test).onAwayState(parser, true, "");
+        verify(test).onAwayState(parser, AwayState.UNKNOWN, AwayState.AWAY, "");
     }
 
 }