1 files changed, 148 insertions, 0 deletions
diff --git a/comm/chat/protocols/irc/test/test_tryNewNick.js b/comm/chat/protocols/irc/test/test_tryNewNick.js
new file mode 100644
index 0000000000..dbd2692d4c
--- /dev/null
+++ b/comm/chat/protocols/irc/test/test_tryNewNick.js
@@ -0,0 +1,148 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+var { ircProtocol } = ChromeUtils.importESModule(
+  "resource:///modules/irc.sys.mjs"
+);
+var { ircAccount } = ChromeUtils.importESModule(
+  "resource:///modules/ircAccount.sys.mjs"
+);
+
+var fakeProto = {
+  id: "fake-proto",
+  options: { alternateNicks: "" },
+  _getOptionDefault(aOption) {
+    return this.options[aOption];
+  },
+  usernameSplits: ircProtocol.prototype.usernameSplits,
+  splitUsername: ircProtocol.prototype.splitUsername,
+};
+
+function test_tryNewNick() {
+  const testData = {
+    clokep: "clokep1",
+    clokep1: "clokep2",
+    clokep10: "clokep11",
+    clokep0: "clokep1",
+    clokep01: "clokep02",
+    clokep09: "clokep10",
+
+    // Now put a number in the "first part".
+    clo1kep: "clo1kep1",
+    clo1kep1: "clo1kep2",
+    clo1kep10: "clo1kep11",
+    clo1kep0: "clo1kep1",
+    clo1kep01: "clo1kep02",
+    clo1kep09: "clo1kep10",
+  };
+
+  let account = new ircAccount(fakeProto, {
+    name: "clokep@instantbird.org",
+  });
+  account.LOG = function (aStr) {};
+  account.normalize = aStr => aStr;
+
+  for (let currentNick in testData) {
+    account._sentNickname = currentNick;
+    account.sendMessage = (aCommand, aNewNick) =>
+      equal(aNewNick, testData[currentNick]);
+
+    account.tryNewNick(currentNick);
+  }
+
+  run_next_test();
+}
+
+// This tests a bunch of cases near the max length by maintaining the state
+// through a series of test nicks.
+function test_maxLength() {
+  let testData = [
+    // First try adding a digit, as normal.
+    ["abcdefghi", "abcdefghi1"],
+    // The "received" nick back will now be the same though, so it was too long.
+    ["abcdefghi", "abcdefgh1"],
+    // And just ensure we're iterating properly.
+    ["abcdefgh1", "abcdefgh2"],
+    ["abcdefgh2", "abcdefgh3"],
+    ["abcdefgh3", "abcdefgh4"],
+    ["abcdefgh4", "abcdefgh5"],
+    ["abcdefgh5", "abcdefgh6"],
+    ["abcdefgh6", "abcdefgh7"],
+    ["abcdefgh7", "abcdefgh8"],
+    ["abcdefgh8", "abcdefgh9"],
+    ["abcdefgh9", "abcdefgh10"],
+    ["abcdefgh1", "abcdefg10"],
+    ["abcdefg10", "abcdefg11"],
+    ["abcdefg99", "abcdefg100"],
+    ["abcdefg10", "abcdef100"],
+    ["a99999999", "a100000000"],
+    ["a10000000", "a00000000"],
+  ];
+
+  let account = new ircAccount(fakeProto, {
+    name: "clokep@instantbird.org",
+  });
+  account.LOG = function (aStr) {};
+  account._sentNickname = "abcdefghi";
+  account.normalize = aStr => aStr;
+
+  for (let currentNick of testData) {
+    account.sendMessage = (aCommand, aNewNick) =>
+      equal(aNewNick, currentNick[1]);
+
+    account.tryNewNick(currentNick[0]);
+  }
+
+  run_next_test();
+}
+
+function test_altNicks() {
+  const altNicks = ["clokep_", "clokep|"];
+  const testData = {
+    // Test account nick.
+    clokep: [altNicks, "clokep_"],
+    // Test first element in list.
+    clokep_: [altNicks, "clokep|"],
+    // Test last element in list.
+    "clokep|": [altNicks, "clokep|1"],
+    // Test element not in list with number at end.
+    clokep1: [altNicks, "clokep2"],
+
+    // Test messy alternatives.
+    "clokep[": [" clokep ,\n clokep111,,,\tclokep[, clokep_", "clokep_"],
+  };
+
+  let account = new ircAccount(fakeProto, {
+    name: "clokep@instantbird.org",
+  });
+  account.LOG = function (aStr) {};
+  account.normalize = aStr => aStr;
+
+  for (let currentNick in testData) {
+    // Only one pref is touched in here, override the default to return
+    // what this test needs.
+    account.getString = function (aStr) {
+      let data = testData[currentNick][0];
+      if (Array.isArray(data)) {
+        return data.join(",");
+      }
+      return data;
+    };
+    account._sentNickname = currentNick;
+
+    account.sendMessage = (aCommand, aNewNick) =>
+      equal(aNewNick, testData[currentNick][1]);
+
+    account.tryNewNick(currentNick);
+  }
+
+  run_next_test();
+}
+
+function run_test() {
+  add_test(test_tryNewNick);
+  add_test(test_maxLength);
+  add_test(test_altNicks);
+
+  run_next_test();
+}