mozilla-central/testing/firefox-ui/tests/functional/safebrowsing/test_initial_download.py

Source code

Revision control

Copy as Markdown

Other Tools

Test Info: Warnings

# 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/.
import os
from functools import reduce
from marionette_driver import Wait
from marionette_harness import MarionetteTestCase
class TestSafeBrowsingInitialDownload(MarionetteTestCase):
v2_file_extensions = [
"vlpset",
"sbstore",
]
v4_file_extensions = [
"vlpset",
"metadata",
]
prefs_download_lists = [
"urlclassifier.blockedTable",
"urlclassifier.downloadAllowTable",
"urlclassifier.downloadBlockTable",
"urlclassifier.malwareTable",
"urlclassifier.phishTable",
"urlclassifier.trackingTable",
"urlclassifier.trackingWhitelistTable",
]
prefs_provider_update_time = {
# Force an immediate download of the safebrowsing files
"browser.safebrowsing.provider.mozilla.nextupdatetime": 1,
}
prefs_safebrowsing = {
"browser.safebrowsing.debug": True,
"browser.safebrowsing.update.enabled": True,
}
def get_safebrowsing_files(self, is_v4):
files = []
if is_v4:
my_file_extensions = self.v4_file_extensions
else: # v2
my_file_extensions = self.v2_file_extensions
for pref_name in self.prefs_download_lists:
base_names = self.marionette.get_pref(pref_name).split(",")
# moztest- lists are not saved to disk
# pylint --py3k: W1639
base_names = list(
filter(lambda x: not x.startswith("moztest-"), base_names)
)
for ext in my_file_extensions:
files.extend(
[
"{name}.{ext}".format(name=f, ext=ext)
for f in base_names
if f and f.endswith("-proto") == is_v4
]
)
return set(sorted(files))
def setUp(self):
super(TestSafeBrowsingInitialDownload, self).setUp()
self.safebrowsing_v2_files = self.get_safebrowsing_files(False)
if any(
f.startswith("goog-") or f.startswith("googpub-")
for f in self.safebrowsing_v2_files
):
self.prefs_provider_update_time.update(
{
"browser.safebrowsing.provider.google.nextupdatetime": 1,
}
)
self.safebrowsing_v4_files = self.get_safebrowsing_files(True)
if any(
f.startswith("goog-") or f.startswith("googpub-")
for f in self.safebrowsing_v4_files
):
self.prefs_provider_update_time.update(
{
"browser.safebrowsing.provider.google4.nextupdatetime": 1,
}
)
# Force the preferences for the new profile
enforce_prefs = self.prefs_safebrowsing
enforce_prefs.update(self.prefs_provider_update_time)
self.marionette.enforce_gecko_prefs(enforce_prefs)
self.safebrowsing_path = os.path.join(
self.marionette.instance.profile.profile, "safebrowsing"
)
def tearDown(self):
try:
# Restart with a fresh profile
self.marionette.restart(in_app=False, clean=True)
finally:
super(TestSafeBrowsingInitialDownload, self).tearDown()
def test_safe_browsing_initial_download(self):
def check_downloaded(_):
return reduce(
lambda state, pref: state and int(self.marionette.get_pref(pref)) != 1,
list(self.prefs_provider_update_time),
True,
)
try:
Wait(self.marionette, timeout=170).until(
check_downloaded,
message="Not all safebrowsing files have been downloaded",
)
finally:
files_on_disk_toplevel = os.listdir(self.safebrowsing_path)
for f in self.safebrowsing_v2_files:
self.assertIn(f, files_on_disk_toplevel)
if len(self.safebrowsing_v4_files) > 0:
files_on_disk_google4 = os.listdir(
os.path.join(self.safebrowsing_path, "google4")
)
for f in self.safebrowsing_v4_files:
self.assertIn(f, files_on_disk_google4)