OSGKeyboard/OSGKeyboardTests/SettingsCloudSyncTests.swift

// SettingsCloudSyncTests.swift
// OSGKeyboardTests
//
// Hermetic tests for iCloud KVS settings sync + preference toggles.

import XCTest
@testable import OSGKeyboardShared

@MainActor
final class SettingsCloudSyncTests: XCTestCase {

    private var suiteName: String!
    private var defaults: UserDefaults!
    private var store: AppGroupStore!
    private var kvs: FakeUbiquitousKeyValueStore!
    private var settingsSync: SettingsCloudSync!
    private let deviceA = "device-a"
    private let deviceB = "device-b"

    override func setUp() {
        super.setUp()
        suiteName = "group.com.osgkeyboard.shared.tests.settings.\(UUID().uuidString)"
        defaults = UserDefaults(suiteName: suiteName)!
        defaults.removePersistentDomain(forName: suiteName)
        defaults.set(deviceA, forKey: "sync.deviceID.v1")
        store = AppGroupStore(defaults: defaults)
        kvs = FakeUbiquitousKeyValueStore()
        settingsSync = SettingsCloudSync(kvs: kvs) { [unowned self] in store }
    }

    override func tearDown() {
        defaults.removePersistentDomain(forName: suiteName)
        try? Keychain.deleteAPIKey(for: "openai", useICloudSync: false)
        try? Keychain.deleteAPIKey(for: "openai", useICloudSync: true)
        try? Keychain.deleteAPIKey(for: "qwen", useICloudSync: false)
        try? Keychain.deleteAPIKey(for: "qwen", useICloudSync: true)
        super.tearDown()
    }

    func testPerFieldMergeKeepsIndependentChanges() {
        let stampA = Date(timeIntervalSince1970: 100)
        let stampB = Date(timeIntervalSince1970: 200)
        let local = SyncedAppSettingsV2(
            providerId: SyncedField(value: "openai", updatedAt: stampA, deviceID: deviceA),
            baseURL: SyncedField(value: "https://local.example", updatedAt: stampA, deviceID: deviceA),
            model: SyncedField(value: "gpt-local", updatedAt: stampA, deviceID: deviceA),
            asrProviderId: SyncedField(value: "zhipu", updatedAt: stampA, deviceID: deviceA),
            asrBaseURL: SyncedField(value: "https://asr-local.example", updatedAt: stampA, deviceID: deviceA),
            asrModel: SyncedField(value: "glm-asr-local", updatedAt: stampA, deviceID: deviceA),
            modeId: SyncedField(value: "polish", updatedAt: stampA, deviceID: deviceA),
            localeId: SyncedField(value: "auto", updatedAt: stampA, deviceID: deviceA),
            engineMode: SyncedField(value: "cloud", updatedAt: stampA, deviceID: deviceA),
            hasAcknowledgedCloudSharing: SyncedField(value: false, updatedAt: stampA, deviceID: deviceA),
            uiLanguage: SyncedField(value: .english, updatedAt: stampA, deviceID: deviceA),
            translationTargetLocaleId: SyncedField(
                value: TranslationLanguageCatalog.offLocaleId,
                updatedAt: stampA,
                deviceID: deviceA
            ),
            handednessPreference: SyncedField(value: .left, updatedAt: stampA, deviceID: deviceA),
            cursorDragNavigationEnabled: SyncedField(value: true, updatedAt: stampA, deviceID: deviceA),
            polishIntensity: SyncedField(value: .medium, updatedAt: stampA, deviceID: deviceA),
            llmThinkingEnabled: SyncedField(value: false, updatedAt: stampA, deviceID: deviceA),
            flowSkipAppSwitch: SyncedField(value: true, updatedAt: stampA, deviceID: deviceA),
            flowInactivityDuration: SyncedField(value: .twelveHours, updatedAt: stampA, deviceID: deviceA)
        )
        let remote = SyncedAppSettingsV2(
            providerId: SyncedField(value: "openai", updatedAt: stampA, deviceID: deviceB),
            baseURL: SyncedField(value: "https://remote.example", updatedAt: stampB, deviceID: deviceB),
            model: SyncedField(value: "gpt-remote", updatedAt: stampB, deviceID: deviceB),
            asrProviderId: SyncedField(value: "qwen", updatedAt: stampB, deviceID: deviceB),
            asrBaseURL: SyncedField(value: "https://asr-remote.example", updatedAt: stampB, deviceID: deviceB),
            asrModel: SyncedField(value: "fun-asr-remote", updatedAt: stampB, deviceID: deviceB),
            modeId: SyncedField(value: "polish", updatedAt: stampA, deviceID: deviceB),
            localeId: SyncedField(value: "ja", updatedAt: stampB, deviceID: deviceB),
            engineMode: SyncedField(value: "local", updatedAt: stampB, deviceID: deviceB),
            hasAcknowledgedCloudSharing: SyncedField(value: true, updatedAt: stampB, deviceID: deviceB),
            uiLanguage: SyncedField(value: .chinese, updatedAt: stampB, deviceID: deviceB),
            translationTargetLocaleId: SyncedField(value: "en", updatedAt: stampB, deviceID: deviceB),
            handednessPreference: SyncedField(value: .right, updatedAt: stampB, deviceID: deviceB),
            cursorDragNavigationEnabled: SyncedField(value: false, updatedAt: stampB, deviceID: deviceB),
            polishIntensity: SyncedField(value: .light, updatedAt: stampB, deviceID: deviceB),
            llmThinkingEnabled: SyncedField(value: true, updatedAt: stampB, deviceID: deviceB),
            flowSkipAppSwitch: SyncedField(value: false, updatedAt: stampB, deviceID: deviceB),
            flowInactivityDuration: SyncedField(value: .threeHours, updatedAt: stampB, deviceID: deviceB)
        )

        let merged = SyncedAppSettingsV2.merge(local: local, remote: remote)

        XCTAssertEqual(merged.baseURL.value, "https://remote.example")
        XCTAssertEqual(merged.asrProviderId.value, "qwen")
        XCTAssertEqual(merged.localeId.value, "ja")
        XCTAssertEqual(merged.engineMode.value, "local")
    }

