From cab45a8f02cd2aa6c727b5773e6de90f4b78bb25 Mon Sep 17 00:00:00 2001
From: Marcos Gurgel <mgurgel@users.noreply.github.com>
Date: Tue, 14 Jan 2025 14:21:51 +0000
Subject: [PATCH] Updated tests

---
 .../PrivacyDashboardControllerTests.swift                  | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/Tests/PrivacyDashboardTests/PrivacyDashboardControllerTests.swift b/Tests/PrivacyDashboardTests/PrivacyDashboardControllerTests.swift
index a3ee836fc..39ddc62d7 100644
--- 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")
             }