fix: track calls to HTMLInputElement.select() (#898)

testing-library · Apr 1, 2022 · 6d36828 · 6d36828
1 parent f2e8f8e
commit 6d36828
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/src/document/selection.ts b/src/document/selection.ts
@@ -69,6 +69,19 @@ export function prepareSelectionInterceptor(
       return {realArgs: v}
     },
   )
+
+  prepareInterceptor(
+    element,
+    'select',
+    function interceptorImpl(this: HTMLInputElement | HTMLTextAreaElement) {
+      this[UISelection] = {
+        anchorOffset: 0,
+        focusOffset: getUIValue(element).length,
+      }
+
+      return {realArgs: [] as []}
+    },
+  )
 }
 
 export function setUISelection(

diff --git a/tests/document/index.ts b/tests/document/index.ts
@@ -140,4 +140,23 @@ test('clear UI selection if selection is programmatically set', async () => {
   element.selectionStart = 2
   expect(getUISelection(element)).toHaveProperty('startOffset', 2)
   expect(getUISelection(element)).toHaveProperty('endOffset', 2)
+
+  setUISelection(element, {anchorOffset: 1, focusOffset: 2})
+  element.select()
+  expect(getUISelection(element)).toHaveProperty('startOffset', 0)
+  expect(getUISelection(element)).toHaveProperty('endOffset', 3)
+})
+
+test('select input without selectionRange support', () => {
+  const {element} = render<HTMLInputElement>(
+    `<input type="number" value="123"/>`,
+  )
+
+  prepare(element)
+
+  setUISelection(element, {focusOffset: 1})
+  element.select()
+
+  expect(getUISelection(element)).toHaveProperty('startOffset', 0)
+  expect(getUISelection(element)).toHaveProperty('endOffset', 3)
 })