wehub-resource-sync
142 lines
4.0 KiB
Swift
142 lines
4.0 KiB
Swift
import AppKit
|
|
import XCTest
|
|
@testable import MacTools
|
|
@testable import HideNotchPlugin
|
|
|
|
@MainActor
|
|
final class MockHideNotchWallpaperController: HideNotchWallpaperControlling {
|
|
var onStateChange: (() -> Void)?
|
|
var snapshotValue = HideNotchSnapshot(
|
|
hasSupportedDisplay: false,
|
|
supportedDisplayCount: 0,
|
|
managedDisplayCount: 0,
|
|
unsupportedVisibleDisplayCount: 0,
|
|
pendingRestoreCount: 0,
|
|
isEnabled: false,
|
|
isProcessing: false,
|
|
isAwaitingDisplay: false,
|
|
errorMessage: nil
|
|
)
|
|
private(set) var refreshCallCount = 0
|
|
private(set) var setEnabledCalls: [Bool] = []
|
|
|
|
func snapshot() -> HideNotchSnapshot {
|
|
snapshotValue
|
|
}
|
|
|
|
func refresh() {
|
|
refreshCallCount += 1
|
|
}
|
|
|
|
func setEnabled(_ isEnabled: Bool) {
|
|
setEnabledCalls.append(isEnabled)
|
|
snapshotValue = HideNotchSnapshot(
|
|
hasSupportedDisplay: snapshotValue.hasSupportedDisplay,
|
|
supportedDisplayCount: snapshotValue.supportedDisplayCount,
|
|
managedDisplayCount: isEnabled ? max(snapshotValue.managedDisplayCount, 1) : 0,
|
|
unsupportedVisibleDisplayCount: 0,
|
|
pendingRestoreCount: 0,
|
|
isEnabled: isEnabled,
|
|
isProcessing: false,
|
|
isAwaitingDisplay: isEnabled && !snapshotValue.hasSupportedDisplay,
|
|
errorMessage: snapshotValue.errorMessage
|
|
)
|
|
}
|
|
}
|
|
|
|
@MainActor
|
|
final class InMemoryHideNotchStateStore: HideNotchStateStoring {
|
|
var desiredEnabled = false
|
|
}
|
|
|
|
struct StubHideNotchDisplayCatalog: HideNotchDisplayCatalogProviding {
|
|
let records: [HideNotchDisplayRecord]
|
|
|
|
func listDisplayRecords() -> [HideNotchDisplayRecord] {
|
|
records
|
|
}
|
|
}
|
|
|
|
@MainActor
|
|
final class RecordingHideNotchDesktopMaskWindow: HideNotchDesktopMaskWindowing {
|
|
private(set) var frames: [CGRect]
|
|
private(set) var showCallCount = 0
|
|
private(set) var closeCallCount = 0
|
|
|
|
init(frame: CGRect) {
|
|
frames = [frame]
|
|
}
|
|
|
|
func setFrame(_ frame: CGRect) {
|
|
frames.append(frame)
|
|
}
|
|
|
|
func show() {
|
|
showCallCount += 1
|
|
}
|
|
|
|
func close() {
|
|
closeCallCount += 1
|
|
}
|
|
}
|
|
|
|
enum StubHideNotchDesktopMaskWindowBuilderError: LocalizedError {
|
|
case forcedFailure
|
|
|
|
var errorDescription: String? {
|
|
"forced failure"
|
|
}
|
|
}
|
|
|
|
@MainActor
|
|
final class RecordingHideNotchDesktopMaskWindowBuilder: HideNotchDesktopMaskWindowBuilding {
|
|
var makeWindowError: Error?
|
|
|
|
private(set) var windowsByOriginX: [CGFloat: RecordingHideNotchDesktopMaskWindow] = [:]
|
|
private(set) var createdFrames: [CGRect] = []
|
|
|
|
func makeWindow(frame: CGRect) throws -> HideNotchDesktopMaskWindowing {
|
|
if let makeWindowError {
|
|
throw makeWindowError
|
|
}
|
|
|
|
let window = RecordingHideNotchDesktopMaskWindow(frame: frame)
|
|
createdFrames.append(frame)
|
|
windowsByOriginX[frame.minX] = window
|
|
return window
|
|
}
|
|
}
|
|
|
|
func makeHideNotchDisplayRecord(
|
|
id displayID: CGDirectDisplayID,
|
|
displayIdentifier: String,
|
|
spaces: [HideNotchDisplaySpace] = [HideNotchDisplaySpace(identifier: "space-1", isCurrent: true)],
|
|
name: String = "Built-in Retina Display",
|
|
frame: CGRect = CGRect(x: 0, y: 0, width: 1512, height: 982),
|
|
backingScaleFactor: CGFloat = 2,
|
|
notchHeightPoints: CGFloat = 36,
|
|
isBuiltin: Bool = true,
|
|
isSupported: Bool = true
|
|
) -> HideNotchDisplayRecord {
|
|
HideNotchDisplayRecord(
|
|
context: HideNotchDisplayContext(
|
|
displayID: displayID,
|
|
displayIdentifier: displayIdentifier,
|
|
name: name,
|
|
frame: frame,
|
|
backingScaleFactor: backingScaleFactor,
|
|
notchHeightPoints: notchHeightPoints,
|
|
isBuiltin: isBuiltin,
|
|
isSupported: isSupported
|
|
),
|
|
spaces: spaces
|
|
)
|
|
}
|
|
|
|
func makeIsolatedUserDefaults() -> UserDefaults {
|
|
let suiteName = "HideNotchTests.\(UUID().uuidString)"
|
|
let defaults = UserDefaults(suiteName: suiteName)!
|
|
defaults.removePersistentDomain(forName: suiteName)
|
|
return defaults
|
|
}
|