add test

duckduckgo · May 15, 2024 · a09f90e · a09f90e
1 parent 014eec4
commit a09f90e
Showing 1 changed file with 35 additions and 0 deletions.
diff --git a/IntegrationTests/Tab/AddressBarTests.swift b/IntegrationTests/Tab/AddressBarTests.swift
@@ -861,6 +861,41 @@ class AddressBarTests: XCTestCase {
         let shieldImage = mainViewController.navigationBarViewController.addressBarViewController!.addressBarButtonsViewController!.privacyEntryPointButton.image!
         XCTAssertTrue(shieldImage.isEqualToImage(expectedImage))
     }
+
+    @MainActor
+    func test_ZoomLevelNonDefault_ThenZoomButtonIsVisible() async throws {
+        // GIVEN
+        let tab = Tab(content: .url(.duckDuckGo, credential: nil, source: .userEntered("")))
+        let viewModel = TabCollectionViewModel(tabCollection: TabCollection(tabs: [tab]))
+        viewModel.selectedTabViewModel?.zoomWasSet(to: .percent150)
+        let tabLoadedPromise = tab.webViewDidFinishNavigationPublisher.timeout(5).first().promise()
+
+        // WHEN
+        window = WindowsManager.openNewWindow(with: viewModel)!
+        _=try await tabLoadedPromise.value
+
+        // THEN
+        let zoomButton = mainViewController.navigationBarViewController.addressBarViewController!.addressBarButtonsViewController!.zoomButton!
+        XCTAssertFalse(zoomButton.isHidden)
+    }
+
+    @MainActor
+    func test_ZoomLevelDefault_ThenZoomButtonIsNotVisible() async throws {
+        // GIVEN
+        let tab = Tab(content: .url(.duckDuckGo, credential: nil, source: .userEntered("")))
+        tab.webView.zoomLevel = AccessibilityPreferences.shared.defaultPageZoom
+        let viewModel = TabCollectionViewModel(tabCollection: TabCollection(tabs: [tab]))
+        viewModel.selectedTabViewModel?.zoomWasSet(to: .percent100)
+        let tabLoadedPromise = tab.webViewDidFinishNavigationPublisher.timeout(5).first().promise()
+
+        // WHEN
+        window = WindowsManager.openNewWindow(with: viewModel)!
+        _=try await tabLoadedPromise.value
+
+        // THEN
+        let zoomButton = mainViewController.navigationBarViewController.addressBarViewController!.addressBarButtonsViewController!.zoomButton!
+        XCTAssertTrue(zoomButton.isHidden)
+    }
 }
 
 protocol MainActorPerformer {