Tests for getting / restoring selections across iframes

src/renderers/dom/client/__tests__/ReactInputSelection-test.js

@@ -174,5 +174,39 @@ describe('ReactInputSelection', () => {
 
       document.body.removeChild(input);
     });
+
+    it('gets and restores selection for inputs in an iframe that get remounted', () => {
+      var iframe = document.createElement('iframe');
+      iframe.setAttribute('tabIndex', 0);
+      document.body.appendChild(iframe);
+      iframe.focus();
+      var iframeDoc = iframe.contentDocument;
+      var input = document.createElement('input');
+      input.value = textValue;
+      iframeDoc.body.appendChild(input);
+      // Focus iframe first to get around jsdom limitations
+      iframe.focus();
+      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);
+    });
   });
 });