You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
session-ios/SignalServiceKit/src/TestUtils/OWSFakeMessageSender.m

79 lines
2.1 KiB
Objective-C

//
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
//
#import "OWSFakeMessageSender.h"
#import <SessionServiceKit/SessionServiceKit-Swift.h>
NS_ASSUME_NONNULL_BEGIN
#ifdef DEBUG
@implementation OWSFakeMessageSender
- (void)sendMessage:(TSOutgoingMessage *)message
success:(void (^)(void))successHandler
failure:(void (^)(NSError *error))failureHandler
{
if (self.stubbedFailingError) {
failureHandler(self.stubbedFailingError);
} else {
successHandler();
}
if (self.sendMessageWasCalledBlock) {
self.sendMessageWasCalledBlock(message);
}
}
- (void)sendMessage:(OWSMessageSend *)messageSend
{
if (self.stubbedFailingError) {
messageSend.failure(self.stubbedFailingError);
} else {
messageSend.success();
}
if (self.sendMessageWasCalledBlock) {
self.sendMessageWasCalledBlock(messageSend.message);
}
}
- (void)sendAttachment:(DataSource *)dataSource
contentType:(NSString *)contentType
sourceFilename:(nullable NSString *)sourceFilename
inMessage:(TSOutgoingMessage *)outgoingMessage
success:(void (^)(void))successHandler
failure:(void (^)(NSError *error))failureHandler
{
if (self.stubbedFailingError) {
failureHandler(self.stubbedFailingError);
} else {
successHandler();
}
if (self.sendAttachmentWasCalledBlock) {
self.sendAttachmentWasCalledBlock(outgoingMessage);
}
}
- (void)sendTemporaryAttachment:(DataSource *)dataSource
contentType:(NSString *)contentType
inMessage:(TSOutgoingMessage *)outgoingMessage
success:(void (^)(void))successHandler
failure:(void (^)(NSError *error))failureHandler
{
if (self.stubbedFailingError) {
failureHandler(self.stubbedFailingError);
} else {
successHandler();
}
if (self.sendTemporaryAttachmentWasCalledBlock) {
self.sendTemporaryAttachmentWasCalledBlock(outgoingMessage);
}
}
@end
#endif
NS_ASSUME_NONNULL_END