1 files changed, 78 insertions, 0 deletions
diff --git a/toolkit/components/autocomplete/tests/unit/test_autofillSelectedPopupIndex.js b/toolkit/components/autocomplete/tests/unit/test_autofillSelectedPopupIndex.js
new file mode 100644
index 000000000..5fb93abc1
--- /dev/null
+++ b/toolkit/components/autocomplete/tests/unit/test_autofillSelectedPopupIndex.js
@@ -0,0 +1,78 @@
+"use strict";
+
+add_task(function* sameCaseAsMatch() {
+  yield runTest("moz");
+});
+
+add_task(function* differentCaseFromMatch() {
+  yield runTest("MOZ");
+});
+
+function* runTest(searchStr) {
+  let matches = [
+    "mozilla.org",
+    "example.com",
+  ];
+  let result = new AutoCompleteResultBase(matches);
+  result.defaultIndex = 0;
+
+  let search = new AutoCompleteSearchBase("search", result);
+  registerAutoCompleteSearch(search);
+
+  let input = new AutoCompleteInputBase([search.name]);
+  input.completeSelectedIndex = true;
+  input.completeDefaultIndex = true;
+
+  // Start off with the search string in the input.  The selection must be
+  // collapsed and the caret must be at the end to trigger autofill below.
+  input.textValue = searchStr;
+  input.selectTextRange(searchStr.length, searchStr.length);
+  Assert.equal(input.selectionStart, searchStr.length,
+               "Selection should start at the end of the input");
+  Assert.equal(input.selectionEnd, searchStr.length,
+               "Selection should end at the end of the input");
+
+  let controller = Cc["@mozilla.org/autocomplete/controller;1"].
+                   createInstance(Ci.nsIAutoCompleteController);
+  controller.input = input;
+  input.controller = controller;
+
+  // Start a search.
+  yield new Promise(resolve => {
+    controller.startSearch(searchStr);
+    input.onSearchComplete = () => {
+      // The first match should have autofilled, but the case of the search
+      // string should be preserved.
+      let expectedValue = searchStr + matches[0].substr(searchStr.length);
+      Assert.equal(input.textValue, expectedValue,
+                   "Should have autofilled");
+      Assert.equal(input.selectionStart, searchStr.length,
+                   "Selection should start after search string");
+      Assert.equal(input.selectionEnd, expectedValue.length,
+                   "Selection should end at the end of the input");
+      resolve();
+    };
+  });
+
+  // Key down to select the second match in the popup.
+  controller.handleKeyNavigation(Ci.nsIDOMKeyEvent.DOM_VK_DOWN);
+  let expectedValue = matches[1];
+  Assert.equal(input.textValue, expectedValue,
+               "Should have filled second match");
+  Assert.equal(input.selectionStart, expectedValue.length,
+               "Selection should start at the end of the input");
+  Assert.equal(input.selectionEnd, expectedValue.length,
+               "Selection should end at the end of the input");
+
+  // Key up to select the first match again.  The input should be restored
+  // exactly as it was when the first match was autofilled above: the search
+  // string's case should be preserved, and the selection should be preserved.
+  controller.handleKeyNavigation(Ci.nsIDOMKeyEvent.DOM_VK_UP);
+  expectedValue = searchStr + matches[0].substr(searchStr.length);
+  Assert.equal(input.textValue, expectedValue,
+               "Should have filled first match again");
+  Assert.equal(input.selectionStart, searchStr.length,
+               "Selection should start after search string again");
+  Assert.equal(input.selectionEnd, expectedValue.length,
+               "Selection should end at the end of the input again");
+}