Updated tests

duckduckgo · Jan 14, 2025 · cab45a8 · cab45a8
1 parent 62a81c1
commit cab45a8
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/Tests/PrivacyDashboardTests/PrivacyDashboardControllerTests.swift b/Tests/PrivacyDashboardTests/PrivacyDashboardControllerTests.swift
@@ -37,7 +37,6 @@ final class PrivacyDashboardControllerTests: XCTestCase {
         toggleReportingManagerMock = ToggleReportingManagerMock()
         privacyDashboardController = PrivacyDashboardController(privacyInfo: privacyInfo,
                                                                 entryPoint: entryPoint,
-                                                                variant: .control,
                                                                 toggleReportingManager: toggleReportingManagerMock,
                                                                 eventMapping: EventMapping<PrivacyDashboardEvents> { _, _, _, _ in })
         webView = WKWebView()
@@ -51,16 +50,12 @@ final class PrivacyDashboardControllerTests: XCTestCase {
         let entryPoints: [PrivacyDashboardEntryPoint] = [
             .dashboard,
             .report,
-            .afterTogglePrompt(category: "apple", didToggleProtectionsFixIssue: false),
             .toggleReport(completionHandler: { _ in })
         ]
         for entryPoint in entryPoints {
             makePrivacyDashboardController(entryPoint: entryPoint)
             let currentURL = privacyDashboardController.webView!.url
-            XCTAssertEqual(currentURL?.getParameter(named: "screen"), entryPoint.screen(for: .control).rawValue)
-            if case .afterTogglePrompt = entryPoint {
-                XCTAssertEqual(currentURL?.getParameter(named: "category"), "apple")
-            }
+            XCTAssertEqual(currentURL?.getParameter(named: "screen"), "breakageForm")
             if case .toggleReport = entryPoint {
                 XCTAssertEqual(currentURL?.getParameter(named: "opener"), "menu")
             }