Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
# 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
import os
import shutil
import tempfile
from marionette_harness import MarionetteTestCase
AUTOSTART_PBM_PREF = "browser.privatebrowsing.autostart"
DELETE_PRIVATE_PREF = "browser.download.deletePrivate"
class DeletePrivateDownloadsInPermanentPBM(MarionetteTestCase):
"""
In permanent private browsing mode "last-pb-context-exited" is never
notified, so private downloads must be deleted at shutdown when
browser.download.deletePrivate is set.
"""
def setUp(self):
super().setUp()
self.marionette.set_pref(AUTOSTART_PBM_PREF, True)
self.marionette.set_pref(DELETE_PRIVATE_PREF, True)
# Avoid the remote download reputation check.
self.marionette.set_pref("browser.safebrowsing.downloads.remote.enabled", False)
self.marionette.restart(in_app=True)
self.target_dir = tempfile.mkdtemp()
self.target_path = os.path.join(self.target_dir, "download.html")
def tearDown(self):
self.marionette.restart(in_app=False, clean=True)
shutil.rmtree(self.target_dir, ignore_errors=True)
super().tearDown()
def test_private_download_deleted_on_shutdown(self):
with self.marionette.using_context("chrome"):
self.assertTrue(
self.marionette.execute_script(
"""
const { PrivateBrowsingUtils } = ChromeUtils.importESModule(
"resource://gre/modules/PrivateBrowsingUtils.sys.mjs"
);
return PrivateBrowsingUtils.permanentPrivateBrowsing;
"""
),
"Permanent private browsing mode is enabled",
)
error = self.marionette.execute_async_script(
"""
const [url, targetPath, resolve] = arguments;
const { Downloads } = ChromeUtils.importESModule(
"resource://gre/modules/Downloads.sys.mjs"
);
(async () => {
const download = await Downloads.createDownload({
source: { url, isPrivate: true },
target: targetPath,
});
const list = await Downloads.getList(Downloads.PRIVATE);
await list.add(download);
await download.start();
})().then(() => resolve(null), e => resolve(String(e)));
""",
script_args=(
self.marionette.absolute_url("empty.html"),
self.target_path,
),
)
self.assertIsNone(error, "Private download succeeded")
self.assertTrue(
os.path.exists(self.target_path), "Private download was completed"
)
self.marionette.quit(in_app=True)
self.assertFalse(
os.path.exists(self.target_path),
"Private download was deleted on shutdown",
)
self.marionette.start_session()