better presentation fot the App Store

CableUITestsScreenshot/CableUITestsScreenshot.swift

@@ -8,34 +8,90 @@
 import XCTest
 
 final class CableUITestsScreenshot: XCTestCase {
+    private let springboard = XCUIApplication(bundleIdentifier: "com.apple.springboard")
 
     override func setUpWithError() throws {
-        // Put setup code here. This method is called before the invocation of each test method in the class.
-
-        // In UI tests it is usually best to stop immediately when a failure occurs.
         continueAfterFailure = false
-
-        // In UI tests it’s important to set the initial state - such as interface orientation - required for your tests before they run. The setUp method is a good place to do this.
+        try super.setUpWithError()
+        ensureDoNotDisturbEnabled()
+        dismissSystemOverlays()
     }
 
     override func tearDownWithError() throws {
-        // Put teardown code here. This method is called after the invocation of each test method in the class.
+        try super.tearDownWithError()
+        dismissSystemOverlays()
     }
 
     @MainActor
     func testExample() throws {
-        // UI tests must launch the application that they test.
         let app = XCUIApplication()
         app.launch()
-
-        // Use XCTAssert and related functions to verify your tests produce the correct results.
+        dismissSystemOverlays()
     }
 
-//    @MainActor
-//    func testLaunchPerformance() throws {
-//        // This measures how long it takes to launch your application.
-//        measure(metrics: [XCTApplicationLaunchMetric()]) {
-//            XCUIApplication().launch()
-//        }
-//    }
+    private func ensureDoNotDisturbEnabled() {
+        springboard.activate()
+        let pullStart = springboard.coordinate(withNormalizedOffset: CGVector(dx: 0.98, dy: 0.02))
+        let pullEnd = springboard.coordinate(withNormalizedOffset: CGVector(dx: 0.98, dy: 0.30))
+        pullStart.press(forDuration: 0.1, thenDragTo: pullEnd)
+
+        let focusTile = springboard.otherElements["Focus"]
+        let focusButton = springboard.buttons["Focus"]
+        if focusTile.waitForExistence(timeout: 2) {
+            focusTile.press(forDuration: 1.0)
+        } else if focusButton.waitForExistence(timeout: 2) {
+            focusButton.press(forDuration: 1.0)
+        }
+
+        let dndButton = springboard.buttons["Do Not Disturb"]
+        if dndButton.waitForExistence(timeout: 1) {
+            if !dndButton.isSelected {
+                dndButton.tap()
+            }
+        } else {
+            let dndCell = springboard.cells["Do Not Disturb"]
+            if dndCell.waitForExistence(timeout: 1) && !dndCell.isSelected {
+                dndCell.tap()
+            } else {
+                let dndLabel = springboard.staticTexts["Do Not Disturb"]
+                if dndLabel.waitForExistence(timeout: 1) {
+                    dndLabel.tap()
+                }
+            }
+        }
+
+        let dismissStart = springboard.coordinate(withNormalizedOffset: CGVector(dx: 0.5, dy: 0.85))
+        let dismissEnd = springboard.coordinate(withNormalizedOffset: CGVector(dx: 0.5, dy: 0.15))
+        dismissStart.press(forDuration: 0.1, thenDragTo: dismissEnd)
+    }
+
+    private func dismissSystemOverlays() {
+        let app = XCUIApplication()
+        let alertButtons = [
+            "OK", "Allow", "Later", "Not Now", "Close",
+            "Continue", "Remind Me Later", "Maybe Later",
+        ]
+
+        if app.alerts.firstMatch.exists {
+            handleAlerts(in: app, buttons: alertButtons)
+        }
+
+        if springboard.alerts.firstMatch.exists || springboard.scrollViews.firstMatch.exists {
+            handleAlerts(in: springboard, buttons: alertButtons + ["Enable Later"])
+        }
+    }
+
+    private func handleAlerts(in application: XCUIApplication, buttons: [String]) {
+        for buttonLabel in buttons {
+            let button = application.buttons[buttonLabel]
+            if button.waitForExistence(timeout: 0.5) {
+                button.tap()
+                return
+            }
+        }
+        let closeButton = application.buttons.matching(NSPredicate(format: "identifier CONTAINS[c] %@", "Close")).firstMatch
+        if closeButton.exists {
+            closeButton.tap()
+        }
+    }
 }