update reference tests for protectionsState (#2323)

Co-authored-by: Shane Osbourne <sosbourne@duckduckgo.com>
duckduckgo · Nov 10, 2023 · cf7b24b · cf7b24b
1 parent 0529970
commit cf7b24b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/unit-test/background/reference-tests/broken-site-reporting-tests.js b/unit-test/background/reference-tests/broken-site-reporting-tests.js
@@ -20,6 +20,13 @@ async function submitAndValidateReport (report) {
     })
     tab.upgradedHttps = report.wasUpgraded
 
+    if (report.protectionsEnabled === true) {
+        spyOnProperty(tab.site, 'enabledFeatures').and.returnValue(['contentBlocking'])
+    } else if (report.denylisted === true) {
+        spyOnProperty(tab.site, 'enabledFeatures').and.returnValue([])
+        spyOnProperty(tab.site, 'denylisted').and.returnValue(true)
+    }
+
     const addRequest = (hostname, action, opts = {}) => {
         tab.addToTrackers({
             action,