Add tests for prefix mode mamager.

Change-Id: I66662eddf73c570a0f59e4ffd0f84ab5da15d195
Reviewed-on: http://gerrit.dmdirc.com/3958
Reviewed-by: Greg Holmes <greg@dmdirc.com>
Automatic-Compile: DMDirc Build Manager

src/com/dmdirc/parser/irc/PrefixModeManager.java

@@ -115,7 +115,7 @@ public class PrefixModeManager {
     }
 
     /**
-     * Adds a new mode.
+     * Adds a new mode. Modes must be added in increasing order of importance.
      *
      * @param mode The mode that appears in mode strings (e.g. 'o').
      * @param prefix The prefix that is used to show a user has the mode (e.g. '@')
@@ -198,6 +198,11 @@ public class PrefixModeManager {
             }
             result.append(existingMode);
         }
+
+        if (!found) {
+            result.append(mode);
+        }
+
         return result.toString();
     }
 }

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

@@ -0,0 +1,177 @@
+/*
+ * Copyright (c) 2006-2014 DMDirc Developers
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+package com.dmdirc.parser.irc;
+
+import org.junit.Before;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
+public class PrefixModeManagerTest {
+
+    private PrefixModeManager manager;
+
+    @Before
+    public void setup() {
+        manager = new PrefixModeManager();
+    }
+
+    @Test
+    public void testGetModeFor() {
+        manager.add('m', '/');
+        assertEquals('m', manager.getModeFor('/'));
+    }
+
+    @Test
+    public void testGetPrefixFor() {
+        manager.add('m', '/');
+        assertEquals('/', manager.getPrefixFor('m'));
+    }
+
+    @Test
+    public void testIsPrefixMode() {
+        manager.add('m', '/');
+        assertTrue(manager.isPrefixMode('m'));
+        assertFalse(manager.isPrefixMode('/'));
+        assertFalse(manager.isPrefixMode('o'));
+    }
+
+    @Test
+    public void testIsPrefix() {
+        manager.add('m', '/');
+        assertFalse(manager.isPrefix('m'));
+        assertTrue(manager.isPrefix('/'));
+        assertFalse(manager.isPrefix('+'));
+    }
+
+    @Test
+    public void testGetPrefixesFor() {
+        manager.add('m', '/');
+        manager.add('n', '+');
+        assertEquals("//++//", manager.getPrefixesFor("mmnnmm"));
+        assertEquals("", manager.getPrefixesFor(""));
+        assertEquals("/", manager.getPrefixesFor("m"));
+        assertEquals("+/", manager.getPrefixesFor("nm"));
+    }
+
+    @Test
+    public void testGetModes() {
+        assertEquals(0, manager.getModes().size());
+        manager.add('m', '/');
+        assertEquals(1, manager.getModes().size());
+        assertTrue(manager.getModes().contains('m'));
+        manager.add('n', '+');
+        assertEquals(2, manager.getModes().size());
+        assertTrue(manager.getModes().contains('m'));
+        assertTrue(manager.getModes().contains('n'));
+        manager.clear();
+        assertEquals(0, manager.getModes().size());
+    }
+
+    @Test
+    public void testIsOpped() {
+        assertFalse(manager.isOpped(""));
+        manager.add('k', '_');
+
+        // If there isn't a 'v' mode, then any mode equals opped
+        assertFalse(manager.isOpped(""));
+        assertTrue(manager.isOpped("k"));
+
+        manager.add('v', '+');
+        assertFalse(manager.isOpped(""));
+        assertFalse(manager.isOpped("k"));
+        assertFalse(manager.isOpped("v"));
+
+        manager.add('o', '@');
+        assertFalse(manager.isOpped(""));
+        assertFalse(manager.isOpped("k"));
+        assertFalse(manager.isOpped("v"));
+        assertTrue(manager.isOpped("o"));
+
+        manager.add('z', '*');
+        assertFalse(manager.isOpped(""));
+        assertFalse(manager.isOpped("k"));
+        assertFalse(manager.isOpped("v"));
+        assertTrue(manager.isOpped("o"));
+        assertTrue(manager.isOpped("z"));
+    }
+
+    @Test
+    public void testInsertMode() {
+        manager.add('m', '/');
+        manager.add('n', '+');
+        manager.add('o', '@');
+
+        // Already exist in modes
+        assertEquals("mn", manager.insertMode("mn", 'm'));
+        assertEquals("m", manager.insertMode("m", 'm'));
+
+        // Most important ones go at the start
+        assertEquals("om", manager.insertMode("m", 'o'));
+        assertEquals("nm", manager.insertMode("m", 'n'));
+        assertEquals("onm", manager.insertMode("nm", 'o'));
+
+        // Inserting elsewhere
+        assertEquals("om", manager.insertMode("o", 'm'));
+        assertEquals("onm", manager.insertMode("on", 'm'));
+        assertEquals("onm", manager.insertMode("om", 'n'));
+        assertEquals("nm", manager.insertMode("n", 'm'));
+
+        // Build up a string from scratch
+        assertEquals("onm",
+                manager.insertMode(manager.insertMode(manager.insertMode("", 'n'), 'm'), 'o'));
+    }
+
+    @Test
+    public void testCompareImportantModes() {
+        // No modes, must be equal
+        assertEquals(0, manager.compareImportantModes("", ""));
+
+        manager.add('m', '/');
+        assertEquals(0, manager.compareImportantModes("", ""));
+        assertTrue(manager.compareImportantModes("m", "") > 0);
+        assertTrue(manager.compareImportantModes("", "m") < 0);
+
+        manager.add('n', '+');
+        assertEquals(0, manager.compareImportantModes("", ""));
+        assertTrue(manager.compareImportantModes("m", "") > 0);
+        assertTrue(manager.compareImportantModes("", "m") < 0);
+        assertTrue(manager.compareImportantModes("n", "") > 0);
+        assertTrue(manager.compareImportantModes("", "n") < 0);
+        assertTrue(manager.compareImportantModes("n", "m") > 0);
+        assertTrue(manager.compareImportantModes("m", "n") < 0);
+        assertTrue(manager.compareImportantModes("nm", "m") > 0);
+        assertTrue(manager.compareImportantModes("m", "nm") < 0);
+
+        manager.add('o', '@');
+        assertEquals(0, manager.compareImportantModes("", ""));
+        assertTrue(manager.compareImportantModes("nm", "m") > 0);
+        assertTrue(manager.compareImportantModes("m", "nm") < 0);
+        assertTrue(manager.compareImportantModes("o", "nm") > 0);
+        assertTrue(manager.compareImportantModes("nm", "om") < 0);
+        assertTrue(manager.compareImportantModes("nm", "on") < 0);
+    }
+
+}