Add tests for process001

irc/test/com/dmdirc/parser/irc/processors/Process001Test.java

@@ -22,27 +22,117 @@
 
 package com.dmdirc.parser.irc.processors;
 
-import com.dmdirc.harness.parser.TestParser;
+import com.dmdirc.parser.common.ParserError;
+import com.dmdirc.parser.irc.IRCClientInfo;
+import com.dmdirc.parser.irc.IRCParser;
+import com.dmdirc.parser.irc.ProcessingManager;
+
+import org.junit.Before;
 import org.junit.Test;
-import static org.junit.Assert.*;
+import org.junit.runner.RunWith;
+import org.mockito.ArgumentCaptor;
+import org.mockito.Mock;
+import org.mockito.runners.MockitoJUnitRunner;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assume.assumeFalse;
+import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.anyBoolean;
+import static org.mockito.Matchers.anyString;
+import static org.mockito.Matchers.eq;
+import static org.mockito.Mockito.doAnswer;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
 
+@RunWith(MockitoJUnitRunner.class)
 public class Process001Test {
-       
+
+    @Mock private IRCParser parser;
+    @Mock private ProcessingManager processingManager;
+    @Mock private IRCClientInfo localClient;
+    private Process001 processor;
+
+    @Before
+    public void setup() {
+        when(parser.getLocalClient()).thenReturn(localClient);
+        processor = new Process001(parser, processingManager);
+    }
+
+    @Test
+    public void testSets001Received() {
+        when(localClient.isFake()).thenReturn(true);
+        assumeFalse(parser.got001);
+        processor.process("001", ":test.server.com", "001", "userName", "Hello!");
+        assertTrue(parser.got001);
+    }
+
+    @Test
+    public void testUpdatesServerName() {
+        when(localClient.isFake()).thenReturn(true);
+        processor.process("001", ":test.server.com", "001", "userName", "Hello!");
+        verify(parser).updateServerName("test.server.com");
+    }
+
+    @Test
+    public void testAddsLocalClientIfFake() {
+        when(localClient.isFake()).thenReturn(true);
+        processor.process("001", ":test.server.com", "001", "userName", "Hello!");
+        verify(parser).addClient(localClient);
+    }
+
     @Test
-    public void testDuplicate001() {
-        final TestParser tp = new TestParser();
-        
-        assertTrue(tp.getLocalClient().isFake());
-        
-        tp.injectConnectionStrings();
-        
-        assertEquals("nick", tp.getLocalClient().getNickname());
-        assertFalse(tp.getLocalClient().isFake());
-        
-        tp.injectLine(":server 001 nick2 :Crazy second 001 for you, nick2");
-        
-        assertEquals("nick2", tp.getLocalClient().getNickname());
-        assertFalse(tp.getLocalClient().isFake());
+    public void testUpdatesLocalClientIfFake() {
+        when(localClient.isFake()).thenReturn(true);
+        processor.process("001", ":test.server.com", "001", "userName", "Hello!");
+        verify(localClient).setUserBits(eq("userName"), eq(true), anyBoolean());
+        verify(localClient).setFake(false);
+    }
+
+    @Test
+    public void testIgnoresDuplicate001WithSameNick() {
+        when(localClient.getNickname()).thenReturn("UsernaME");
+        processor.process("001", ":test.server.com", "001", "userName", "Hello!");
+        verify(parser, never()).addClient(any());
+        verify(parser, never()).forceRemoveClient(any());
+    }
+
+    @Test
+    public void testReplacesLocalUserOnDuplicate001() {
+        when(localClient.getNickname()).thenReturn("UsernaME");
+        processor.process("001", ":test.server.com", "001", "newName", "Hello!");
+        verify(parser).forceRemoveClient(localClient);
+        verify(localClient).setUserBits(eq("newName"), eq(true), anyBoolean());
+        verify(parser).addClient(localClient);
+    }
+
+    @Test
+    public void testRaisesFatalErrorIfDuplicate001CausesNicknameCollision() {
+        setupLocalClientToTrackNicknameChanges();
+
+        when(parser.isKnownClient("newName")).thenReturn(true);
+        when(parser.getClient("newName")).thenReturn(mock(IRCClientInfo.class));
+
+        processor.process("001", ":test.server.com", "001", "newName", "Hello!");
+
+        final ArgumentCaptor<ParserError> errorCaptor = ArgumentCaptor.forClass(ParserError.class);
+        verify(parser).callErrorInfo(errorCaptor.capture());
+        assertEquals(ParserError.ERROR_FATAL, errorCaptor.getValue().getLevel());
+    }
+
+    /**
+     * Ensures that the local client mock returns new nicknames after setUserBits is called.
+     */
+    private void setupLocalClientToTrackNicknameChanges() {
+        final StringBuilder nickname = new StringBuilder("UserName");
+        when(localClient.getNickname()).thenAnswer(invocation -> nickname.toString());
+        doAnswer(invocation -> {
+            nickname.setLength(0);
+            nickname.append(invocation.getArgumentAt(0, String.class));
+            return null;
+        }).when(localClient).setUserBits(anyString(), eq(true), anyBoolean());
     }
 
 }