136 lines
5.6 KiB
Objective-C
136 lines
5.6 KiB
Objective-C
//
|
|
// Copyright (c) 2019 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "MockSSKEnvironment.h"
|
|
#import "ContactDiscoveryService.h"
|
|
#import "OWS2FAManager.h"
|
|
#import "OWSAttachmentDownloads.h"
|
|
#import "OWSBatchMessageProcessor.h"
|
|
#import "OWSBlockingManager.h"
|
|
#import "OWSDisappearingMessagesJob.h"
|
|
#import "OWSFakeCallMessageHandler.h"
|
|
#import "OWSFakeContactsUpdater.h"
|
|
#import "OWSFakeMessageSender.h"
|
|
#import "OWSFakeNetworkManager.h"
|
|
#import "OWSFakeProfileManager.h"
|
|
#import "OWSIdentityManager.h"
|
|
#import "OWSMessageDecrypter.h"
|
|
#import "OWSMessageManager.h"
|
|
#import "OWSMessageReceiver.h"
|
|
#import "OWSOutgoingReceiptManager.h"
|
|
#import "OWSPrimaryStorage.h"
|
|
#import "OWSReadReceiptManager.h"
|
|
#import "TSAccountManager.h"
|
|
#import "TSSocketManager.h"
|
|
#import <SignalServiceKit/SignalServiceKit-Swift.h>
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
#ifdef DEBUG
|
|
|
|
@interface OWSPrimaryStorage (Tests)
|
|
|
|
@property (atomic) BOOL areAsyncRegistrationsComplete;
|
|
@property (atomic) BOOL areSyncRegistrationsComplete;
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
@implementation MockSSKEnvironment
|
|
|
|
+ (void)activate
|
|
{
|
|
MockSSKEnvironment *instance = [self new];
|
|
[self setShared:instance];
|
|
[instance configure];
|
|
}
|
|
|
|
- (instancetype)init
|
|
{
|
|
OWSPrimaryStorage *primaryStorage = [MockSSKEnvironment createPrimaryStorageForTests];
|
|
id<ContactsManagerProtocol> contactsManager = [OWSFakeContactsManager new];
|
|
TSNetworkManager *networkManager = [OWSFakeNetworkManager new];
|
|
OWSMessageSender *messageSender = [OWSFakeMessageSender new];
|
|
SSKMessageSenderJobQueue *messageSenderJobQueue = [SSKMessageSenderJobQueue new];
|
|
|
|
OWSMessageManager *messageManager = [[OWSMessageManager alloc] initWithPrimaryStorage:primaryStorage];
|
|
OWSBlockingManager *blockingManager = [[OWSBlockingManager alloc] initWithPrimaryStorage:primaryStorage];
|
|
OWSIdentityManager *identityManager = [[OWSIdentityManager alloc] initWithPrimaryStorage:primaryStorage];
|
|
id<OWSUDManager> udManager = [[OWSUDManagerImpl alloc] initWithPrimaryStorage:primaryStorage];
|
|
OWSMessageDecrypter *messageDecrypter = [[OWSMessageDecrypter alloc] initWithPrimaryStorage:primaryStorage];
|
|
OWSBatchMessageProcessor *batchMessageProcessor =
|
|
[[OWSBatchMessageProcessor alloc] initWithPrimaryStorage:primaryStorage];
|
|
OWSMessageReceiver *messageReceiver = [[OWSMessageReceiver alloc] initWithPrimaryStorage:primaryStorage];
|
|
TSSocketManager *socketManager = [[TSSocketManager alloc] init];
|
|
TSAccountManager *tsAccountManager = [[TSAccountManager alloc] initWithPrimaryStorage:primaryStorage];
|
|
OWS2FAManager *ows2FAManager = [[OWS2FAManager alloc] initWithPrimaryStorage:primaryStorage];
|
|
OWSDisappearingMessagesJob *disappearingMessagesJob =
|
|
[[OWSDisappearingMessagesJob alloc] initWithPrimaryStorage:primaryStorage];
|
|
ContactDiscoveryService *contactDiscoveryService = [[ContactDiscoveryService alloc] initDefault];
|
|
OWSReadReceiptManager *readReceiptManager = [[OWSReadReceiptManager alloc] initWithPrimaryStorage:primaryStorage];
|
|
OWSOutgoingReceiptManager *outgoingReceiptManager =
|
|
[[OWSOutgoingReceiptManager alloc] initWithPrimaryStorage:primaryStorage];
|
|
id<SSKReachabilityManager> reachabilityManager = [SSKReachabilityManagerImpl new];
|
|
id<OWSSyncManagerProtocol> syncManager = [[OWSMockSyncManager alloc] init];
|
|
id<OWSTypingIndicators> typingIndicators = [[OWSTypingIndicatorsImpl alloc] init];
|
|
OWSAttachmentDownloads *attachmentDownloads = [[OWSAttachmentDownloads alloc] init];
|
|
|
|
self = [super initWithContactsManager:contactsManager
|
|
messageSender:messageSender
|
|
messageSenderJobQueue:messageSenderJobQueue
|
|
profileManager:[OWSFakeProfileManager new]
|
|
primaryStorage:primaryStorage
|
|
contactsUpdater:[OWSFakeContactsUpdater new]
|
|
networkManager:networkManager
|
|
messageManager:messageManager
|
|
blockingManager:blockingManager
|
|
identityManager:identityManager
|
|
udManager:udManager
|
|
messageDecrypter:messageDecrypter
|
|
batchMessageProcessor:batchMessageProcessor
|
|
messageReceiver:messageReceiver
|
|
socketManager:socketManager
|
|
tsAccountManager:tsAccountManager
|
|
ows2FAManager:ows2FAManager
|
|
disappearingMessagesJob:disappearingMessagesJob
|
|
contactDiscoveryService:contactDiscoveryService
|
|
readReceiptManager:readReceiptManager
|
|
outgoingReceiptManager:outgoingReceiptManager
|
|
reachabilityManager:reachabilityManager
|
|
syncManager:syncManager
|
|
typingIndicators:typingIndicators
|
|
attachmentDownloads:attachmentDownloads];
|
|
|
|
if (!self) {
|
|
return nil;
|
|
}
|
|
|
|
self.callMessageHandler = [OWSFakeCallMessageHandler new];
|
|
self.notificationsManager = [NoopNotificationsManager new];
|
|
return self;
|
|
}
|
|
|
|
+ (OWSPrimaryStorage *)createPrimaryStorageForTests
|
|
{
|
|
OWSPrimaryStorage *primaryStorage = [[OWSPrimaryStorage alloc] initStorage];
|
|
[OWSPrimaryStorage protectFiles];
|
|
return primaryStorage;
|
|
}
|
|
|
|
- (void)configure
|
|
{
|
|
__block dispatch_semaphore_t semaphore = dispatch_semaphore_create(0);
|
|
[OWSStorage registerExtensionsWithMigrationBlock:^() {
|
|
dispatch_semaphore_signal(semaphore);
|
|
}];
|
|
dispatch_semaphore_wait(semaphore, DISPATCH_TIME_FOREVER);
|
|
}
|
|
|
|
@end
|
|
|
|
#endif
|
|
|
|
NS_ASSUME_NONNULL_END
|