Fix uncaught DOMException error

atlassian · Feb 18, 2020 · ff8b2d4 · ff8b2d4
1 parent 4721a51
commit ff8b2d4
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 3 deletions.
diff --git a/src/view/use-announcer/use-announcer.js b/src/view/use-announcer/use-announcer.js
@@ -42,9 +42,11 @@ export default function useAnnouncer(contextId: ContextId): Announce {
         // unmounting after a timeout to let any announcements
         // during a mount be published
         setTimeout(function remove() {
-          // not clearing the ref as it might have been set by a new effect
-          getBodyElement().removeChild(el);
-
+          // checking if element exists before remove to prevent errors
+          if (getBodyElement().contains(el)) {
+            // not clearing the ref as it might have been set by a new effect
+            getBodyElement().removeChild(el);
+          }
           // if el was the current ref - clear it so that
           // we can get a warning if announce is called
           if (el === ref.current) {

diff --git a/test/unit/view/announcer.spec.js b/test/unit/view/announcer.spec.js
@@ -60,6 +60,24 @@ it('should remove the element when unmounting after a timeout', () => {
   expect(getElement('5')).not.toBeTruthy();
 });
 
+it('should not remove the element when unmounting after a timeout if the element does not exist', () => {
+  const { unmount } = render(
+    <WithAnnouncer contextId="5">{getMock()}</WithAnnouncer>,
+  );
+
+  const doc = new DOMParser().parseFromString(
+    '<!doctype html><title>wat</title>',
+    'text/html',
+  );
+  document.write(doc);
+  unmount();
+  expect(() => jest.runOnlyPendingTimers()).not.toThrow(
+    new DOMException(
+      "Failed to execute 'removeChild' on 'Node': The node to be removed is not a child of this node.",
+    ),
+  );
+});
+
 it('should set the text content of the announcement element', () => {
   // arrange
   const mock = getMock();