testing/firefox-ui/tests/functional/safebrowsing/test_notification.py
author M. Sirringhaus <msirringhaus@suse.de>
Mon, 10 May 2021 13:00:45 +0000
changeset 579234 ecb1490c9fba32eebcbf6fd3b5b80981799ed8ab
parent 575362 f7ccaffc49b86d3f081d8ecf25667f94d0927b0e
permissions -rw-r--r--
Bug 1710099 - Update the minidump_writer_linux crate to replace the unmaintained memmap crate. r=gsvelto Differential Revision: https://phabricator.services.mozilla.com/D114728

# 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 http://mozilla.org/MPL/2.0/.

from __future__ import absolute_import

import time

from marionette_driver import By, expected, Wait
from marionette_harness import MarionetteTestCase, WindowManagerMixin


class TestSafeBrowsingNotificationBar(WindowManagerMixin, MarionetteTestCase):
    def setUp(self):
        super(TestSafeBrowsingNotificationBar, self).setUp()

        self.test_data = [
            # Unwanted software URL
            {"unsafe_page": "https://www.itisatrap.org/firefox/unwanted.html"},
            # Phishing URL info
            {"unsafe_page": "https://www.itisatrap.org/firefox/its-a-trap.html"},
            # Malware URL object
            {"unsafe_page": "https://www.itisatrap.org/firefox/its-an-attack.html"},
        ]

        self.default_homepage = self.marionette.get_pref("browser.startup.homepage")

        self.marionette.set_pref("browser.safebrowsing.phishing.enabled", True)
        self.marionette.set_pref("browser.safebrowsing.malware.enabled", True)

        # Give the browser a little time, because SafeBrowsing.jsm takes a while
        # between start up and adding the example urls to the db.
        # hg.mozilla.org/mozilla-central/file/46aebcd9481e/browser/base/content/browser.js#l1194
        time.sleep(3)

        # Run this test in a new tab.
        new_tab = self.open_tab()
        self.marionette.switch_to_window(new_tab)

    def tearDown(self):
        try:
            self.marionette.clear_pref("browser.safebrowsing.phishing.enabled")
            self.marionette.clear_pref("browser.safebrowsing.malware.enabled")

            self.remove_permission("https://www.itisatrap.org", "safe-browsing")
            self.close_all_tabs()
        finally:
            super(TestSafeBrowsingNotificationBar, self).tearDown()

    def test_notification_bar(self):
        for item in self.test_data:
            unsafe_page = item["unsafe_page"]

            # Return to the unsafe page
            # Check "ignore warning" link then notification bar's "get me out" button
            self.marionette.navigate(unsafe_page)
            # Wait for the DOM to receive events for about:blocked
            time.sleep(1)
            self.check_ignore_warning_link(unsafe_page)
            self.check_get_me_out_of_here_button()

            # Return to the unsafe page
            # Check "ignore warning" link then notification bar's "X" button
            self.marionette.navigate(unsafe_page)
            # Wait for the DOM to receive events for about:blocked
            time.sleep(1)
            self.check_ignore_warning_link(unsafe_page)
            self.check_x_button()

    def get_final_url(self, url):
        self.marionette.navigate(url)
        return self.marionette.get_url()

    def remove_permission(self, host, permission):
        with self.marionette.using_context("chrome"):
            self.marionette.execute_script(
                """
              Components.utils.import("resource://gre/modules/Services.jsm");
              let uri = Services.io.newURI(arguments[0], null, null);
              let principal = Services.scriptSecurityManager.createContentPrincipal(uri, {});
              Services.perms.removeFromPrincipal(principal, arguments[1]);
            """,
                script_args=[host, permission],
            )

    def check_ignore_warning_link(self, unsafe_page):
        button = self.marionette.find_element(By.ID, "seeDetailsButton")
        button.click()
        time.sleep(1)
        link = self.marionette.find_element(By.ID, "ignore_warning_link")
        link.click()

        Wait(self.marionette, timeout=self.marionette.timeout.page_load).until(
            expected.element_present(By.ID, "main-feature"),
            message='Expected target element "#main-feature" has not been found',
        )
        self.assertEquals(self.marionette.get_url(), self.get_final_url(unsafe_page))

        # Clean up here since the permission gets set in this function
        self.remove_permission("https://www.itisatrap.org", "safe-browsing")

    def check_get_me_out_of_here_button(self):
        with self.marionette.using_context("chrome"):
            if self.marionette.get_pref("browser.proton.enabled"):
                notification_box = self.marionette.find_element(
                    By.CSS_SELECTOR, 'vbox.notificationbox-stack[slot="selected"]'
                )
                message = notification_box.find_element(
                    By.CSS_SELECTOR, "notification-message"
                )
                button_container = message.get_property("buttonContainer")
                button = button_container.find_element(
                    By.CSS_SELECTOR, 'button[label="Get me out of here!"]'
                )
            else:
                button = self.marionette.find_element(
                    By.ID, "tabbrowser-tabbox"
                ).find_element(By.CSS_SELECTOR, 'button[label="Get me out of here!"]')
            button.click()

        Wait(self.marionette, timeout=self.marionette.timeout.page_load).until(
            lambda mn: self.default_homepage in mn.get_url(),
            message="The default home page has not been loaded",
        )

    def check_x_button(self):
        with self.marionette.using_context("chrome"):
            if self.marionette.get_pref("browser.proton.enabled"):
                notification_box = self.marionette.find_element(
                    By.CSS_SELECTOR, 'vbox.notificationbox-stack[slot="selected"]'
                )
                message = notification_box.find_element(
                    By.CSS_SELECTOR, "notification-message[value=blocked-badware-page]"
                )
                button = message.get_property("closeButton")
            else:
                button = (
                    self.marionette.find_element(By.ID, "tabbrowser-tabbox")
                    .find_element(
                        By.CSS_SELECTOR, "notification[value=blocked-badware-page]"
                    )
                    .find_element(By.CSS_SELECTOR, ".messageCloseButton")
                )
            button.click()

            Wait(self.marionette, timeout=self.marionette.timeout.page_load).until(
                expected.element_stale(button),
                message="The notification bar has not been closed",
            )