fix: Use frame query that supports shadow dom (#492)

dequelabs · Aug 22, 2017 · 94008ff · 94008ff
1 parent 508fae2
commit 94008ff
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/lib/core/public/cleanup-plugins.js b/lib/core/public/cleanup-plugins.js
@@ -23,9 +23,11 @@ function cleanupPlugins(resolve, reject) {
     });
   });
 
-  axe.utils.toArray(document.querySelectorAll('frame, iframe')).forEach(function (frame) {
+  var flattenedTree = axe.utils.getFlattenedTree(document.body)
+
+  axe.utils.querySelectorAll(flattenedTree, 'iframe, frame').forEach(function (node) {
     q.defer(function (res, rej) {
-      return axe.utils.sendCommandToFrame(frame, {
+      return axe.utils.sendCommandToFrame(node.actualNode, {
         command: 'cleanup-plugin'
       }, res, rej);
     });

diff --git a/lib/core/public/plugins.js b/lib/core/public/plugins.js
@@ -43,4 +43,4 @@ Plugin.prototype.add = function (impl) {
 axe.registerPlugin = function (plugin) {
 	'use strict';
 	axe.plugins[plugin.id] = new Plugin(plugin);
-};
+};