Gaze/GazeTests/TestHelpers.swift

//
//  TestHelpers.swift
//  GazeTests
//
//  Test helpers and utilities for unit testing.
//

import Foundation
import XCTest
@testable import Gaze

// MARK: - Enhanced MockSettingsManager

/// Enhanced mock settings manager with full control over state
@MainActor
final class EnhancedMockSettingsManager: ObservableObject, SettingsProviding {
    @Published var settings: AppSettings
    
    var settingsPublisher: Published<AppSettings>.Publisher {
        $settings
    }
    
    private let timerConfigKeyPaths: [TimerType: WritableKeyPath<AppSettings, TimerConfiguration>] = [
        .lookAway: \.lookAwayTimer,
        .blink: \.blinkTimer,
        .posture: \.postureTimer,
    ]
    
    // Track method calls for verification
    private(set) var saveCallCount = 0
    private(set) var saveImmediatelyCallCount = 0
    private(set) var loadCallCount = 0
    private(set) var resetToDefaultsCallCount = 0
    
    init(settings: AppSettings = .defaults) {
        self.settings = settings
    }
    
    func timerConfiguration(for type: TimerType) -> TimerConfiguration {
        guard let keyPath = timerConfigKeyPaths[type] else {
            preconditionFailure("Unknown timer type: \(type)")
        }
        return settings[keyPath: keyPath]
    }
    
    func updateTimerConfiguration(for type: TimerType, configuration: TimerConfiguration) {
        guard let keyPath = timerConfigKeyPaths[type] else {
            preconditionFailure("Unknown timer type: \(type)")
        }
        settings[keyPath: keyPath] = configuration
    }
    
    func allTimerConfigurations() -> [TimerType: TimerConfiguration] {
        var configs: [TimerType: TimerConfiguration] = [:]
        for (type, keyPath) in timerConfigKeyPaths {
            configs[type] = settings[keyPath: keyPath]
        }
        return configs
    }
    
    func save() {
        saveCallCount += 1
    }
    
    func saveImmediately() {
        saveImmediatelyCallCount += 1
    }
    
    func load() {
        loadCallCount += 1
    }
    
    func resetToDefaults() {
        resetToDefaultsCallCount += 1
        settings = .defaults
    }
    
    // Test helpers
    func reset() {
        saveCallCount = 0
        saveImmediatelyCallCount = 0
        loadCallCount = 0
        resetToDefaultsCallCount = 0
        settings = .defaults
    }
}

// MARK: - Mock Smart Mode Services

@MainActor
final class MockFullscreenDetectionService: ObservableObject, FullscreenDetectionProviding {
    @Published var isFullscreenActive: Bool = false
    
    var isFullscreenActivePublisher: Published<Bool>.Publisher {
        $isFullscreenActive
    }
    
    private(set) var forceUpdateCallCount = 0
    
    func forceUpdate() {
        forceUpdateCallCount += 1
    }
    
    func simulateFullscreen(_ active: Bool) {
        isFullscreenActive = active
    }
}

@MainActor
final class MockIdleMonitoringService: ObservableObject, IdleMonitoringProviding {
    @Published var isIdle: Bool = false
    
    var isIdlePublisher: Published<Bool>.Publisher {
        $isIdle
    }
    
    private(set) var thresholdMinutes: Int = 5
    private(set) var forceUpdateCallCount = 0
    
    func updateThreshold(minutes: Int) {
        thresholdMinutes = minutes
    }
    
    func forceUpdate() {
        forceUpdateCallCount += 1
    }
    
    func simulateIdle(_ idle: Bool) {
        isIdle = idle
    }
}

// MARK: - Test Fixtures

extension AppSettings {
    /// Settings with all timers disabled
    static var allTimersDisabled: AppSettings {
        var settings = AppSettings.defaults
        settings.lookAwayTimer.enabled = false
        settings.blinkTimer.enabled = false
        settings.postureTimer.enabled = false
        return settings
    }
    
    /// Settings with only lookAway timer enabled
    static var onlyLookAwayEnabled: AppSettings {
        var settings = AppSettings.defaults
        settings.lookAwayTimer.enabled = true
        settings.blinkTimer.enabled = false
        settings.postureTimer.enabled = false
        return settings
    }
    
    /// Settings with short intervals for testing
    static var shortIntervals: AppSettings {
        var settings = AppSettings.defaults
        settings.lookAwayTimer.intervalSeconds = 5
        settings.blinkTimer.intervalSeconds = 3
        settings.postureTimer.intervalSeconds = 7
        return settings
    }
    
    /// Settings with onboarding completed
    static var onboardingCompleted: AppSettings {
        var settings = AppSettings.defaults
        settings.hasCompletedOnboarding = true
        return settings
    }
    
    /// Settings with smart mode fully enabled
    static var smartModeEnabled: AppSettings {
        var settings = AppSettings.defaults
        settings.smartMode.autoPauseOnFullscreen = true
        settings.smartMode.autoPauseOnIdle = true
        settings.smartMode.idleThresholdMinutes = 5
        return settings
    }
}

// MARK: - Test Utilities

/// Creates a service container configured for testing
@MainActor
func createTestContainer(
    settings: AppSettings = .defaults
) -> ServiceContainer {
    return ServiceContainer.forTesting(settings: settings)
}

/// Creates a complete test environment with all mocks
@MainActor
struct TestEnvironment {
    let container: ServiceContainer
    let windowManager: MockWindowManager
    let settingsManager: EnhancedMockSettingsManager
    let timeProvider: MockTimeProvider
    
    init(settings: AppSettings = .defaults) {
        self.settingsManager = EnhancedMockSettingsManager(settings: settings)
        self.container = ServiceContainer(settingsManager: settingsManager)
        self.windowManager = MockWindowManager()
        self.timeProvider = MockTimeProvider()
    }
    
    /// Creates an AppDelegate with all test dependencies
    func createAppDelegate() -> AppDelegate {
        return AppDelegate(serviceContainer: container, windowManager: windowManager)
    }
    
    /// Resets all mock state
    func reset() {
        windowManager.reset()
        settingsManager.reset()
    }
}

// MARK: - XCTest Extensions

extension XCTestCase {
    /// Waits for a condition to be true with timeout
    @MainActor
    func waitFor(
        _ condition: @escaping () -> Bool,
        timeout: TimeInterval = 1.0,
        message: String = "Condition not met"
    ) async throws {
        let deadline = Date().addingTimeInterval(timeout)
        while !condition() {
            if Date() > deadline {
                XCTFail(message)
                return
            }
            try await Task.sleep(nanoseconds: 10_000_000) // 10ms
        }
    }
    
    /// Waits for a published value to change
    @MainActor
    func waitForPublisher<T: Equatable>(
        _ publisher: Published<T>.Publisher,
        toEqual expectedValue: T,
        timeout: TimeInterval = 1.0
    ) async throws {
        let expectation = XCTestExpectation(description: "Publisher value changed")
        var cancellable: AnyCancellable?
        
        cancellable = publisher.sink { value in
            if value == expectedValue {
                expectation.fulfill()
            }
        }
        
        await fulfillment(of: [expectation], timeout: timeout)
        cancellable?.cancel()
    }
}

// MARK: - Import Statement for Combine
import Combine