import Foundation import Combine import VnidropCore @testable import VniDrop enum TestError: Error { case unimplemented } /// In-memory `CoreGateway`, mirroring `support/Fakes.kt`'s `FakeCoreGateway`. /// Lets model tests drive core state/signals and stub results without the FFI. @MainActor final class FakeCoreGateway: CoreGateway { private let stateSubject = CurrentValueSubject(CoreState()) private let signalsSubject = PassthroughSubject() var state: CoreState { stateSubject.value } var statePublisher: AnyPublisher { stateSubject.eraseToAnyPublisher() } var signals: AnyPublisher { signalsSubject.eraseToAnyPublisher() } // Stubbed results var requests: [UInt64: [ReceiverRequestModel]] = [:] var responseResult: Result = .success(()) var shareResult: Result = .failure(TestError.unimplemented) var inspectionResult: Result = .failure(TestError.unimplemented) var receiveResult: Result = .success(()) var cancelResult: Result = .success(()) var deleteResult: Result = .success(()) var clearReceiveHistoryResult: Result = .success(0) var initializeResult: Result = .success(()) var initializeResults: [Result] = [] // Recorded calls private(set) var responses: [(id: String, accepted: Bool, reason: String?)] = [] private(set) var deletedTransfers: [UInt64] = [] private(set) var cancelledTransfers: [UInt64] = [] private(set) var clearReceiveHistoryCount = 0 private(set) var receiveCount = 0 private(set) var lastReceiveTicket: String? private(set) var lastReceiveReceiverName: String? private(set) var lastShareAccessPolicy: ShareAccessPolicy? private(set) var initializedNetworkConfigurations: [RelayConfiguration] = [] func setState(_ state: CoreState) { stateSubject.send(state) } func emit(_ signal: CoreSignal) { signalsSubject.send(signal) } func initialize( appDataDir: String, networkConfiguration: RelayConfiguration ) async -> Result { initializedNetworkConfigurations.append(networkConfiguration) let result = initializeResults.isEmpty ? initializeResult : initializeResults.removeFirst() guard case .success = result else { return result } var s = stateSubject.value s.isInitialized = true stateSubject.send(s) return .success(()) } func shutdown() {} func shareSources(_ sources: [ShareSource], transferName: String, senderName: String, accessPolicy: ShareAccessPolicy) async -> Result { lastShareAccessPolicy = accessPolicy return shareResult } func inspectTicket(_ ticket: String) async -> Result { inspectionResult } func receive(ticket: String, outputDir: String, receiverName: String) async -> Result { receiveCount += 1; lastReceiveTicket = ticket; lastReceiveReceiverName = receiverName return receiveResult } func receiveIntoSecurityScopedDirectory(ticket: String, outputDirectoryUrl: String, receiverName: String) async -> Result { receiveCount += 1; lastReceiveTicket = ticket; lastReceiveReceiverName = receiverName return receiveResult } func cancel(transferId: UInt64) async -> Result { cancelledTransfers.append(transferId); return cancelResult } func delete(transferId: UInt64) async -> Result { deletedTransfers.append(transferId); return deleteResult } func clearReceiveHistory() async -> Result { clearReceiveHistoryCount += 1; return clearReceiveHistoryResult } func storageUsage() async -> Result { .success(CoreStorageUsageModel(blobStoreBytes: 0, appDataBytes: 0)) } func receivedArtifacts() async -> Result<[ReceivedArtifactModel], Error> { .success([]) } func receiverRequests(transferId: UInt64) async -> Result<[ReceiverRequestModel], Error> { .success(requests[transferId] ?? []) } func respondReceiverRequest(requestId: String, accepted: Bool, reason: String?) async -> Result { responses.append((requestId, accepted, reason)) return responseResult } func refresh() async -> Result { .success(()) } // MARK: Device history var contactsResult: Result<[DeviceContact], Error> = .success([]) var pairings: [PendingPairingModel] = [] var offers: [IncomingOfferModel] = [] var respondToPairingResult: Result = .success(true) /// Ticket handed back when an offer is accepted; nil models a declined one. var offerTicket: String? = "vnd1:offered" var sendToContactResult: Result = .failure(TestError.unimplemented) var forgetContactResult: Result = .success(()) var blockedResult: Result<[String], Error> = .success([]) private(set) var allowedDevices: [(endpointId: String, displayName: String?)] = [] private(set) var pairingResponses: [(endpointId: String, accepted: Bool)] = [] private(set) var offerResponses: [(offerId: String, accepted: Bool)] = [] private(set) var forgottenContacts: [String] = [] private(set) var forgetAllCount = 0 private(set) var blockedContactIds: [String] = [] private(set) var unblockedContactIds: [String] = [] private(set) var contactLabels: [(endpointId: String, label: String?)] = [] private(set) var grantLifetimes: [GrantLifetimeOption] = [] private(set) var sentToContacts: [String] = [] func contacts() async -> Result<[DeviceContact], Error> { contactsResult } func pendingPairings() async -> [PendingPairingModel] { pairings } func pendingOffers() async -> [IncomingOfferModel] { offers } func allowDeviceToReachMe(endpointId: String, displayName: String?) async -> Result { allowedDevices.append((endpointId, displayName)) return .success(()) } func respondToPairing(endpointId: String, accepted: Bool) async -> Result { pairingResponses.append((endpointId, accepted)) if case .success = respondToPairingResult { pairings.removeAll { $0.endpointId == endpointId } } return respondToPairingResult } func respondToOffer(offerId: String, accepted: Bool) async -> String? { offerResponses.append((offerId, accepted)) offers.removeAll { $0.offerId == offerId } return accepted ? offerTicket : nil } func sendToContact( endpointId: String, sources: [ShareSource], transferName: String, senderName: String ) async -> Result { sentToContacts.append(endpointId) return sendToContactResult } func forgetContact(endpointId: String) async -> Result { forgottenContacts.append(endpointId) return forgetContactResult } func forgetAllContacts() async -> Result { forgetAllCount += 1 return .success(0) } func blockContact(endpointId: String) async -> Result { blockedContactIds.append(endpointId) return .success(()) } func unblockContact(endpointId: String) async -> Result { unblockedContactIds.append(endpointId) return .success(()) } func blockedContacts() async -> Result<[String], Error> { blockedResult } func setContactLabel(endpointId: String, label: String?) async -> Result { contactLabels.append((endpointId, label)) return .success(()) } func setGrantLifetime(_ lifetime: GrantLifetimeOption) async { grantLifetimes.append(lifetime) } } /// Minimal `FileSystemService` fake — a writable path receive folder, no reveal. @MainActor final class FakeFileSystemService: FileSystemService { var supportsCustomReceiveFolders = false var folder = ReceiveFolder(kind: .fileSystemPath, value: "/tmp/vnidrop-tests", displayName: "Documents") func defaultReceiveFolder() -> ReceiveFolder { folder } func validateReceiveFolder(_ folder: ReceiveFolder) async -> FolderAccessStatus { .writable } func canRevealReceiveFolder(_ folder: ReceiveFolder) -> Bool { false } func sharePickedFiles(repository: CoreGateway, files: [PickedShareFile], transferName: String, senderName: String, accessPolicy: ShareAccessPolicy) async -> Result { await repository.shareSources([], transferName: transferName, senderName: senderName, accessPolicy: accessPolicy) } } @MainActor final class FakeDeviceInfoProvider: DeviceInfoProvider { func load() async -> DeviceInfo { DeviceInfo(deviceName: "Test Device", deviceModel: "TestModel", operatingSystem: "TestOS 1.0", network: nil, batteryLevel: nil) } } // MARK: - Factories @MainActor enum Fixtures { static func preferences(username: String = "Tester") -> AppPreferencesRepository { let defaults = UserDefaults(suiteName: "vnidrop.tests.\(UUID().uuidString)")! return AppPreferencesRepository( defaults: defaults, fallback: AppPreferencesDefaults( username: username, receiveFolder: ReceiveFolder(kind: .fileSystemPath, value: "/tmp/vnidrop-tests", displayName: "Documents"), themeMode: .system ) ) } static func request(id: String, requestedAt: Int64, transferId: UInt64 = 1, status: ReceiverDeliveryStatus = .requested) -> ReceiverRequestModel { ReceiverRequestModel( id: id, transferId: transferId, remoteEndpointId: "endpoint-\(id)", transferName: "Photos", receiverName: "Peer", receiverDeviceName: "Phone", appVersion: "1.0", status: status, reason: nil, requestedAt: requestedAt, respondedAt: nil, completedAt: nil ) } static func transfer(id: UInt64, direction: TransferDirection, status: TransferStatus) -> Transfer { Transfer( localId: "local-\(id)", transferId: id, direction: direction, status: status, peerId: nil, transferName: "Photos", contentHash: nil, fileCount: 1, totalSize: 1024, ticket: "ticket", accessPolicy: .requireApproval, createdAt: 0, updatedAt: 0 ) } } /// Polls `condition` on the main actor until true or `timeout` elapses. Used to /// await the models' internal `Task`s, which XCTest can't join directly. @MainActor func waitUntil(timeout: TimeInterval = 2, _ condition: @escaping () -> Bool) async { let deadline = Date().addingTimeInterval(timeout) while !condition() && Date() < deadline { try? await Task.sleep(nanoseconds: 5_000_000) } }