felt_tests_thunderbird.py

# This Source Code Form is subject to the terms of the Mozilla Public

# License, v. 2.0. If a copy of the MPL was not distributed with this

# file, You can obtain one at https://mozilla.org/MPL/2.0/.

import os

import sys

here = os.path.abspath(__file__)

repo_root = os.path.normpath(os.path.join(here, *[".."] * 6))

sys.path.append(os.path.join(repo_root, "testing", "enterprise"))

from felt_tests import FeltTests

class FeltTestsThunderbird(FeltTests):

    def open_tab_child(self, url):

        with self._child_driver.using_context(self._child_driver.CONTEXT_CHROME):

            self._child_driver.execute_script("""

                const url = arguments[0];

                const mainWin = Services.wm.getMostRecentWindow("mail:3pane");

                const tabmail = mainWin.document.getElementById("tabmail");

                const oldShouldSwitchTo = tabmail.tabModes["contentTab"].shouldSwitchTo;

                tabmail.tabModes["contentTab"].shouldSwitchTo = () => {

                    return -1;

};

                tabmail.openTab("contentTab", { url });

                tabmail.tabModes["contentTab"].shouldSwitchTo = oldShouldSwitchTo;

            """, [ url ])

    def close_tab_child(self):

        with self._child_driver.using_context(self._child_driver.CONTEXT_CHROME):

            self._child_driver.execute_script("""

                const mainWin = Services.wm.getMostRecentWindow("mail:3pane");

                const tabmail = mainWin.document.getElementById("tabmail");

                tabmail.closeTab();

            """)

    def get_tab(self, url):

        with self._child_driver.using_context(self._child_driver.CONTEXT_CHROME):

            return self._child_driver.execute_script("""

                const url = arguments[0];

                const mainWin = Services.wm.getMostRecentWindow("mail:3pane");

                return [...mainWin.document.querySelectorAll("browser")].filter(b => b.docShell?.document?.location == url)[0];

            """, [ url ])

    def get_current_tab(self):

        with self._child_driver.using_context(self._child_driver.CONTEXT_CHROME):

            return self._child_driver.execute_script("""

                const mainWin = Services.wm.getMostRecentWindow("mail:3pane");

                const tabmail = mainWin.document.getElementById("tabmail");

                return tabmail._getTabContextForTabbyThing(null, true)[1].browser;

            """)

    def get_current_uri(self):

        with self._child_driver.using_context(self._child_driver.CONTEXT_CHROME):

            tab = self.get_current_tab()

            return self._child_driver.execute_script("""

                return arguments[0].docShell?.currentDocumentChannel?.originalURI?.spec;

            """, [tab])

    def get_elem_child(self, selector):

        with self._child_driver.using_context(self._child_driver.CONTEXT_CHROME):

            tab = self.get_current_tab()

            return self._child_driver.execute_script("""

                return arguments[0].docShell.document.querySelector(arguments[1]);

            """, [ tab, selector ])

class FeltTestsThunderbirdUi(FeltTestsThunderbird):

    def get_elem_child(self, selector):

        with self._child_driver.using_context(self._child_driver.CONTEXT_CHROME):

            return self._child_driver.execute_script("""

                return document.querySelector(arguments[0]);

            """, [ selector ])