1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
# 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"):
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"):
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",
)
|
