diff options
author | Wojtek Kosior <koszko@koszko.org> | 2022-01-17 11:20:52 +0100 |
---|---|---|
committer | Wojtek Kosior <koszko@koszko.org> | 2022-01-17 11:24:56 +0100 |
commit | 7bedbcbd80eba9359d2e905b7693923c76ce563d (patch) | |
tree | 5059ac406e29b1b1e81639fc11316dde280fe218 /test/unit/utils.py | |
parent | ede3a55ba22d2560ec7c0deebffd73623488acc1 (diff) | |
download | browser-extension-7bedbcbd80eba9359d2e905b7693923c76ce563d.tar.gz browser-extension-7bedbcbd80eba9359d2e905b7693923c76ce563d.zip |
move policy enforcing code to a new file, include basic test
Diffstat (limited to 'test/unit/utils.py')
-rw-r--r-- | test/unit/utils.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/test/unit/utils.py b/test/unit/utils.py index 96ebf60..8e04d91 100644 --- a/test/unit/utils.py +++ b/test/unit/utils.py @@ -187,3 +187,16 @@ def is_prime(n): return n > 1 and all([n % i != 0 for i in range(2, n)]) broker_js = lambda: load_script('background/broadcast_broker.js') + ';start();' + +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) |