fix(flatten-tree): do not call deprecated getDistributedNodes (#1729)

dequelabs · Jul 31, 2019 · 46a5d11 · 46a5d11
1 parent ecc99f2
commit 46a5d11
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/lib/core/utils/flattened-tree.js b/lib/core/utils/flattened-tree.js
@@ -80,7 +80,10 @@ function flattenTree(node, shadowId, parent) {
 
 		return [retVal];
 	} else {
-		if (nodeName === 'content') {
+		if (
+			nodeName === 'content' &&
+			typeof node.getDistributedNodes === 'function'
+		) {
 			realArray = Array.from(node.getDistributedNodes());
 			return realArray.reduce((res, child) => {
 				return reduceShadowDOM(res, child, parent);

diff --git a/test/core/utils/flattened-tree.js b/test/core/utils/flattened-tree.js
@@ -291,6 +291,21 @@ describe('axe.utils.getFlattenedTree', function() {
 				assert.isDefined(vNode);
 				assert.equal(virtualDOM[0].actualNode, vNode.actualNode);
 			});
+			it('should not throw if getDistributedNodes is missing', function() {
+				var getDistributedNodes = fixture.getDistributedNodes;
+				fixture.getDistributedNodes = undefined;
+				try {
+					var virtualDOM = axe.utils.getFlattenedTree(fixture);
+					var vNode = axe.utils.getNodeFromTree(
+						virtualDOM[0],
+						virtualDOM[0].actualNode
+					);
+					assert.isDefined(vNode);
+					assert.equal(virtualDOM[0].actualNode, vNode.actualNode);
+				} finally {
+					fixture.getDistributedNodes = getDistributedNodes;
+				}
+			});
 		});
 	} else {
 		it('does not throw when slot elements are used', function() {