Fix #3700: Fix tests

brave · Mar 15, 2022 · 7118f6e · 7118f6e
1 parent 661af7e
commit 7118f6e
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 6 deletions.
diff --git a/ClientTests/DomainUserScriptTests.swift b/ClientTests/DomainUserScriptTests.swift
@@ -15,17 +15,17 @@ class DomainUserScriptTests: XCTestCase {
                         URL(string: "https://search-dev.brave.com/custom/path")].compactMap { $0 }
 
         goodURLs.forEach {
-            XCTAssert(DomainUserScript.get(for: $0) == .braveSearch)
+            XCTAssertEqual(DomainUserScript(for: $0), .braveSearchHelper)
         }
 
         let badURLs = [URL(string: "https://talk.brave.com"),
-                        URL(string: "https://search.brave.software.com"),
+                       URL(string: "https://search.brave.software.com"),
                        URL(string: "https://community.brave.com"),
                        URL(string: "https://subdomain.search.brave.com"),
                        URL(string: "https://brave.com")].compactMap { $0 }
 
         badURLs.forEach {
-            XCTAssertFalse(DomainUserScript.get(for: $0) == .braveSearch)
+            XCTAssertNotEqual(DomainUserScript(for: $0), .braveSearchHelper)
         }
     }
 
@@ -40,7 +40,7 @@ class DomainUserScriptTests: XCTestCase {
                         URL(string: "https://talk.brave.com/account")].compactMap { $0 }
 
         goodURLs.forEach {
-            XCTAssert(DomainUserScript.get(for: $0) == .braveTalk)
+            XCTAssertEqual(DomainUserScript(for: $0), .braveTalkHelper)
         }
 
         let badURLs = [URL(string: "https://search.brave.com"),
@@ -52,7 +52,24 @@ class DomainUserScriptTests: XCTestCase {
                        URL(string: "https://brave.com")].compactMap { $0 }
 
         badURLs.forEach {
-            XCTAssertFalse(DomainUserScript.get(for: $0) == .braveTalk)
+            XCTAssertNotEqual(DomainUserScript(for: $0), .braveTalkHelper)
+        }
+    }
+
+    func testYouTubeAdBlockAvailability() throws {
+        let goodURLs = [URL(string: "https://youtube.com")!,
+                        URL(string: "https://www.youtube.com")!]
+
+        goodURLs.forEach {
+            XCTAssertEqual(DomainUserScript(for: $0), .youtubeAdBlock)
+        }
+
+        let badURLs = [URL(string: "https://youtube.org.com")!,
+                       URL(string: "https://www.youtube.org.com")!,
+                       URL(string: "https://metube.com")!]
+
+        badURLs.forEach {
+            XCTAssertNotEqual(DomainUserScript(for: $0), .youtubeAdBlock)
         }
     }
 }
diff --git a/ClientTests/TabSessionTests.swift b/ClientTests/TabSessionTests.swift
@@ -600,7 +600,7 @@ class TabSessionTests: XCTestCase {
         let group = DispatchGroup()
         for tab in self.tabManager.allTabs {
             // include all scripts
-            tab.userScriptManager?.isFingerprintingProtectionEnabled = true
+            tab.userScriptManager?.domainScriptTypes = [.farblingProtection(etld: "fake.com")]
             tab.userScriptManager?.isCookieBlockingEnabled = true
 
             group.enter()