tools/wptrunner/wptrunner/testdriver-extra.js

"use strict";

(function(){
    let pending_resolve = null;
    let pending_reject = null;
    window.addEventListener("message", function(event) {
        const data = event.data;

        if (typeof data !== "object" && data !== null) {
            return;
        }

        if (data.type !== "testdriver-complete") {
            return;
        }

        if (data.status === "success") {
            pending_resolve();
        } else {
            pending_reject();
        }
    });

    const get_frame = function(element, frame) {
        let foundFrame = frame;
        let frameDocument = frame == window ? window.document : frame.contentDocument;
        if (!frameDocument.contains(element)) {
          foundFrame = null;
          let frames = document.getElementsByTagName("iframe");
          for (let i = 0; i < frames.length; i++) {
            if (get_frame(element, frames[i])) {
              foundFrame = frames[i];
              break;
            }
          }
        }
        return foundFrame;
    };

    const get_selector = function(element) {
        let selector;

        if (element.id) {
            const id = element.id;

            selector = "#";
            // escape everything, because it's easy to implement
            for (let i = 0, len = id.length; i < len; i++) {
                selector += '\\' + id.charCodeAt(i).toString(16) + ' ';
            }
        } else {
            // push and then reverse to avoid O(n) unshift in the loop
            let segments = [];
            for (let node = element;
                 node.parentElement;
                 node = node.parentElement) {
                let segment = "*|" + node.localName;
                let nth = Array.prototype.indexOf.call(node.parentElement.children, node) + 1;
                segments.push(segment + ":nth-child(" + nth + ")");
            }
            segments.push(":root");
            segments.reverse();

            selector = segments.join(" > ");
        }

        return selector;
    };

    window.test_driver_internal.in_automation = true;

    window.test_driver_internal.click = function(element) {
        const selector = get_selector(element);
        const pending_promise = new Promise(function(resolve, reject) {
            pending_resolve = resolve;
            pending_reject = reject;
        });
        window.__wptrunner_message_queue.push({"type": "action", "action": "click", "selector": selector});
        return pending_promise;
    };

    window.test_driver_internal.send_keys = function(element, keys) {
        const selector = get_selector(element);
        const pending_promise = new Promise(function(resolve, reject) {
            pending_resolve = resolve;
            pending_reject = reject;
        });
        window.__wptrunner_message_queue.push({"type": "action", "action": "send_keys", "selector": selector, "keys": keys});
        return pending_promise;
    };

    window.test_driver_internal.action_sequence = function(actions) {
        const pending_promise = new Promise(function(resolve, reject) {
            pending_resolve = resolve;
            pending_reject = reject;
        });
        for (let actionSequence of actions) {
            if (actionSequence.type == "pointer") {
                for (let action of actionSequence.actions) {
                    // The origin of each action can only be an element or a string of a value "viewport" or "pointer".
                    if (action.type == "pointerMove" && typeof(action.origin) != 'string') {
                        let frame = get_frame(action.origin, window);
                        if (frame != null) {
                            if (frame == window)
                                action.frame = {frame: "window"};
                            else
                                action.frame = {frame: frame};
                            action.origin = {selector: get_selector(action.origin)};
                        } 
                    }
                }
            }
        }
        window.__wptrunner_message_queue.push({"type": "action", "action": "action_sequence", "actions": actions});
        return pending_promise;
    };

    window.test_driver_internal.generate_test_report = function(message) {
        const pending_promise = new Promise(function(resolve, reject) {
            pending_resolve = resolve;
            pending_reject = reject;
        });
        window.__wptrunner_message_queue.push({"type": "action", "action": "generate_test_report", "message": message});
        return pending_promise;
    };
})();