From 7bedbcbd80eba9359d2e905b7693923c76ce563d Mon Sep 17 00:00:00 2001 From: Wojtek Kosior Date: Mon, 17 Jan 2022 11:20:52 +0100 Subject: move policy enforcing code to a new file, include basic test --- test/unit/test_webrequest.py | 14 +------------- 1 file changed, 1 insertion(+), 13 deletions(-) (limited to 'test/unit/test_webrequest.py') diff --git a/test/unit/test_webrequest.py b/test/unit/test_webrequest.py index ae617aa..598f43b 100644 --- a/test/unit/test_webrequest.py +++ b/test/unit/test_webrequest.py @@ -22,6 +22,7 @@ from hashlib import sha256 import pytest from ..script_loader import load_script +from .utils import are_scripts_allowed def webrequest_js(): return (load_script('background/webrequest.js', @@ -50,19 +51,6 @@ def webrequest_js(): start("somesecret"); ''') -def are_scripts_allowed(driver, nonce=None): - return driver.execute_script( - ''' - document.scripts_allowed = false; - const script = document.createElement("script"); - script.innerHTML = "document.scripts_allowed = true;"; - if (arguments[0]) - script.setAttribute("nonce", arguments[0]); - document.head.append(script); - return document.scripts_allowed; - ''', - nonce) - @pytest.mark.ext_data({'background_script': webrequest_js}) @pytest.mark.usefixtures('webextension') def test_on_headers_received(driver, execute_in_page): -- cgit v1.2.3