/
TPStatsBlocklistsTests.swift
68 lines (56 loc) · 2.82 KB
/
TPStatsBlocklistsTests.swift
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
@testable import Client
import XCTest
class TPStatsBlocklistsTests: XCTestCase {
var blocklists: TPStatsBlocklists!
override func setUp() {
super.setUp()
blocklists = TPStatsBlocklists()
}
override func tearDown() {
super.tearDown()
blocklists = nil
}
func testLoadPerformance() {
self.measureMetrics([.wallClockTime], automaticallyStartMeasuring: true) {
blocklists.load()
self.stopMeasuring()
}
}
func testURLInListPerformance() {
blocklists.load()
let whitelistedRegexs = ["*google.com"].compactMap { (domain) -> String? in
return wildcardContentBlockerDomainToRegex(domain: domain)
}
self.measureMetrics([.wallClockTime], automaticallyStartMeasuring: true) {
for _ in 0..<100 {
_ = blocklists.urlIsInList(URL(string: "https://www.firefox.com")!, mainDocumentURL: URL(string: "http://foo.com")!, whitelistedDomains: whitelistedRegexs)
}
self.stopMeasuring()
}
}
func testURLInList() {
blocklists.load()
func blocklist(_ urlString: String, _ mainDoc: String = "https://foo.com", _ whitelistedDomains: [String] = []) -> BlocklistCategory? {
let whitelistedRegexs = whitelistedDomains.compactMap { (domain) -> String? in
return wildcardContentBlockerDomainToRegex(domain: domain)
}
let mainDoc = URL(string: mainDoc)!
return blocklists.urlIsInList(URL(string: urlString)!, mainDocumentURL: mainDoc, whitelistedDomains: whitelistedRegexs)
}
XCTAssertEqual(blocklist("https://www.facebook.com", "https://atlassolutions.com"), nil)
XCTAssertEqual(blocklist("https://www.firefox.com"), nil)
XCTAssertEqual(blocklist("https://2leep.com/track"), .advertising)
XCTAssertEqual(blocklist("https://sub.2leep.com/ad"), .advertising)
XCTAssertEqual(blocklist("https://admeld.com"), .advertising)
XCTAssertEqual(blocklist("https://admeld.com/popup"), .advertising)
XCTAssertEqual(blocklist("https://sub.admeld.com"), .advertising)
XCTAssertEqual(blocklist("https://subadmeld.com"), nil)
XCTAssertEqual(blocklist("https://aol.com.aolanswers.com", "https://foo.com", ["ers.com"]), nil)
XCTAssertEqual(blocklist("https://sub.xiti.com/track"), .analytics)
XCTAssertEqual(blocklist("https://atlassolutions.com"), .advertising)
XCTAssertEqual(blocklist("https://atlassolutions.com", "https://foo.com", ["*solutions.com"]), nil)
}
}