123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192 |
- /*
- * Copyright (c) 2006-2012 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.addons.identd;
-
- import com.dmdirc.harness.TestConfigManagerMap;
- import com.dmdirc.config.IdentityManager;
-
- import org.junit.BeforeClass;
- import org.junit.Test;
- import static org.junit.Assert.*;
- import static org.mockito.Mockito.*;
-
- public class IdentClientTest {
-
- @BeforeClass
- public static void setUpClass() throws Exception {
- IdentityManager.getIdentityManager().initialise();
- }
-
- protected IdentClient getClient() {
- IdentdPlugin plugin = mock(IdentdPlugin.class);
-
- when(plugin.getDomain()).thenReturn("plugin-Identd");
-
- return new IdentClient(null, null, plugin);
- }
-
- @Test
- public void testInvalidIdent() {
- final String response = getClient().getIdentResponse("invalid request!",
- IdentityManager.getGlobalConfig());
-
- assertContains("Illegal requests must result in an ERROR response",
- response, "ERROR");
- }
-
- @Test
- public void testQuoting() {
- final String response = getClient().getIdentResponse("in\\valid:invalid",
- IdentityManager.getGlobalConfig());
-
- assertStartsWith("Special chars in illegal requests must be quoted",
- response, "in\\\\valid\\:invalid");
- }
-
- @Test
- public void testQuoting2() {
- final String response = getClient().getIdentResponse("in\\\\valid\\ inv\\:alid",
- IdentityManager.getGlobalConfig());
-
- assertStartsWith("Escaped characters in illegal requests shouldn't be doubly-escaped",
- response, "in\\\\valid\\ inv\\:alid");
- }
-
- @Test
- public void testNonNumericPort() {
- final String response = getClient().getIdentResponse("abc, def",
- IdentityManager.getGlobalConfig());
-
- assertContains("Non-numeric ports must result in an ERROR response",
- response, "ERROR");
- assertStartsWith("Specified ports must be returned in the response",
- response.replaceAll("\\s+", ""), "abc,def:");
- }
-
- private void doPortTest(final String ports) {
- final String response = getClient().getIdentResponse(ports,
- IdentityManager.getGlobalConfig());
-
- assertContains("Illegal ports must result in an ERROR response",
- response, "ERROR");
- assertContains("Illegal ports must result in an INVALID-PORT response",
- response, "INVALID-PORT");
- assertStartsWith("Port numbers must be returned as part of the response",
- response.replaceAll("\\s+", ""), ports.replaceAll("\\s+", ""));
- }
-
- @Test
- public void testOutOfRangePorts() {
- doPortTest("0, 50");
- doPortTest("65536, 50");
- doPortTest("50, 0");
- doPortTest("50, 65536");
- }
-
- @Test
- public void testAlwaysOn() {
- final TestConfigManagerMap tcm = new TestConfigManagerMap();
- tcm.settings.put("plugin-Identd.advanced.alwaysOn", "false");
-
- final String response = getClient().getIdentResponse("50, 50", tcm);
- assertContains("Unknown port requests must return an ERROR response",
- response, "ERROR");
- assertContains("Unknown port requests must return a NO-USER response",
- response, "NO-USER");
- }
-
- @Test
- public void testHidden() {
- final TestConfigManagerMap tcm = new TestConfigManagerMap();
- tcm.settings.put("plugin-Identd.advanced.alwaysOn", "true");
- tcm.settings.put("plugin-Identd.advanced.isHiddenUser", "true");
-
- final String response = getClient().getIdentResponse("50, 50", tcm);
- assertContains("Hidden requests must return an ERROR response",
- response, "ERROR");
- assertContains("Hidden requests must return a HIDDEN-USER response",
- response, "HIDDEN-USER");
- }
-
- @Test
- public void testSystemNameQuoting() {
- final TestConfigManagerMap tcm = new TestConfigManagerMap();
- tcm.settings.put("plugin-Identd.advanced.alwaysOn", "true");
- tcm.settings.put("plugin-Identd.advanced.isHiddenUser", "false");
- tcm.settings.put("plugin-Identd.advanced.useCustomSystem", "true");
- tcm.settings.put("plugin-Identd.advanced.customSystem", "a:b\\c,d");
- tcm.settings.put("plugin-Identd.general.useCustomName", "false");
- tcm.settings.put("plugin-Identd.general.customName", "");
-
- final String response = getClient().getIdentResponse("50, 50", tcm);
- assertContains("Special characters must be quoted in system names",
- response, "a\\:b\\\\c\\,d");
- }
-
- @Test
- public void testCustomNameQuoting() {
- final TestConfigManagerMap tcm = new TestConfigManagerMap();
- tcm.settings.put("plugin-Identd.advanced.alwaysOn", "true");
- tcm.settings.put("plugin-Identd.advanced.isHiddenUser", "false");
- tcm.settings.put("plugin-Identd.advanced.useCustomSystem", "false");
- tcm.settings.put("plugin-Identd.advanced.customSystem", "");
- tcm.settings.put("plugin-Identd.general.useCustomName", "true");
- tcm.settings.put("plugin-Identd.general.customName", "a:b\\c,d");
-
- final String response = getClient().getIdentResponse("50, 50", tcm);
- assertContains("Special characters must be quoted in custom names",
- response, "a\\:b\\\\c\\,d");
- }
-
- @Test
- public void testCustomNames() {
- final TestConfigManagerMap tcm = new TestConfigManagerMap();
- tcm.settings.put("plugin-Identd.advanced.alwaysOn", "true");
- tcm.settings.put("plugin-Identd.advanced.isHiddenUser", "false");
- tcm.settings.put("plugin-Identd.advanced.useCustomSystem", "true");
- tcm.settings.put("plugin-Identd.advanced.customSystem", "system");
- tcm.settings.put("plugin-Identd.general.useCustomName", "true");
- tcm.settings.put("plugin-Identd.general.customName", "name");
-
- final String response = getClient().getIdentResponse("50, 60", tcm);
- final String[] bits = response.split(":");
-
- assertTrue("Responses must include port pair",
- bits[0].matches("\\s*50\\s*,\\s*60\\s*"));
- assertEquals("Positive response must include USERID",
- "USERID", bits[1].trim());
- assertEquals("Must use custom system name", "system", bits[2].trim());
- assertEquals("Must use custom name", "name", bits[3].trim());
- }
-
- private static void assertContains(final String msg, final String haystack,
- final String needle) {
- assertTrue(msg, haystack.indexOf(needle) > -1);
- }
-
- private static void assertStartsWith(final String msg, final String haystack,
- final String needle) {
- assertTrue(msg, haystack.startsWith(needle));
- }
-
- }
|