PoC for TDS Perf Tests

Author: bwaresiak

.swiftpm/xcode/xcshareddata/xcschemes/TrackerRadarKit.xcscheme

@@ -53,6 +53,16 @@
                ReferencedContainer = "container:">
             </BuildableReference>
          </TestableReference>
+         <TestableReference
+            skipped = "NO">
+            <BuildableReference
+               BuildableIdentifier = "primary"
+               BlueprintIdentifier = "TrackerRadarKitPerformanceTests"
+               BuildableName = "TrackerRadarKitPerformanceTests"
+               BlueprintName = "TrackerRadarKitPerformanceTests"
+               ReferencedContainer = "container:">
+            </BuildableReference>
+         </TestableReference>
       </Testables>
    </TestAction>
    <LaunchAction

Package.swift

@@ -37,6 +37,9 @@ let package = Package(
             resources: [
                 .process("Resources/trackerData.json"),
                 .process("Resources/mockTrackerData.json")
-            ])
+            ]),
+        .testTarget(
+            name: "TrackerRadarKitPerformanceTests",
+            dependencies: ["TrackerRadarKit"])
     ]
 )

Tests/TrackerRadarKitPerformanceTests/NextTrackerDataSetPerformanceTests.swift

@@ -0,0 +1,80 @@
+//
+//  NextTrackerDataSetPerformanceTests.swift
+//
+//  Copyright © 2021 DuckDuckGo. All rights reserved.
+//
+//  Licensed under the Apache License, Version 2.0 (the "License");
+//  you may not use this file except in compliance with the License.
+//  You may obtain a copy of the License at
+//
+//  http://www.apache.org/licenses/LICENSE-2.0
+//
+//  Unless required by applicable law or agreed to in writing, software
+//  distributed under the License is distributed on an "AS IS" BASIS,
+//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+//  See the License for the specific language governing permissions and
+//  limitations under the License.
+//
+
+
+import Foundation
+import XCTest
+import TrackerRadarKit
+import WebKit
+
+class NextTrackerDataSetPerformanceTests: XCTestCase {
+
+    static func nextURL(filename: String) -> URL {
+        return URL(string: "https://staticcdn.duckduckgo.com/trackerblocking/v5/next/\(filename)")!
+    }
+
+    var nextiOSTDS: TrackerData!
+
+
+    override func setUp() async throws {
+        try await super.setUp()
+
+        let (data, _) = try await URLSession.shared.data(from: Self.nextURL(filename: "ios-tds.json"))
+
+        nextiOSTDS = try JSONDecoder().decode(TrackerData.self, from: data)
+
+        print("Next TDS prepared")
+    }
+
+    func testPerformanceOfNext_iOSTDS() throws {
+
+        let rules = ContentBlockerRulesBuilder(trackerData: nextiOSTDS).buildRules()
+        
+        let data = try JSONEncoder().encode(rules)
+        let ruleList = String(data: data, encoding: .utf8)!
+
+        var store: WKContentRuleListStore!
+
+        let warmUpExpectation = expectation(description: "Warmed up")
+
+        DispatchQueue.main.async {
+            store = WKContentRuleListStore(url: FileManager.default.temporaryDirectory)
+            store.compileContentRuleList(forIdentifier: UUID().uuidString, encodedContentRuleList: ruleList) { _, error in
+                XCTAssertNil(error)
+                warmUpExpectation.fulfill()
+            }
+        }
+        wait(for: [warmUpExpectation], timeout: 40)
+
+        measure {
+            let time = CACurrentMediaTime()
+            let expectation = expectation(description: "Compiled")
+
+            store.compileContentRuleList(forIdentifier: UUID().uuidString, encodedContentRuleList: ruleList) { _, error in
+                XCTAssertNil(error)
+
+                Thread.sleep(forTimeInterval: 1)
+
+                expectation.fulfill()
+            }
+
+            wait(for: [expectation], timeout: 40)
+            print("Compiled in \(CACurrentMediaTime() - time)")
+        }
+    }
+}