36 lines
1.0 KiB
Swift
36 lines
1.0 KiB
Swift
//
|
|
// Copyright 2022 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import Foundation
|
|
|
|
#if TESTABLE_BUILD
|
|
|
|
class FakeMessageSender: MessageSender {
|
|
public var stubbedFailingErrors = [Error?]()
|
|
public var sentMessages = [TSOutgoingMessage]()
|
|
public var sendMessageWasCalledBlock: ((TSOutgoingMessage) -> Void)?
|
|
|
|
init() {
|
|
super.init(groupSendEndorsementStore: GroupSendEndorsementStoreImpl())
|
|
}
|
|
|
|
override func sendMessage(_ preparedMessage: PreparedOutgoingMessage) async throws {
|
|
try await preparedMessage.send { message in
|
|
sendMessageWasCalledBlock?(message)
|
|
sentMessages.append(message)
|
|
}
|
|
if let stubbedFailingError = stubbedFailingErrors.removeFirst() { throw stubbedFailingError }
|
|
}
|
|
|
|
override func sendTransientContactSyncAttachment(
|
|
dataSource: DataSource,
|
|
thread: TSThread
|
|
) async throws {
|
|
if let stubbedFailingError = stubbedFailingErrors.removeFirst() { throw stubbedFailingError }
|
|
}
|
|
}
|
|
|
|
#endif
|