move policy enforcing code to a new file, include basic test

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)