Tests for getting / restoring selections across iframes

facebook · Dec 2, 2016 · b48f472 · b48f472
1 parent e948e95
commit b48f472
Showing 1 changed file with 51 additions and 2 deletions.
diff --git a/src/renderers/dom/client/__tests__/ReactInputSelection-test.js b/src/renderers/dom/client/__tests__/ReactInputSelection-test.js
@@ -22,6 +22,12 @@ describe('ReactInputSelection', () => {
     var instance = ReactTestUtils.renderIntoDocument(element);
     return ReactDOM.findDOMNode(instance);
   };
+  var makeGetSelection = (win = window) => () => ({
+    anchorNode: win.document.activeElement,
+    focusNode: win.document.activeElement,
+    anchorOffset: win.document.activeElement && win.document.activeElement.selectionStart,
+    focusOffset: win.document.activeElement && win.document.activeElement.selectionEnd,
+  });
 
   beforeEach(() => {
     jest.resetModuleRegistry();
@@ -108,7 +114,7 @@ describe('ReactInputSelection', () => {
     });
 
     it('gets selection on inputs in iframes', () => {
-      var iframe = document.createElement('iframe');
+      const iframe = document.createElement('iframe');
       document.body.appendChild(iframe);
       const input = document.createElement('input');
       input.value = textValue;
@@ -135,7 +141,7 @@ describe('ReactInputSelection', () => {
     });
 
     it('sets selection on inputs in iframes', () => {
-      var iframe = document.createElement('iframe');
+      const iframe = document.createElement('iframe');
       document.body.appendChild(iframe);
       const input = document.createElement('input');
       input.value = textValue;
@@ -150,6 +156,9 @@ describe('ReactInputSelection', () => {
 
   describe('getSelectionInformation/restoreSelection', () => {
     it('gets and restores selection for inputs that get remounted', () => {
+      // Mock window getSelection if needed
+      var originalGetSelection = window.getSelection;
+      window.getSelection = window.getSelection || makeGetSelection(window);
       var input = document.createElement('input');
       input.value = textValue;
       document.body.appendChild(input);
@@ -173,6 +182,46 @@ describe('ReactInputSelection', () => {
       expect(input.selectionEnd).toBe(10);
 
       document.body.removeChild(input);
+      window.getSelection = originalGetSelection;
+    });
+
+    it('gets and restores selection for inputs in an iframe that get remounted', () => {
+      var iframe = document.createElement('iframe');
+      document.body.appendChild(iframe);
+      var iframeDoc = iframe.contentDocument;
+      var iframeWin = iframeDoc.defaultView;
+      // Mock window and iframe getSelection if needed
+      var originalGetSelection = window.getSelection;
+      var originalIframeGetSelection = iframeWin.getSelection;
+      window.getSelection = window.getSelection || makeGetSelection(window);
+      iframeWin.getSelection = iframeWin.getSelection || makeGetSelection(iframeWin);
+
+      var input = document.createElement('input');
+      input.value = textValue;
+      iframeDoc.body.appendChild(input);
+      input.focus();
+      input.selectionStart = 1;
+      input.selectionEnd = 10;
+      var selectionInfo = ReactInputSelection.getSelectionInformation();
+      expect(selectionInfo.focusedElement).toBe(input);
+      expect(selectionInfo.activeElements[0].selectionRange).toEqual({start: 1, end: 10});
+      expect(document.activeElement).toBe(iframe);
+      expect(iframeDoc.activeElement).toBe(input);
+
+      input.setSelectionRange(0, 0);
+      iframeDoc.body.removeChild(input);
+      expect(iframeDoc.activeElement).not.toBe(input);
+      expect(input.selectionStart).not.toBe(1);
+      expect(input.selectionEnd).not.toBe(10);
+      iframeDoc.body.appendChild(input);
+      ReactInputSelection.restoreSelection(selectionInfo);
+      expect(iframeDoc.activeElement).toBe(input);
+      expect(input.selectionStart).toBe(1);
+      expect(input.selectionEnd).toBe(10);
+
+      document.body.removeChild(iframe);
+      window.getSelection = originalGetSelection;
+      iframeWin.getSelection = originalIframeGetSelection;
     });
   });
 });