Signal-iOS/SignalServiceKit/tests/SSKBaseTestObjC.m
Matthew Chen 86b8eb08b8 Remove YapDatabase.
Apply asset from design.

Fix rebase breakage.
2021-03-25 11:41:16 -03:00

65 lines
1.3 KiB
Objective-C

//
// Copyright (c) 2021 Open Whisper Systems. All rights reserved.
//
#import "SSKBaseTestObjC.h"
#import "SSKEnvironment.h"
#import "TestAppContext.h"
#import <CocoaLumberjack/CocoaLumberjack.h>
#import <CocoaLumberjack/DDTTYLogger.h>
#import <SignalServiceKit/SDSDatabaseStorage+Objc.h>
#import <SignalServiceKit/SignalServiceKit-Swift.h>
NS_ASSUME_NONNULL_BEGIN
#ifdef DEBUG
@implementation SSKBaseTestObjC
- (void)setUp
{
OWSLogInfo(@"%@ setUp", self.logTag);
[super setUp];
[DDLog addLogger:DDTTYLogger.sharedInstance];
ClearCurrentAppContextForTests();
SetCurrentAppContext([TestAppContext new]);
[MockSSKEnvironment activate];
[GroupManager forceV1Groups];
}
- (void)tearDown
{
OWSLogInfo(@"%@ tearDown", self.logTag);
OWSAssertIsOnMainThread();
// Spin the main run loop to flush any remaining async work.
__block BOOL done = NO;
dispatch_async(dispatch_get_main_queue(), ^{ done = YES; });
while (!done) {
(void)CFRunLoopRunInMode(kCFRunLoopDefaultMode, 0.0, true);
}
[super tearDown];
}
-(void)readWithBlock:(void (^)(SDSAnyReadTransaction *))block
{
[SDSDatabaseStorage.shared readWithBlock:block];
}
-(void)writeWithBlock:(void (^)(SDSAnyWriteTransaction *))block
{
DatabaseStorageWrite(SDSDatabaseStorage.shared, block);
}
@end
#endif
NS_ASSUME_NONNULL_END