    func testLegacyV1PullDoesNotClearKeychain() async throws {
        try Keychain.setAPIKey("sk-local-openai", for: "openai", useICloudSync: false)
        store.setSettingsICloudSyncEnabled(true)

        let legacy = SyncedAppSettings(
            updatedAt: Date().addingTimeInterval(3600),
            providerId: "openai",
            baseURL: "https://remote.example",
            model: "remote-model",
            modeId: "polish",
            localeId: "ja",
            engineMode: "cloud",
            hasAcknowledgedCloudSharing: false,
            uiLanguage: .english,
            translationTargetLocaleId: TranslationLanguageCatalog.offLocaleId,
            handednessPreference: .left,
            cursorDragNavigationEnabled: true,
            polishIntensity: .medium,
            flowSkipAppSwitch: true,
            flowInactivityDuration: .twelveHours
        )
        let encoder = JSONEncoder()
        encoder.dateEncodingStrategy = .iso8601
        let data = try encoder.encode(legacy)
        kvs.set(data, forKey: SettingsCloudSync.legacyKVSKey)

        await settingsSync.pullAndMerge(store: store)

        XCTAssertEqual(Keychain.apiKey(for: "openai", preferICloudSync: false), "sk-local-openai")
        XCTAssertEqual(store.localeId, "ja")
    }

    func testEnableSyncMigratesKeysToICloudKeychain() async throws {
        try Keychain.setAPIKey("sk-local-openai", for: "openai", useICloudSync: false)

        try await settingsSync.enableSync()

        XCTAssertTrue(store.settingsICloudSyncEnabled)
        XCTAssertTrue(store.personalDictionaryICloudSyncEnabled)
        XCTAssertEqual(Keychain.apiKey(for: "openai", preferICloudSync: true), "sk-local-openai")
    }

    func testPullAndMergeAppliesRemoteSettingsToAppGroup() async throws {
        store.setSettingsICloudSyncEnabled(true)
        store.setLocaleId("auto")

        let deviceID = SyncDeviceID.current(defaults: defaults)
        let stamp = Date(timeIntervalSince1970: 900)
        var remote = SyncedAppSettingsV2.seeded(
            from: AppGroupConfiguration.load(fromAvailable: defaults),
            deviceID: deviceB,
            updatedAt: stamp
        )
        remote.localeId = SyncedField(value: "ja", updatedAt: stamp, deviceID: deviceB)
        try settingsSync.push(remote)

        await settingsSync.pullAndMerge(store: store)

        XCTAssertEqual(store.localeId, "ja")
        XCTAssertEqual(store.settingsCloudUpdatedAt?.timeIntervalSince1970 ?? 0, 900, accuracy: 1)
    }

    func testPushLocalIfEnabledSkipsWhenDisabled() async throws {
        store.setSettingsICloudSyncEnabled(false)
        store.setLocaleId("ko")

        try await settingsSync.pushLocalIfEnabled()

        XCTAssertNil(settingsSync.loadRemote())
    }

    func testICloudSyncPreferencesMigrateLegacyToggles() {
        store.setSettingsICloudSyncEnabled(false)
        store.setPersonalDictionaryICloudSyncEnabled(true)

        ICloudSyncPreferences.migrateLegacyTogglesIfNeeded(kvs: kvs, store: store)

        XCTAssertEqual(kvs.object(forKey: ICloudSyncPreferences.settingsEnabledKey) as? Bool, false)
        XCTAssertEqual(kvs.object(forKey: ICloudSyncPreferences.dictionaryEnabledKey) as? Bool, true)
    }

    func testAppGroupConfigurationDefaultsSettingsICloudSyncToOn() {
        let config = AppGroupConfiguration.load(fromAvailable: defaults)
        XCTAssertTrue(config.settingsICloudSyncEnabled)
    }
}