session-ios/Signal/test/SignalBaseTest.m
2018-10-16 13:01:24 -04:00

55 lines
1 KiB
Objective-C

//
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
//
#import "SignalBaseTest.h"
#import "Environment.h"
#import <SignalServiceKit/OWSPrimaryStorage.h>
#import <SignalServiceKit/TestAppContext.h>
NS_ASSUME_NONNULL_BEGIN
@implementation SignalBaseTest
- (void)setUp
{
OWSLogInfo(@"");
[super setUp];
ClearCurrentAppContextForTests();
[Environment clearSharedForTests];
[SSKEnvironment clearSharedForTests];
SetCurrentAppContext([TestAppContext new]);
[MockSSKEnvironment activate];
[MockEnvironment activate];
}
- (void)tearDown
{
OWSLogInfo(@"");
[super tearDown];
}
- (void)readWithBlock:(void (^)(YapDatabaseReadTransaction *transaction))block
{
OWSAssert(block);
[[SSKEnvironment.shared.primaryStorage newDatabaseConnection] readWithBlock:block];
}
- (void)readWriteWithBlock:(void (^)(YapDatabaseReadWriteTransaction *transaction))block
{
OWSAssert(block);
[[SSKEnvironment.shared.primaryStorage newDatabaseConnection] readWriteWithBlock:block];
}
@end
NS_ASSUME_NONNULL_END